Segment upper clothes only

process.py

@@ -124,9 +124,9 @@ def generate_mask(input_image, net, palette, device = 'cpu'):
     classes_to_save = []
 
     # Check which classes are present in the image
-    for cls in range(1, 4):  # Exclude background class (0)
-        if np.any(output_arr == cls):
-            classes_to_save.append(cls)
+    #for cls in range(1, 4):  # Exclude background class (0)
+    if np.any(output_arr == 1):
+        classes_to_save.append(1)
 
     # Save alpha masks
     for cls in classes_to_save: