latest.zip

anydoor/run_inference_train_x.py

@@ -256,7 +256,7 @@ if __name__ == '__main__':
         os.makedirs(save_dir)
 
     # Create list of cloth image names
-    cloth_image_names = os.listdir(cloth_dir)[:5]
+    cloth_image_names = os.listdir(cloth_dir)[:40]
 
     # First loop for the top half
     for cloth_image_name in cloth_image_names:
@@ -288,13 +288,13 @@ if __name__ == '__main__':
 
 
     # Update the cloth directory and mask path for the bottom half processing
-    cloth_dir = '/work/pink_girl/cloth/trou'
+    cloth_dir1 = '/work/pink_girl/cloth/trou'
     fixed_ref_mask_path = os.path.join(image_parse_v3_dir, 'trou1.png')
-    cloth_image_names = os.listdir(cloth_dir)[:5]
+    cloth_image_names = os.listdir(cloth_dir1)[:40]
 
     # Second loop for the bottom half
     for cloth_image_name in cloth_image_names:
-        cloth_image_path = os.path.join(cloth_dir, cloth_image_name)
+        cloth_image_path = os.path.join(cloth_dir1, cloth_image_name)
         cloth_mask_path = os.path.join(cloth_mask_dir, cloth_image_name)
         
         cloth_image = cv2.imread(cloth_image_path)
@@ -304,7 +304,7 @@ if __name__ == '__main__':
         ref_image = cv2.imread(fixed_ref_image_path)
         ref_image = cv2.cvtColor(ref_image, cv2.COLOR_BGR2RGB)
         ref_mask = Image.open(fixed_ref_mask_path).convert('P')
-        ref_mask = np.array(ref_mask) == 2  
+        ref_mask = np.array(ref_mask) == 1  
 
         gen_image = inference_single_image(cloth_image, cloth_mask, ref_image, ref_mask)
         gen_path = os.path.join(save_dir, 'gen_pant_' + cloth_image_name)
@@ -318,5 +318,5 @@ if __name__ == '__main__':
 
         # Replace the resizing and padding code for the bottom half
         bottom_image_path = os.path.join(save_dir, 'bottom_z_' + cloth_image_name)
-        bottom_half_cropped = resize_and_crop(gen_image, 512, 600, False, True)
+        bottom_half_cropped = resize_and_crop(gen_image, 512, 700, False, True)
         cv2.imwrite(bottom_image_path, cv2.cvtColor(bottom_half_cropped, cv2.COLOR_RGB2BGR))

adbase-step=48745.ckpt → latest.zip

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ed60705f20bdde1951e9dd280eacc1849a16bc5e6aff277aa52e5cc6f577c6a8
-size 16836620107
+oid sha256:e2dff0561936fc60ed86ab1c4da9a2654a43e98228c15d84fe189d2bfd3952ce
+size 109296642