cloth-segmentation

Running

mevCJ commited on Feb 29, 2024

Commit

d374ff2

verified ·

1 Parent(s): f62d400

Update process.py

Files changed (1) hide show

process.py CHANGED Viewed

@@ -172,7 +172,7 @@ def main(args):
     # Create an instance of your model
     model = load_seg_model(args.checkpoint_path, device=device)
-    palette = get_palette(4)
     img = Image.open(args.image).convert('RGB')

     # Create an instance of your model
     model = load_seg_model(args.checkpoint_path, device=device)
+    palette = get_palette(1)
     img = Image.open(args.image).convert('RGB')