Faster-RCNN_TF
Faster-RCNN_TF copied to clipboard
How many images used for each batch?
I was trying to understand the get_minibatch function in /lib/roi_data_layer/minibatch.py. The assert statement in the function always spit an error (assert len(im_scales) == 1, "Single batch only";assert len(roidb) == 1, "Single batch only").
I traced back to /lib/roi_data_layer/layer.py, and found the size of roidb is 2, which is defined by cfg.TRAIN.IMS_PER_BATCH (minibatch_db is assigned to roidb in a RoIDatalayer object and the size of minibatch is determined by cfg.TRAIN.IMS_PER_BATCH) . If so, the condition len(roidb)==1 is always false. Does anyone run into this?
I was trying to understand the get_minibatch function in /lib/roi_data_layer/minibatch.py. The assert statement in the function always spit an error (assert len(im_scales) == 1, "Single batch only";assert len(roidb) == 1, "Single batch only").
I traced back to /lib/roi_data_layer/layer.py, and found the size of roidb is 2, which is defined by cfg.TRAIN.IMS_PER_BATCH (minibatch_db is assigned to roidb in a RoIDatalayer object and the size of minibatch is determined by cfg.TRAIN.IMS_PER_BATCH) . If so, the condition len(roidb)==1 is always false. Does anyone run into this?
the config in experiments/cfgs/faster_rcnn_end2end.yml will overwrite them.