Yolo v4用于pytorch,tensorflow渴望模式和onnx(通过Trident api)
感谢
所有预先准备好的模型权重和cfg均来自官方网站:
还要感谢Ultralytics的项目,它确实很棒而且很有帮助。
yolo v4的搜索结果
让我们看看有关yolo v4(pytorch后端)的出色性能!!
更新(5/3):增强小物品
在yolo v4中,缺少缺少对小物件的检测的缺点。 我试图解决短缺问题。 我发现解决此问题的最佳方法是在stride = 8 Yolo Layer(76 * 76)中修改对象
您所需要做的就是设置YoloLayer small_item_enhance = True(仅效果76 * 76 head)
for module in detector.model.modules():
if isinstance(module,Yolo
1