push

inference/cli_demo_camera_i2v_pcd.py

@@ -294,7 +294,7 @@ def generate_video(
             input_images = reference_video[0].unsqueeze(0)
         else:
             input_images = torch.tensor(np.array(Image.open(image))).permute(2,0,1).unsqueeze(0)/255
-            pixel_transforms = [transforms.Resize((480, 720)),
+            pixel_transforms = [transforms.Resize((height, width)),
                                 transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5], inplace=True)]
             for transform in pixel_transforms:
                 input_images = transform(input_images)