bubbliiiing / yolov4-tf2

这是一个yolo4-tf2(tensorflow2)的源码,可以用于训练自己的模型。
MIT License
314 stars 111 forks source link

训练效果问题 #38

Open 11wswhl opened 2 years ago

11wswhl commented 2 years ago

upup 请问为啥yolo4的mAP特别的低啊 别的网络都是80多 这个就特别低

bubbliiiing commented 2 years ago

结合数据集分析,请上图

11wswhl commented 2 years ago

不知道怎么上图啊 别的yolo网络的map0.75都80-90多 但是yolo4却只有40多

11wswhl commented 2 years ago

upup 我b站私发您图片了

bubbliiiing commented 2 years ago

emm你好像发太快了,图都没上传好就确定了

11wswhl commented 2 years ago

抱歉 我再重发一下吧

11wswhl commented 2 years ago

IMG_20220428_103903 IMG_20220428_103912 IMG_20220428_103919 IMG_20220428_103919 IMG_20220428_103925 IMG_20220428_103929 IMG_20220428_103932 IMG_20220428_103939 IMG_20220428_103942 IMG_20220428_103945

bubbliiiing commented 2 years ago

其实主要是检测的内容……这个看不出来的饿

bubbliiiing commented 2 years ago

就是你的训练集啥的,以及标注情况

11wswhl commented 2 years ago

用您之前的yolo4代码 mAP都是80多的 这次也不知道是咋了😢

bubbliiiing commented 2 years ago

主要用了sgd吧,对小数据并不友好

bubbliiiing commented 2 years ago

你换成adam不一定有意外收获。

11wswhl commented 2 years ago

up 我是用的训练集和测试集 在您其他网络上性能都没问题的 那我再重新跑一次看看吧

11wswhl commented 2 years ago

好的 我试一下 之前用的是sgd cos init-lr0.001

bubbliiiing commented 2 years ago

按建议参数设置啊,别看错了。

11wswhl commented 2 years ago

设置成0.001是因为在别的网络上性能更好 为了对比就都统一参数了😖

bubbliiiing commented 2 years ago

emm难怪你效果这么差……不同网络的学习率本来就不一样

bubbliiiing commented 2 years ago

不同优化器的学习率也不一样

11wswhl commented 2 years ago

主要是没想到会这么差。。。。

11wswhl commented 2 years ago

导师怀疑我在造假了。。

bubbliiiing commented 2 years ago

先试试adam吧,按建议参数

11wswhl commented 2 years ago

好的

11wswhl commented 2 years ago

up 我正在重新跑 loss30多没问题吧😄

bubbliiiing commented 2 years ago

正则化的loss在里面,没什么问题

11wswhl commented 2 years ago

up昨天用adam优化器跑了一遍 效果还是好差啊 mAP0.75只有50多 yolo3都有70多 yolo5和yolox都是90多 这个效果是不是问题有点大啊

bubbliiiing commented 2 years ago

yolo5和x用的是我提供的么

11wswhl commented 2 years ago

对 都是您的代码

bubbliiiing commented 2 years ago

确实不合理啊……有没有示例的图片……

bubbliiiing commented 2 years ago

你的测试集什么都没变嘛

11wswhl commented 2 years ago

都没变 用的是一样的配置 而且之前用别的数据集 yolo4也是很低

11wswhl commented 2 years ago

而且yolo4主要是mAP0.75特别低。但是mAP0.5不算低。

11wswhl commented 2 years ago

和其他网络相比

bubbliiiing commented 2 years ago

你这数据集才1000多张是吧

11wswhl commented 2 years ago

数据集1900多张 8:1:1划分的

bubbliiiing commented 2 years ago

训练参数?详细点

11wswhl commented 2 years ago

batch 8 adam cos init-lr0.001; 冻结训练100 epoch 解冻后训练900 epoch; confendence 0.5; nms 0.5;然后选择mAP0.75最优的权重 mAP大概50多 yolo3是70.0几

bubbliiiing commented 2 years ago

confendence 0.5; nms 0.5这个应该不是在get_map.py设置的吧?

11wswhl commented 2 years ago

yolo里设置的

11wswhl commented 2 years ago

momentum和weights -decay是您的默认参数

bubbliiiing commented 2 years ago

……咱还想说关了mosaic= =但发现yolov5和x效果都还不错……咱也不懂

11wswhl commented 2 years ago

尴尬啊 我现在都想编数据了😂

bubbliiiing commented 2 years ago

代码都是新改的代码吗0 0

11wswhl commented 2 years ago

是啊 前天重新下了之后再跑了一次 效果还是差。哈哈 已经想合理预估数据了。

bubbliiiing commented 2 years ago

其它的呢 也是最新的吗0 0

11wswhl commented 2 years ago

都是更新了weights-decay和moment的版本 多gpu的版本有的更新了 有的没更新

bubbliiiing commented 2 years ago

我入土了……要不试试SGD?因为我新改的代码已经是更注重高精度框的拟合的了……入土了……

bubbliiiing commented 2 years ago

1e-2开始这样

11wswhl commented 2 years ago

再跑一次试试吧。不行就合理的编一下数据了。😂

bubbliiiing commented 2 years ago

兄弟……只能说建议关闭mosaic然后按照sgd且学习率为1e-2跑一遍……我也不知道为什么

11wswhl commented 2 years ago

好的 我试试

11wswhl commented 2 years ago

up是真的牛啊 改成0.01之后效果就好了