preparing for frontend and backend communication

api/app.py

@@ -39,24 +39,6 @@ def current_endpoint():
 
 @app.route('/generate-design', methods=['GET'])
 def generate_design():
-    
-    '''# Esegui il comando di generazione
-    command = [
-        'python', '/api/model/src/eval.py',
-        '--dataset_path', '/api/model/assets/data/vitonhd',
-        '--batch_size', '1',
-        '--mixed_precision', 'fp16',
-        '--output_dir', output_dir,
-        '--save_name', 'generato_paired',
-        '--num_workers_test', '4',
-        '--sketch_cond_rate', '0.2',
-        '--dataset', 'vitonhd',
-        '--start_cond_rate', '0.0',
-        '--test_order', 'unpaired'
-    ]
-    
-    # Esegui il comando senza gestire il JSON, lascia che l'errore venga stampato
-    subprocess.run(command, check=True)'''
     try:
         # Creiamo una lista di argomenti come quelli che passeresti via CLI
         sys.argv = [
@@ -64,8 +46,6 @@ def generate_design():
             '--dataset_path', '/api/model/assets/data/vitonhd',
             '--batch_size', '1',
             '--mixed_precision', 'fp16',
-            #'--output_dir', '/api/output/generato_paired_paired/images',
-            '--save_name', 'generato_paired',
             '--num_workers_test', '4',
             '--sketch_cond_rate', '0.2',
             '--dataset', 'vitonhd',

api/model/src/datasets/vitonhd.py

@@ -28,7 +28,7 @@ class VitonHDDataset(data.Dataset):
             phase: str,
             tokenizer,
             radius=5,
-            caption_folder='captions.json',
+            caption_folder='captions.json', #######################################################3
             sketch_threshold_range: Tuple[int, int] = (20, 127),
             order: str = 'paired',
             outputlist: Tuple[str] = ('c_name', 'im_name', 'image', 'im_cloth', 'shape', 'pose_map',
@@ -153,7 +153,11 @@ class VitonHDDataset(data.Dataset):
                     os.path.join(dataroot, self.phase, 'im_sketch_unpaired',
                                  os.path.splitext(im_name)[0] + '_' + c_name.replace(".jpg", ".png")))
             elif self.order == 'paired':
-                im_sketch = Image.open(os.path.join(dataroot, self.phase, 'im_sketch', im_name.replace(".jpg", ".png")))
+                im_sketch = Image.open(os.path.join(dataroot, self.phase, 'im_sketch', im_name.replace(".jpg", ".jpeg")))
+                # define a transform to convert the image to grayscale
+                transform = transforms.Grayscale()
+                # apply the above transform on the image
+                im_sketch = transform(im_sketch)
             else:
                 raise ValueError(
                     f"Order should be either paired or unpaired"

api/model/src/utils/arg_parser.py

@@ -87,7 +87,7 @@ def eval_parse_args() -> argparse.Namespace:
 
     # miscelaneous parameters
     parser.add_argument("--seed", type=int, default=1234, help="A seed for reproducible training.")
-    parser.add_argument("--save_name", type=str, required=True, help="Folder name of the saved images")
+    parser.add_argument("--save_name", type=str, required=False, help="Folder name of the saved images")
 
     args = parser.parse_args()