本期给大家介绍的是126基于deeplabv3+图像分割检测墙体裂缝,效果图如下:
代码下载或视频演示地址:
126基于deeplabv3+图像分割检测墙体裂缝_哔哩哔哩_bilibili
代码整体是非常简便的,主要是dataset数据集文件夹和3个py文件。
运行01json转mask格式.py能将图片数据集转成标准格式,并划分训练集和验证集。
运行02train.py会对标准格式的数据集进行训练,训练好的模型保存在checkpoints文件夹下。并生成评价指标图在IOU。
运行03pyqt.py可以选择自己感兴趣的图片进行识别,通过调用上一步训练好的模型识别图片中的墙体裂缝。
注:
下载本代码环境自行安装
安装python和pytorch即可运行,
安装环境可参考博客: