DeepLabv3+训练Crack500

一、训练集

Crack500是一个路面裂缝分割数据集，原本是500张图片，但是因为每张图片分辨率普遍为1440*2560或2560*1440,如下图所示。

由于图像数量有限、每张图像尺寸较大、计算资源有限，作者团队将每张图像裁剪成16个不重叠的图像区域，只保留包含超过1000个裂纹像素的区域，分辨率为640*360，如下图所示。

裁剪后的数据集有图片3368张。

本文将Crack500数据集按照8：1：1，划分为训练集、验证集、测试集，并按照voc数据集格式进行组织。如下图所示。

JPEGImages里放原图片，SegmentationClass里面放mask图片，ImageSets内还有个Segmentation目录，放划分的训练集的txt文件。数据集百度网盘链接：链接：

链接: https://pan.baidu.com/s/1x8aKznkjzba_ENJ5FkQYHg 提取码: 1234 
 

提出该数据集的论文名《Feature Pyramid and Hierarchical Boosting Network for Pavement Crack Detection》，github地址：链接，也提供了他的数据集百度网盘链接，如下图。

二、分割模型

DeepLabv3+是个语义分割模型，2018年的。

代码修改参考该文章：《【windows10】使用pytorch版本deeplabv3+训练自己数据集》

自己踩的坑：

该项目用的PyTorch 0.4.1，比较老，我自己在安装这个版本时，始终不成功。所以我直接使用了yolo11的conda环境，python3.11.4，PyTorch2.4.1。然后其他包，缺什么安装什么就好了。按照上面的文章修改代码之后，输入命令运行，会报一个错误：TypeError: make_grid() got an unexpected keyword argument ‘range‘，解决方法：将range改为value_range，版本更新的问题，参考文章：链接。

最后就可以了。