def __getitem__(self, idx): if self.videoFramesPath[idx].endswith(".webp"): frame = webp.load_image(self.videoFramesPath[idx], "RGB") else: frame = Image.open(self.videoFramesPath[idx]).convert('RGB') frame = self.transform(frame) return frame
def test_image_palette_opaque(self, image_bars_palette_opaque): with TemporaryDirectory() as tmpdir: file_name = os.path.join(tmpdir, 'image.webp') assert image_bars_palette_opaque.mode == 'P' webp.save_image(image_bars_palette_opaque, file_name, lossless=True) dec_img = webp.load_image(file_name, 'RGB') actual = np.asarray(dec_img, dtype=np.uint8) image_bars_rgb = image_bars_palette_opaque.convert('RGB') expected = np.asarray(image_bars_rgb, dtype=np.uint8) assert_array_equal(actual, expected)
def __getitem__(self, idx): if self.videoFramesPath[idx].endswith(".webp"): frame = webp.load_image(self.videoFramesPath[idx], "RGB") else: frame = Image.open(self.videoFramesPath[idx]).convert('RGB') frame = self.transform(frame) flow = flowlib.readFlow( os.path.join(self.flow_dir, "{:06d}.flo".format(idx))) flow = transforms.ToTensor()(flow) occ = self.occTransform( Image.open( os.path.join(self.occlusion_dir, "{:05d}.png".format(idx))).convert('RGB')) return (frame, flow, occ)
def test_image_simple(self): width = 256 height = 64 img = Image.new('RGB', (width, height)) draw = ImageDraw.Draw(img) draw.rectangle([0, 0, width-1, height-1], fill=(0, 0, 255)) draw.rectangle([0, 0, (width/4-1), height-1], fill=(255, 0, 0)) with TemporaryDirectory() as tmpdir: file_name = os.path.join(tmpdir, 'image.webp') webp.save_image(img, file_name, lossless=True) dec_img = webp.load_image(file_name, 'RGB') actual = np.asarray(dec_img, dtype=np.uint8) expected = np.asarray(img, dtype=np.uint8) assert_array_equal(actual, expected)
source_video.start, source_video.start + args.duration) source_audio = source_video.audio source_fps = source_video.fps source_duration = source_video.duration print("Video Info:") print("Name:", driving_video_path) print("Dimensions:", 'x'.join([str(x) for x in source_video.size])) print("Start:", args.start or source_video.start) print("Duration:", source_duration) print("End:", (args.start or source_video.start) + source_duration) print("FPS:", source_fps) print() # Read Image if image_extension.lower() == 'webp': source_image = np.array(webp.load_image(source_image_path, 'RGBA')) else: source_image = imageio.imread(source_image_path) print("Image Info:") print("Name:", source_image_path) print("Dimensions:", 'x'.join([str(x) for x in source_image.shape[:2]][::-1])) print() # Read and Resize image and video to 256x256 if source_image.shape[:2] != [256, 256]: print("Resizing image to 256x256 with ", end='') if args.image_resize == 'fill': print("Mode: Fill") height, width = source_image.shape[:2]
def load_webp(image): image = webp.load_image(image, 'RGB') return image