Faster-RCNN_TF icon indicating copy to clipboard operation
Faster-RCNN_TF copied to clipboard

How many images used for each batch?

Open wmmxk opened this issue 7 years ago • 1 comments

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?

wmmxk avatar May 18 '17 00:05 wmmxk

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.

eraofelix avatar Nov 15 '18 07:11 eraofelix