002unet墙体瑕疵检测-python-pytorch

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

整个代码运行流程已经放在b站上找002期：

到此一游7758258的个人空间_哔哩哔哩_bilibili

效果图如下所示：

数据集如下：

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

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

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

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

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