search

Stinky-Tofu / Stronger-yolo

🔥Improve yolo with latest paper
MIT License
3 stars 0 forks source link

如何使用focal loss 、soft nms等这些改进呢？ #74

Open zychen2016 opened 5 years ago

zychen2016 commented 5 years ago

在工程中是否都有实现呢？

Stinky-Tofu commented 5 years ago

默认使用了focal loss，nms，使用soft nms需要在evaluator.py中改一下

zychen2016 commented 5 years ago
baseline 84.3 0.0 yes
data agumentation 85.8 +1.5 yes
multi scale train 86.3 +0.5 yes
focal loss 88.3 +2.0 yes
des xxx +1.0 no
group normalization xxx xxx yes
soft nms xxx -0.6 yes
label smooth 88.6 +0.3 yes
cosine learning rate 88.6 0.0 yes
GIOU 88.8 +0.2 yes
multi scale test 90.7 +1.9 yes

下面这些哪些是需要修改代码才能实现的？哪些是已经实现了的？

Stinky-Tofu commented 5 years ago

@zychen2016 都已经实现了，能够提升性能的都默认使用了，不能提升性能的都没有使用，des没有开源