How to modify the categories of the coco dataset?

[MACHUNHAI]

在mmyolo中训练yolov8时，使用的是数据集是自己的coco格式的数据集（在mmdetection中可用），训练之后跑测试，发现检测出的类别还只是标准的coco数据集的类别，请问大佬如何修改coco数据集的类别，需要把coco数据集重新做成yolo格式的吗？目前的目录如下：

[josh3255]

If you initialized the model with init_detector function as follows

model = init_detector(args.config, args.checkpoint, device=args.device)

try adding the code as below

model.dataset_meta['classes'] = ('your', 'custom', 'classes')