PeterH0323/Construction-Crack-Segmentation

This project for Construction-Crack-Segmentation

一、环境配置

# python3
conda create --name=Construction-Crack-Segmentation python=3.7
conda activate Construction-Crack-Segmentation
pip install -r requirements.txt

二、生成训练集、验证集、测试集

1. 打开 dataset\gen_train_val_test_dataset.py，修改好路径，执行

   python3 dataset\gen_train_val_test_dataset.py

2. 将生成好的数据集，根据 dataset文件夹中的结构放入即可

三、训练

1. 修改 train.py 里面的参数，执行下面进行训练

python train.py

四、测试

1. 修改 test.py 里面的参数，执行下面进行测试

python test.py --checkpoint 权重路径 

五、推理

1. 修改 predict.py 里面的参数，执行下面进行推理

python predict.py

六、参考

[1] https://github.com/xiaoyufenfei/Efficient-Segmentation-Networks