cv2.py 文件源码

def create_test_data(self):
        im = cv2.imread(root('resources', 'color_circle.png'),
                        flags=cv2.IMREAD_UNCHANGED)
        if self.mode == 'RGB':
            im = im[:, :, :3]
        elif self.mode == 'RGBA':
            pass
        elif self.mode == 'L':
            im = cv2.cvtColor(im, cv2.COLOR_BGR2GRAY)
        else:
            raise ValueError('Unknown mode: {}'.format(self.mode))

        # Fine for upscaling
        im = cv2.resize(im, tuple(self.size), interpolation=cv2.INTER_CUBIC)
        return [im]