Saad0KH commited on
Commit
fc3d381
Β·
verified Β·
1 Parent(s): 95ddb06

Update SegCloth.py

Browse files
Files changed (1) hide show
  1. SegCloth.py +1 -1
SegCloth.py CHANGED
@@ -20,7 +20,7 @@ def encode_image_to_base64(image):
20
  image.save(buffered, format="PNG")
21
  return base64.b64encode(buffered.getvalue()).decode('utf-8')
22
 
23
- def segment_and_enhance_clothing(img, clothes=["Hat", "Upper-clothes", "Skirt", "Pants", "Dress", "Belt", "Left-shoe", "Right-shoe", "Scarf"]):
24
  # Segment image
25
  segments = segmenter(img)
26
 
 
20
  image.save(buffered, format="PNG")
21
  return base64.b64encode(buffered.getvalue()).decode('utf-8')
22
 
23
+ def segment_clothing(img, clothes=["Hat", "Upper-clothes", "Skirt", "Pants", "Dress", "Belt", "Left-shoe", "Right-shoe", "Scarf"]):
24
  # Segment image
25
  segments = segmenter(img)
26