mmdet训练cityscapes数据集-maskrcnn

由于网上找不到对应的json文件,却不知道该文件可以通过cityscapesscrips生成,经过查资料发现,应该这样做。

1.去github上,找到这个脚本(mcordts/cityscapesScripts),下载下来。

2.在cityscapes官网下载(Login – Cityscapes Dataset)数据集的前三个。

3.放置文件到以下位置

 4.这一步还不清楚是什么作用,跟着网上的跑了一遍,用pycharm打开这个项目,在cityscapes->preparation->createTrainIdLabelImgs,运行这个py文件,如果路径放对,一般就不会报错。

5.打开mmdet工程,找到数据集转换的这部分代码,如下图,import原本应该是爆红的,在tools同级目录下将cityscapes脚本文件复制过来,就不会报错了。

6.在5不会报错之后,运行该文件,需要配置数据集和输出annotation的位置。根据自己下载数据集的路径进行配置,只要保证,--cityscapes_path写的路径下有leftImg8bit和gtFine文件。

 7.在configs->base->datasets->cityscapes_instance.py中配置图片和annotation的位置data_root。我的文件摆放如下:

 8.配置maskrcnn文件。在configs/_base_/models/mask_rcnn_r50_fpn.py中修改类别数num_classes=8,

 9.在train.py文件中,选择maskrcnn对应的配置文件。进行训练,项目正常启动。

 

 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值