test_data_handler.py 文件源码

def next(self):
        sz = self.target_size
        output = np.ones([1, sz, sz, 1]).astype(np.float32)
        img = scipy.misc.imread(
            self._image_paths[self._index], mode='L').astype(np.float32)
        original_size = img.shape
        bigger_size = max(original_size[0], original_size[1])

        mult = 1
        if bigger_size > self.target_size:
          mult = self.target_size / float(bigger_size)



        resized_size = (int(original_size[0] * mult), int(original_size[1]*mult))
        img = scipy.misc.imresize(img, resized_size)
        img = (img - 128.0) / 128.0
        output[0, 0:resized_size[0], 0:resized_size[1], 0] = img

        self._index += 1

        return output, original_size, resized_size