def get_batch(self):
cur_from = self.cur
cur_to = min(cur_from + self.batch_size, self.size)
roidb = [self.roidb[self.index[i]] for i in range(cur_from, cur_to)]
if self.has_rpn:
data, label, im_info = get_rpn_testbatch(roidb)
else:
data, label, im_info = get_rcnn_testbatch(roidb)
self.data = [mx.nd.array(data[name]) for name in self.data_name]
self.im_info = im_info
def get_batch(self):
cur_from = self.cur
cur_to = min(cur_from + self.batch_size, self.size)
roidb = [self.roidb[self.index[i]] for i in range(cur_from, cur_to)]
if self.has_rpn:
data, label, im_info = get_rpn_testbatch(roidb)
else:
data, label, im_info = get_rcnn_testbatch(roidb)
self.data = [mx.nd.array(data[name]) for name in self.data_name]
self.im_info = im_info
def get_batch(self):
cur_from = self.cur
cur_to = min(cur_from + self.batch_size, self.size)
roidb = [self.roidb[self.index[i]] for i in range(cur_from, cur_to)]
if self.has_rpn:
if not self.withlabel:
data, label = get_rpn_testbatch(roidb)
else:
data, label = get_rpn_batch(roidb)
else:
data, label = get_rcnn_testbatch(roidb)
#print('in_loader', label['gt_boxes'].shape, file=sys.stderr)
self.im_info = data['im_info']
self.data = [mx.nd.array(data[name]) for name in self.data_name]