search

cwq159 / PyTorch-Spiking-YOLOv3

A PyTorch implementation of Spiking-YOLOv3. Two branches are provided, based on two common PyTorch implementation of YOLOv3(ultralytics/yolov3 & eriklindernoren/PyTorch-YOLOv3), with support for Spiking-YOLOv3-Tiny at present.
GNU General Public License v3.0
207 stars 58 forks source link

VOC数据集训练精度不高的问题 #45

Open Agent-QG opened 2 years ago

Agent-QG commented 2 years ago

@cwq159 作者您好，我在尝试使用这个算法训练时，voc-yolov3-tiny-ours.cfg，300个epoch ，VOC2012数据集的map只有0.24，与之对比的是，使用voc-tinyv3-tiny.cfg相同epoch数与数据集，map为0.4。 我想问一下为什么VOC数据集训练出来的map这么低，很难达到您在read.me里提到的精度。 期待您的回复，谢谢。

WuTi0525 commented 1 year ago

请问您解决了这个问题了吗 我同样遇到了这样的问题

123456ZF commented 1 year ago

我ultralytics分支时，用自己的数据集，出现了RuntimeError: shape '[4, 3, 9, 16, 16]' is invalid for input of size 261120，filters我也改了，可是还是出现了这个问题，有什么解决办法吗，谢谢

123456ZF commented 1 year ago

@Agent-QG

suojinhui commented 1 year ago

学习率调小就行。