printing output

api/model/src/datasets/vitonhd.py

@@ -111,11 +111,17 @@ class VitonHDDataset(data.Dataset):
         #self.c_names = c_names
         #self.dataroot_names = dataroot_names
 
+        print(im_names)
+        print(c_names)
+        print(dataroot_names)
         # Filtra le immagini e i nomi dei vestiti in base a quelle presenti in captions_dict
         self.im_names = [im for im in im_names if im.split('_')[0] in self.captions_dict]
         self.c_names = [c_names[idx] for idx, im in enumerate(im_names) if im.split('_')[0] in self.captions_dict]
         self.dataroot_names = [dataroot_names[idx] for idx, im in enumerate(im_names) if im.split('_')[0] in self.captions_dict]
-
+        
+        print(im_names)
+        print(c_names)
+        print(dataroot_names)
 
     def __getitem__(self, index):
         """
@@ -129,6 +135,11 @@ class VitonHDDataset(data.Dataset):
         im_name = self.im_names[index]
         dataroot = self.dataroot_names[index]
 
+        
+        print(im_name)
+        print(c_name)
+        print(dataroot)
+
         sketch_threshold = random.randint(self.sketch_threshold_range[0], self.sketch_threshold_range[1])
 
         print("##############################", self.captions_dict)