我正在使用yolov5开发一个头盔检测模型,我只需要检测头盔级的安全帽和其他头盔(自行车头盔、板球头盔、无头盔)为无头盔级。那么,是否有任何方法可以在上执行对象检测,没有头盔类而不标记数据?我们能在这样的数据集上进行异常检测吗?
发布于 2022-05-17 16:11:42
YOLO和其他任何物体检测模型一样,会输出那些东西,你会给它们贴上标签--如果你只给安全帽贴上标签,受过良好训练的模型会试着只给你提供安全帽,而忽略所有其他类型的头盔(或者没有头盔的头)。如果你想检测到“没有头盔”,你需要给另一个类别贴上“没有头盔”的标签。
当然,这里可以使用异常检测(例如,简单的编解码结构)以上检测(异常检测模型的输入将输出,包围盒,YOLO)。结果可能会更好,但是这个解决方案将更加资源密集型(运行两个模型和一个模型)。
https://stackoverflow.com/questions/72253865
复制相似问题