是否可以将 yolo3 与 yolo9000 权重用于更多类?



现在我正在使用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)

相关内容

  • 没有找到相关文章

最新更新