cwq159/PyTorch-Spiking-YOLOv3

源码中没有coco.data啊,怎么解决?自己下载吗

DnkNju opened this issue · 4 comments

源码中没有coco.data啊,怎么解决?自己下载吗

I have used the following code from https://github.com/pjreddie/darknet/blob/master/cfg/coco.data

classes = 80
train = /PyTorch-Spiking-YOLOv3/data/coco/trainvalno5k.txt
valid = /PyTorch-Spiking-YOLOv3/data/coco/5k.txt
names = /PyTorch-Spiking-YOLOv3/data/coco.names
backup = /PyTorch-Spiking-YOLOv3/backup
eval = coco

Hope it helps.

换到eriklindernoren分支

我用自己的数据集跑ultralytics分支时,出现了RuntimeError: shape '[4, 3, 9, 16, 16]' is invalid for input of size 261120。换到eriklindernoren分支后,可以训练其他数据集吗?有什么办法可以解决吗? 谢谢