DeepLabV3 +
使用PyTorch的语义分割实现
火车
运行python train.py进行培训
数据集结构(类似于CamVid数据集)
├── Dataset folder
├── train
├── 1111.png
├── 2222.png
├── train_labels
├── 1111_L.png
├── 2222_L.png
├── class_dict.csv
笔记
默认功能提取器是
更改培训配置,更改utils/config.py参数
默认的损失函数是weighted cross entropy
1