现在我正在使用yolo3,但它没有所有的9000个类。
是否可以将 yolo3 与 yolo9000 类和权重一起使用?
我希望有尽可能多的对象为某些图片分配标签,但我不确定我是否应该使用 yolo9000 或者是否可以将 yolo3 与 9000 个类一起使用。
更新: 所以,我厌倦了用 yolo9000 加载权重 yolo3,模型找不到任何对象。看来我需要调整其他东西。
知道了, 调用检测时需要更改阈值,默认值为 0.5
https://github.com/pjreddie/darknet/blob/61c9d02ec461e30d55762ec7669d6a1d3c356fb2/python/darknet.py#L125
det = darknet.detect(self._net, self._meta, im, thresh=.1, hier_thresh=.1, **kwargs)