search

Alan-D-Chen / CDIoU-CDIoUloss

🔥CDIoU and CDIoU loss is like a convenient plug-in that can be used in multiple models. CDIoU and CDIoU loss have different excellent performances in several models such as Faster R-CNN, YOLOv4, RetinaNet and . There is a maximum AP improvement of 1.9% and an average AP of 0.8% improvement on MS COCO dataset, compared to traditional evaluation-feedback modules. Here we just use as an example to illustrate the code.
Other
27 stars 6 forks source link

loss #5

Open cymdhx opened 3 years ago

cymdhx commented 3 years ago

你好 我想问一下定义的yolo_loss函数是在哪 还想问一下cdiou的 λ 在yolo网络里等于多少好一点 谢谢啦

Alan-D-Chen commented 3 years ago

It （lambda）depends on the circumstances. You can refer to the situation of ATSS. The code here is an example applied to ATSS. You just need to modify the code of Yolo by yourself. It's very simple.