YOLOv4检测

Question

我已经训练了6个班级的YOLOv4，在看到视频上的结果后，我想删除一个班级(id=3)，因为我发现它对我的项目没有必要。我是否可以在视频上使用相同的模型，并强制其忽略某个类，并且只检测其他5个类，而不重新训练5个类上的模型。谢谢

Answer 1

不幸的是，经过训练的模型将有一个最终的fcn层，也就是网格单元，每个网格单元都有一个(B x (5 + C))维度。

抑制该类的可能方法是在后期处理期间，您可以根据id过滤出类。

Answer 2

我认为在神经网络端是不可能的，也许你可以尝试在应用程序端修改它，比如：

# get detections
# perform NMS
for index in indices:
    if classes[class_ids[index] == "something I want to avoid":
        pass
    else:
        # draw the bounding box

**注意：**代码松散地基于Learn OpenCV博客中的this示例。