mmdetection COCO格式数据集
RuntimeError: DataLoader worker (pid 34888) is killed by signal: Terminated.
1、定义数据种类,需要修改的地方在mmdetection/mmdet/datasets/coco.py。把CLASSES的那个tuple改为自己数据集对应的种类tuple即可。例如:
CLASSES = ('belt', 'sunglasses', 'boot', 'cowboy_hat', 'jacket')
1
2、接着在mmdetection/mmdet/core/evaluation/class_names.py修改coco_classes数据集类别,这个关系到后面test的时候结果图中显示的类别名称。例如:
def coco_classes():
return [
'belt', 'sunglasses', 'boot', 'cowboy_hat', 'jacket'
]
1
2
3
4
2
3
4
上次更新: 2021/08/02, 21:04:52
- 02
- README 美化05-20
- 03
- 常见 Tricks 代码片段05-12