复现Mask2Former从头开始

一、环境配置

参考我之前的博客

二、配置数据集

这里我使用的数据集是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改了就能运行了

  • 9
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值