是否可以将yolo3与yolo9000权重一起使用以进行更多类学习?

时间:2019-09-09 12:05:43

标签: yolo darknet

现在我正在使用yolo3,但它没有全部9000个类。

是否可以将yolo3与yolo9000类和权重一起使用?

我想拥有尽可能多的对象来为某些图片分配标签,并且我不确定是否应该使用yolo9000还是可以将yolo3与9000类一起使用。

更新: 因此,我厌倦了用yolo3加载权重yolo9000,并且该模型找不到任何对象。看来我需要调整其他内容。

1 个答案:

答案 0 :(得分:0)

知道了, 调用检测时需要更改阈值,默认值为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)