yolov4-pytorch 替换损失函数

替换损失函数

Open SCIKings opened this issue 2 years ago • 5 comments

导师，您好！我想问一下，在YOLOv4中，损失函数替换为focalloss的话，def fit_one_epoch函数中的前向传播、反向传播需要保留吗，或者说utils_fit_py文件有什么需要改的呢，（我参考的是您efficientdet里面focalloss的写法）

Apr 29 '22 11:04 SCIKings

@SCIKings 你好，这个版本里面已经集成Focal loss了，可以在nets/yolo_training.py 以及train.py中查看

Apr 29 '22 13:04 XuecWu

确实添加了…可以调用，但效果一般

Apr 29 '22 13:04 bubbliiiing

如果使用了效果好可以反馈

Apr 29 '22 13:04 bubbliiiing

铁子，这个版本是指哪个版本啊，我说的是B导Yolov4的程序哦，里面的损失貌似用的：1-CIOU和 BCE 我想把focalloss用在yolov4中试试，大佬们，该咋个搞勒

------------------ 原始邮件 ------------------ 发件人: "bubbliiiing/yolov4-pytorch" @.>; 发送时间: 2022年4月29日(星期五) 晚上9:38 @.>; @.@.>; 主题: Re: [bubbliiiing/yolov4-pytorch] 替换损失函数 (Issue #304)

@SCIKings 你好，这个版本里面已经集成Focal loss了，可以在nets/yolo_training.py 以及train.py中查看

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you were mentioned.Message ID: @.***>

Apr 29 '22 16:04 SCIKings

train.py不是直接给了一个focal loss的接口用于设置嘛

Apr 30 '22 11:04 bubbliiiing

yolov4-pytorch yolov4-pytorch copied to clipboard

替换损失函数

yolov4-pytorch
yolov4-pytorch copied to clipboard