bharatsingh430 / py-R-FCN-multiGPU

Code for training py-faster-rcnn and py-R-FCN on multiple GPUs in caffe

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

The error when using coco dataset

chengshuai opened this issue · comments

when using coco dataset for training, the error is that:

File "/home/aaa/soft-nms-master/tools/../lib/rpn/anchor_target_layer.py", line 138, in forward
argmax_overlaps = overlaps.argmax(axis=1)
ValueError: attempt to get argmax of an empty sequence

I find the error is because the ration of the image width and height is too large or too small.

i want to know the solution or how do you train?