Unet墙体裂缝识别-python-pytorch

unet图像分割网络对墙体裂缝图像进行训练,得到的模型可以对图片中带有裂缝的墙体进行识别。

整个代码运行流程已经放在b站上:

银色子弹zg的个人空间-银色子弹zg个人主页-哔哩哔哩视频

效果图如下所示:

Unet墙体裂缝识别-python-pytorch_第1张图片

 数据集如下:

Unet墙体裂缝识别-python-pytorch_第2张图片

 完整的代码文件放置如下:

Unet墙体裂缝识别-python-pytorch_第3张图片

 其中datasets文件夹下放置的是标签文件json格式和图裂缝瑕疵图片

通过运行01make_dataset.py可以将json文件生成png的标签格式

再运行02train.py会对数据集进行读取进行训练,最后训练得到的模型放置在weights文件夹下

最后运行03predict.py文件即可调用训练好的模型进行识别,比如识别当前文件夹下的102.jpg图像。

完整的代码下载地址:

正在为您运送作品详情

你可能感兴趣的:(pytorch,深度学习,神经网络)