数据集官方格式:
- 在
mmseg/datasets
下对数据集进行初始定义
- 在
configs/_ _base_ _/datasets
下对数据加载进行定义
- 在
configs/
下选择需要的模型参数进行修改
找了两个模型fcn和danet进行训练
类别数应该等于 n + 1
, 也就是多少类别+背景。46类应该是47
- 返回
tools/train.py
进行训练
python tools/train.py ${配置文件} --work-dir ${YOUR_WORK_DIR}
- 训练结果
第2000次迭代:
第20000次迭代:
结果预料之中的不好。
- After
dms数据集分类问题 【dms数据集有52个类别,但是在6个类别上表现不佳,所以选择了46种材料标签作为基准。【合并类别
先解决数据集问题、再用segnext模型训练【最新的模型?】