mmdet训练cityscapes数据集-maskrcnn

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

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

mmdet训练cityscapes数据集-maskrcnn_第1张图片

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

mmdet训练cityscapes数据集-maskrcnn_第2张图片

3.放置文件到以下位置

mmdet训练cityscapes数据集-maskrcnn_第3张图片

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

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

mmdet训练cityscapes数据集-maskrcnn_第4张图片

mmdet训练cityscapes数据集-maskrcnn_第5张图片

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

mmdet训练cityscapes数据集-maskrcnn_第6张图片

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

mmdet训练cityscapes数据集-maskrcnn_第7张图片

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

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

mmdet训练cityscapes数据集-maskrcnn_第8张图片

 

 

你可能感兴趣的:(深度学习,人工智能)