'YOLOLayer' object has no attribute 'grid'

[spurslipu]

'YOLOLayer' object has no attribute 'grid'

[SpursLipu]

你用的训练指令是什么？

[seek-ing]

@SpursLipu
谢谢回复
python3 train.py --data data/obj.data -pt --batch-size 16 --weights weights/yolov3.pt --cfg cfg/yolov3.cfg

[SpursLipu]

我在训练的过程中没有遇到过这个问题，你可以检查一下你是用数据集的分类数是否与yolov3.cfg中的分类数相等(80类)，-pt是使用在coco上进行预训练的.weights文件时使用的命令，不知道是不是和你的指令有冲突。

另外这个仓库中的detect.py程序是使用原图的分辨率进行检测的，你可以测试一下效果。

[seek-ing]

@SpursLipu
谢谢～
我的是7分类，我把 -pt去除也遇到了同样的问题
你师兄是缩放成256x416的？

[SpursLipu]

你用multi-scale训练暂时就没问题了，我在排查你说的问题。

缩放都是使用原分辨率除以输入图片长边算出的缩放倍数进行缩放的，检测时的缩放是没有填充。

[SpursLipu]

应该是解决了，你可以再试一下

[seek-ing]

@SpursLipu 非常感谢～

[Fly-dream12]

When i try to prune the model, I also encounter this error " YOLOLayer object has no attribute 'grid'". My comand is:
python train.py --data --batch-size --weights big_model/best.pt --cfg --sr --s 0.001 --prune 0 --single_cls
Can you give some suggestions? @SpursLipu

[SpursLipu]

I have already solve this problem, you can download new version of this repository.