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

Why batch_size = 1 throws an error? #17

Closed arasharchor closed 6 years ago

arasharchor commented 6 years ago

Hi, 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

arasharchor commented 6 years ago

thanks I figured it out.

keck91 commented 6 years ago

Could you share your insights?

Viktor-Paul commented 4 years ago

thanks I figured it out.

Could you share your insights?

Viktor-Paul commented 4 years ago

Could you share your insights?

Did you finally solve this problem?