argusswift / YOLOv4-pytorch

This is a pytorch repository of YOLOv4, attentive YOLOv4 and mobilenet YOLOv4 with PASCAL VOC and COCO
1.68k stars 330 forks source link

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

Open daiyuanm opened 3 years ago

daiyuanm commented 3 years ago

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

我在使用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 commented 3 years ago

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

我在使用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 commented 2 years ago

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

huiling-liao commented 2 years ago

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

daiyuanm commented 2 years ago

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