wakanawakana / YOLOv2

YOLOv2にGRAD-CAMを追加します(Grad-cam applied to YOLOv2)
MIT License
3 stars 1 forks source link

problem with shapes when running gradcam in yolo #1

Closed adelshafiei closed 4 years ago

adelshafiei commented 5 years ago

I am trying your code and getting this error:

python3 yolov2_darknet_parser.py yolo.weights yolov2.conv21.W.data = dat[38718432:67029984].reshape(1024, 3072, 3, 3) Traceback (most recent call last): File "yolov2_darknet_parser.py", line 91, in exec(txt) File "", line 1, in ValueError: cannot reshape array of size 12265129 into shape (1024,3072,3,3)

also download_images.py cannot download animal pics.

wakanawakana commented 5 years ago

Is chainer 1.16 & python2.7? It is unknown if the binary works on other versions as the binary structure of the model changes often.

wakanawakana commented 5 years ago

Use 'yolov2_gcma.py' if you can read weight files