yolov4-pytorch icon indicating copy to clipboard operation
yolov4-pytorch copied to clipboard

训练两个epoch后loss就变为零点几

Open yuxuan-wang-tj opened this issue 2 years ago • 21 comments

大佬您好,我采用新版的Pytorch训练火焰识别,数据集2100张,采用yolo4_voc_weights.pth,训练完成之后loss很低但是AP只有0.125,请问有什么解决方法吗

yuxuan-wang-tj avatar Apr 07 '22 13:04 yuxuan-wang-tj

而且刚刚发现lr是3*1e-5,不太明白这个,不是已经设置了Min_lr = Init_lr * 0.01=1e-4吗

yuxuan-wang-tj avatar Apr 07 '22 14:04 yuxuan-wang-tj

写了会学习率自动调整。

bubbliiiing avatar Apr 08 '22 14:04 bubbliiiing

要看训练了多久吧

bubbliiiing avatar Apr 08 '22 14:04 bubbliiiing

要看训练了多久吧

目前训练了120epoch解冻,50epoch冻结,AP是26%,需要继续训练吗

yuxuan-wang-tj avatar Apr 08 '22 15:04 yuxuan-wang-tj

之前多少……你有用过吗,一般多少?

bubbliiiing avatar Apr 08 '22 16:04 bubbliiiing

之前多少……你有用过吗,一般多少?

之前使用darknet编译训练的,收敛之后AP差不多88%

yuxuan-wang-tj avatar Apr 09 '22 02:04 yuxuan-wang-tj

之前多少……你有用过吗,一般多少?

大佬我之前冻结50训练50Epoch后停止了,然后更换了权重之后重新进行解冻训练,这样会不会导致相同的图像被训练了两遍呢。

yuxuan-wang-tj avatar Apr 09 '22 04:04 yuxuan-wang-tj

我也是这个问题

503092317 avatar Apr 15 '22 03:04 503092317

我也是这个问题

loss低并不是问题……

bubbliiiing avatar Apr 15 '22 17:04 bubbliiiing

之前多少……你有用过吗,一般多少?

大佬我之前冻结50训练50Epoch后停止了,然后更换了权重之后重新进行解冻训练,这样会不会导致相同的图像被训练了两遍呢。 每个epoch本来就会遍历所有的图像

bubbliiiing avatar Apr 15 '22 17:04 bubbliiiing

在v5上80 v3上百分之二十 

------------------ 原始邮件 ------------------ 发件人: Bubbliiiing @.> 发送时间: 2022年4月16日 01:37 收件人: bubbliiiing/yolov4-pytorch @.> 抄送: 503092317 @.>, Comment @.> 主题: Re: [bubbliiiing/yolov4-pytorch] 训练两个epoch后loss就变为零点几 (Issue #291)

之前多少……你有用过吗,一般多少?

大佬我之前冻结50训练50Epoch后停止了,然后更换了权重之后重新进行解冻训练,这样会不会导致相同的图像被训练了两遍呢。 每个epoch本来就会遍历所有的图像

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

503092317 avatar Apr 15 '22 23:04 503092317

v5是我提供的那个么

bubbliiiing avatar Apr 18 '22 16:04 bubbliiiing

v5是我提供的那个么

不是 用的u版

503092317 avatar Apr 19 '22 01:04 503092317

那你把我的改成300epoch

bubbliiiing avatar Apr 19 '22 02:04 bubbliiiing

那你把我的改成300epoch

使用yolov3,加载预训练权重 使用416416输入 ,默认anchor 300代, 精度54 使用yolov3 ,加载预训练权重 使用640640输入 ,使用k-means调整anchor 300代 ,使用640640推理精度30,使用640640推理精度50 由于想使用up的code做对比实验,所以想使用640640作为输入,我的数据集是基本上是780120,不知道为什么精度会这样

503092317 avatar Apr 20 '22 03:04 503092317

你的图像780x120,这宽高比例太骚了吧0 0,先验框也要重新调整一下,这种预测得开lettebox_image。

bubbliiiing avatar Apr 20 '22 14:04 bubbliiiing

哇!  B导 精度暴涨到80比我改进的v5还高!我用你的第一版v3精度会低一点吗?

------------------ 原始邮件 ------------------ 发件人: Bubbliiiing @.> 发送时间: 2022年4月20日 22:58 收件人: bubbliiiing/yolov4-pytorch @.> 抄送: 503092317 @.>, Comment @.> 主题: Re: [bubbliiiing/yolov4-pytorch] 训练两个epoch后loss就变为零点几 (Issue #291)

503092317 avatar Apr 20 '22 15:04 503092317

你咋做的0 0按我说的调先验框嘛

bubbliiiing avatar Apr 21 '22 14:04 bubbliiiing

是的,调整先验框,然后预测的时候,使用lettebox_image!

------------------ 原始邮件 ------------------ 发件人: Bubbliiiing @.> 发送时间: 2022年4月21日 22:49 收件人: bubbliiiing/yolov4-pytorch @.> 抄送: 503092317 @.>, Comment @.> 主题: Re: [bubbliiiing/yolov4-pytorch] 训练两个epoch后loss就变为零点几 (Issue #291)

你咋做的0 0按我说的调先验框嘛

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

503092317 avatar Apr 21 '22 16:04 503092317

好的~很强

bubbliiiing avatar Apr 22 '22 14:04 bubbliiiing

感谢B导

------------------ 原始邮件 ------------------ 发件人: Bubbliiiing @.> 发送时间: 2022年4月22日 22:02 收件人: bubbliiiing/yolov4-pytorch @.> 抄送: 503092317 @.>, Comment @.> 主题: Re: [bubbliiiing/yolov4-pytorch] 训练两个epoch后loss就变为零点几 (Issue #291)

好的~很强

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

503092317 avatar Apr 22 '22 14:04 503092317