|
@@ -82,13 +82,13 @@ class Unet2ClothSession(BaseSession):
|
|
mask1.putpalette(palette1)
|
|
mask1.putpalette(palette1)
|
|
mask1 = mask1.convert("RGB").convert("L")
|
|
mask1 = mask1.convert("RGB").convert("L")
|
|
masks.append(mask1)
|
|
masks.append(mask1)
|
|
-
|
|
|
|
|
|
+
|
|
def lower_cloth():
|
|
def lower_cloth():
|
|
mask2 = mask.copy()
|
|
mask2 = mask.copy()
|
|
mask2.putpalette(palette2)
|
|
mask2.putpalette(palette2)
|
|
mask2 = mask2.convert("RGB").convert("L")
|
|
mask2 = mask2.convert("RGB").convert("L")
|
|
masks.append(mask2)
|
|
masks.append(mask2)
|
|
-
|
|
|
|
|
|
+
|
|
def full_cloth():
|
|
def full_cloth():
|
|
mask3 = mask.copy()
|
|
mask3 = mask.copy()
|
|
mask3.putpalette(palette3)
|
|
mask3.putpalette(palette3)
|