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

使用CoordAttention-YOLOv4作为backbone的问题。 #227

Open
daiyuanm opened this issue Sep 19, 2021 · 4 comments
Open

使用CoordAttention-YOLOv4作为backbone的问题。 #227

daiyuanm opened this issue Sep 19, 2021 · 4 comments

Comments

@daiyuanm
Copy link

你好,非常感谢作者的工作!

我在使用CoordAttention-YOLOv4作为backbone开始训练时,使用作者提供的mobilenetv2.pth作为预训练模型,会出现预训练模型与当前模型不匹配的情况。
于是我没有使用任何预训练模型,直接在我的数据集(大约1k张图片)上从头开始训练,目前训练到270个epoch,得到的map很低(不到0.2)。之前使用预训练模型以及使用其他backbone时,使用同样的参数(使用CoordAttention-YOLOv4作为backbone时,只修改了eval的image_size:640→320,640会使得任务被kill),到270epoch时,效果就会很不错了。

我想请问:
(1)不使用预训练模型,直接从头开始训练,会对最终得到的map会有影响吗?我目前设置的epoch 是500,等我跑完,我编辑更新一下结果。
(2)作者是否可以提供使用CoordAttention-YOLOv4作为backbone的预训练模型呢?

@QQANG
Copy link

QQANG commented Nov 1, 2021

你好,非常感谢作者的工作!

我在使用CoordAttention-YOLOv4作为backbone开始训练时,使用作者提供的mobilenetv2.pth作为预训练模型,会出现预训练模型与当前模型不匹配的情况。 于是我没有使用任何预训练模型,直接在我的数据集(大约1k张图片)上从头开始训练,目前训练到270个epoch,得到的map很低(不到0.2)。之前使用预训练模型以及使用其他backbone时,使用同样的参数(使用CoordAttention-YOLOv4作为backbone时,只修改了eval的image_size:640→320,640会使得任务被kill),到270epoch时,效果就会很不错了。

我想请问: (1)不使用预训练模型,直接从头开始训练,会对最终得到的map会有影响吗?我目前设置的epoch 是500,等我跑完,我编辑更新一下结果。 (2)作者是否可以提供使用CoordAttention-YOLOv4作为backbone的预训练模型呢?

你好,你使用CoordAttention-YOLOv4作为backbone训练的结果如何?我想请问你不使用任何预训练模型,从头开始训练,代码修改了哪些地方呢?

@gll-sketch
Copy link

您好,请问你有coordAttention的预训练权重了吗?

1 similar comment
@huiling-liao
Copy link

您好,请问你有coordAttention的预训练权重了吗?

@daiyuanm
Copy link
Author

daiyuanm commented Mar 2, 2022

你好,我后面没有继续做这个工作了,许多东西也没有保存,抱歉。预训练权重我一直没有

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

4 participants