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]