Skip to content
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

Open
chenyang9799 opened this issue Mar 20, 2020 · 7 comments
Open

修改损失函数 #19

chenyang9799 opened this issue Mar 20, 2020 · 7 comments

Comments

@chenyang9799
Copy link

如果我要修改损失函数CIOU,我只需要修改src/box.c的float box_ciou(box a, box b)函数吗

@Zzh-tju
Copy link
Owner

Zzh-tju commented Mar 20, 2020

那个只是可视化loss,实际应在dx_box_iou里修改

@Zzh-tju
Copy link
Owner

Zzh-tju commented Mar 20, 2020

如果你只是要使用CIoU,则在cfg里的每个yolo层修改loss类型

@jialiang99
Copy link

那个只是可视化loss,实际应在dx_box_iou里修改

我想请教下修改iou损失函数的细节,以及如何用wins训练。作者能留个联系方式吗

@Zzh-tju
Copy link
Owner

Zzh-tju commented May 28, 2020

我想请教下修改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)

有任何问题都可以在这里留下,我有时间会回答,其他渠道的联系就不必了,我已经忙的不可开交了。

@jialiang99
Copy link

我想请教下修改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)
有任何问题都可以在这里留下,我有时间会回答,其他渠道的联系就不必了,我已经忙的不可开交了。

好的,非常感谢

@henbucuoshanghai
Copy link

float box_ciou(box a, box b)函数 可视化loss?训练的时候屏幕显示的loss? 测试集的时候用的?

@alamsyahamin
Copy link

在AlexeyAB的代码中,我认为beta1在diounms中可能无法正常工作,或者与您的情况不同。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

5 participants