YuHengsss / YOLOV

This repo is an implementation of PyTorch version YOLOV Series
Apache License 2.0
278 stars 39 forks source link

如果换成自定义数据的类别,需要重新训练yolov还是单独的yolox #63

Closed le02146 closed 9 months ago

le02146 commented 9 months ago

请问如果我现在在自定义的数据集上需要检测某个新类别(不在coco80类), 已经训练好YOLOX了。请问如果在一个视频上进行检测,可以直接用yolov的权重吗,还是需要重新训练yolov ? 谢谢

YuHengsss commented 9 months ago

需要使用 yolox 的权重初始化 yolov 进行训练

le02146 commented 9 months ago

@YuHengsss ,您好,谢谢回复。 我只是在图片上训练一个新的类别A(不在COCO_CLASSES,也不在VID_classes)的图片yolox检测器,然后用到了视频上检测A。 因为视频出现漏检(有的帧没有检测到)所以想利用yolov 来提高。 请问您的意思我还不能直接结合我自己训练的yolox 和yolov的方法吗 (比如简单替换您训练的yolox为我自己的yolox,然后VID_classes换成我自己的类?)?需要我自己标注视频目标检测的数据来重新训练/微调yolov才能用吗 谢谢

YuHengsss commented 9 months ago

是的,因为yolov的分类器与yolox是分离的,需要在yolox的基础上进行finetuning

le02146 commented 9 months ago

这样的话,就比较麻烦了。现在只是用图片的标注数据来训练,没有时间和成本去专门再标注类别A的视频检测数据集。 能不能请教您一下,有没有其他方法可以直接利用图片检测器来提高视频检测效果的,比如跟踪方法可以吗,谢谢您。

YuHengsss commented 9 months ago

可以尝试一些后处理的方法,比如:https://github.com/AlbertoSabater/Robust-and-efficient-post-processing-for-video-object-detection 这种方法不需要训练,能出框的检测器都可以直接拿过来用

le02146 commented 9 months ago

谢谢