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

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

Open daiyuanm opened this issue 2 years ago • 4 comments

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

我在使用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的预训练模型呢?

daiyuanm avatar Sep 19 '21 15:09 daiyuanm

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

我在使用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训练的结果如何?我想请问你不使用任何预训练模型,从头开始训练,代码修改了哪些地方呢?

QQANG avatar Nov 01 '21 11:11 QQANG

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

gll-sketch avatar Jan 19 '22 11:01 gll-sketch

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

huiling-liao avatar Feb 18 '22 07:02 huiling-liao

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

daiyuanm avatar Mar 02 '22 07:03 daiyuanm