Started cloth segmentation debugging

2024-07-05 12:44:27 +05:30
parent a2c78015f4
commit abaf432b6f
1 changed files with 1 additions and 1 deletions
@@ -104,7 +104,7 @@ from PIL import Image
 def generate_mask(input_image, net, device='cpu'):
    img = input_image
    img_size = img.size
-    img = img.resize((768, 768), Image.BICUBIC)
+    # img = img.resize((768, 768), Image.BICUBIC)
    image_tensor = apply_transform(img)
    image_tensor = torch.unsqueeze(image_tensor, 0)