search

bubbliiiing / yolov4-tf2

这是一个yolo4-tf2（tensorflow2）的源码，可以用于训练自己的模型。
MIT License
314 stars 111 forks source link

DIoU-NMS #14

Open ZF4444 opened 3 years ago

ZF4444 commented 3 years ago

你好，感谢你的分享，请问代码中 https://github.com/bubbliiiing/yolov4-tf2/blob/a4cf127a49792a296db549d006dbef4f452770d1/nets/yolo4.py#L238 使用的NMS好像不是论文中使用的DIoU-NMS是出于什么考虑呢？谢谢！

bubbliiiing commented 3 years ago

是出于写DIOU很麻烦的考虑，直接用tf库里面的还快一些