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.

Sign up for GitHub

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

Jump to bottom

train_loss下降，val_loss不变，map一直不变，自建数据集预测效果很差。 #363

Open
Chris-zixuan opened this issue Apr 8, 2023 · 9 comments
Open

train_loss下降，val_loss不变，map一直不变，自建数据集预测效果很差。 #363

Chris-zixuan opened this issue Apr 8, 2023 · 9 comments

Comments

@Chris-zixuan
Copy link

epoch_map
epoch_loss

自建的数据集，使用fasterrcnn是可以正常训练和预测的。只有yolo系列不可以。
我查了一下说是可能因为过拟合？ 也试过调参之类的手段，没什么效果。
YOLOV5效果也只能说勉强可以，迟迟找不到原因
@Chris-zixuan
Copy link
Author

会不会是我自建数据集图片大小的问题呢？ 我用的是224*224的图，train脚本和yolo脚本内也修改了对应的参数呀。

@PengPeng-JunJun
Copy link

预训练模型是用的416*416的图像来抽取图像特征的，不过也有可能是你说的这个原因。你的数据集大小是多少呢？ 我之前也遇到过类似你这样的问题，是优化器的选择。如果数据量不大，用sgd很难收敛，我300个E都没有收敛，改成adma后，80个E左右，MAP就有0.8了

@PengPeng-JunJun
Copy link

建议从头到尾仔细看一下UP主写的注释，很详细，会让使用者避坑

@bubbliiiing
Copy link
Owner

额，adam会收敛快点。
224也确实不合适。
不过和具体检测的目标也有关的

@Chris-zixuan
Copy link
Author

预训练模型是用的416*416的图像来抽取图像特征的，不过也有可能是你说的这个原因。你的数据集大小是多少呢？ 我之前也遇到过类似你这样的问题，是优化器的选择。如果数据量不大，用sgd很难收敛，我300个E都没有收敛，改成adma后，80个E左右，MAP就有0.8了

我确实没试过修改优化器。。。但是代码我是吃透了。看了很多遍

@Chris-zixuan
Copy link
Author

额，adam会收敛快点。 224也确实不合适。 不过和具体检测的目标也有关的

我试试adam

@s5377689
Copy link

s5377689 commented May 8, 2023 •
edited
Loading

我發現只要改了東西訓練MAP都會變超級低，不知道要怎麼挑整參數

@oldmonsterlee
Copy link

你好，请问你的这个问题解决了吗？

@MingAze
Copy link

MingAze commented Jan 8, 2024 via email

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
6 participants
@bubbliiiing @MingAze @PengPeng-JunJun @Chris-zixuan @s5377689 @oldmonsterlee