search

liulei01 / DRBox

A deep learning based algorithm to detect rotated object, for example, objects in remote sensing images
Other
421 stars 141 forks source link

IndexError: index 1 is out of bounds for axis 0 with size 1 #46

Open Viktor-Paul opened 4 years ago

Viktor-Paul commented 4 years ago

I wanted to try batch_size 1 rather than 64 because I have only one image at a time and with 64 it takes much longer. However, when I change batch size to 1 both in the deploy.py and deploy.prototxt, I see the following error:

inputdata[count] = subimage IndexError: index 1 is out of bounds for axis 0 with size 1