data_layer.py 文件源码

def _get_image_blob(self, sample):
        im_blob = []
        labels_blob = []
        for i in range(self.batch_size):
        ImgAsColr = 0
        if cfg.CHANNEL_SIZE == 3:
           ImgAsColr = 1
            im = cv2.imread(cfg.IMAGEPATH + sample[i]['picname'],ImgAsColr)
            if sample[i]['flipped']:
                im = im[:, ::-1, :]
            personname = sample[i]['picname'].split('/')[0]
            labels_blob.append(self._data._sample_label[personname])
            im = prep_im_for_blob(im)

            im_blob.append(im)

        # Create a blob to hold the input images
        blob = im_list_to_blob(im_blob)
        return blob, labels_blob