一、环境配置
参考我之前的博客
二、配置数据集
这里我使用的数据集是Cityscapes
按照官方文档准备数据集如下
这里遇到的问题,如果无法使用git就直接去官网下载文件夹,然后直接python文件夹中的py文件
红框圈出来是新生成的,它与labelids的区别是,labelids是像素值表示类别,而labelTrainids是将对应的类型通过一定的映射规则映射到0-255,这里我还没弄懂数据读入是怎么进行的,回头再来填坑
三、运行训练代码
(1)将数据集放到/datasets/cityscapes/gtFine/train
至于为什么我还不清楚
(2)运行训练代码
python train_net.py \
--config-file ./configs/cityscapes/semantic-segmentation/maskformer2_R50_bs16_90k.yaml \
--num-gpus 1 SOLVER.IMS_PER_BATCH 2 SOLVER.BASE_LR 0.0001
遇到的问题:
"Default process group has not been initialized, " RuntimeError: Default process group has not been initialized, please make sure to call init_process_group.
解决办法:参考
将yaml文件NORM改了就能运行了