-
Notifications
You must be signed in to change notification settings - Fork 85
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
修改损失函数 #19
Comments
那个只是可视化loss,实际应在dx_box_iou里修改 |
如果你只是要使用CIoU,则在cfg里的每个yolo层修改loss类型 |
我想请教下修改iou损失函数的细节,以及如何用wins训练。作者能留个联系方式吗 |
所有细节全部在src/box.c,或者参考模拟实验https://github.com/Zzh-tju/DIoU pytorch的版本也可以在里面找到链接。 目前此版本的darknet不支持windows,如有需要,请使用AlexeyAB的darknet (https://github.com/AlexeyAB/darknet),该repo也可以使用CIoU,且包含了更多实用的trick,也即目前最新的YOLOv4 (https://arxiv.org/abs/2004.10934) 有任何问题都可以在这里留下,我有时间会回答,其他渠道的联系就不必了,我已经忙的不可开交了。 |
好的,非常感谢 |
float box_ciou(box a, box b)函数 可视化loss?训练的时候屏幕显示的loss? 测试集的时候用的? |
在AlexeyAB的代码中,我认为beta1在diounms中可能无法正常工作,或者与您的情况不同。 |
如果我要修改损失函数CIOU,我只需要修改src/box.c的float box_ciou(box a, box b)函数吗
The text was updated successfully, but these errors were encountered: