unet图像分割网络对墙体裂缝图像进行训练，得到的模型可以对图片中带有裂缝的墙体进行识别。
整个代码运行流程已经放在b站上找002期：
到此一游7758258的个人空间_哔哩哔哩_bilibili
效果图如下所示：
数据集如下：
完整的代码文件放置如下：
其中datasets文件夹下放置的是标签文件json格式和图裂缝瑕疵图片
通过运行01make_dataset.py可以将json文件生成png的标签格式
再运行02train.py会对数据集进行读取进行训练，最后训练得到的模型放置在weights文件夹下
最后运行03predict.py文件即可调用训练好的模型进行识别，比如识别当前文件夹下的102.jpg图像。
版权声明：本文为bug_creat0r原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。