Hugging Face's logo

Spaces:
CVPR
/
Text2Human
Runtime error

App Files Community
2
yumingj commited on
Commit
cd1147d
·
1 Parent(s): c5bdcdb

process mask

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -87,7 +87,7 @@ class Model:
87
  def process_mask(mask: np.ndarray) -> np.ndarray:
88
  if mask.shape != (512, 256, 3):
89
  mask = mask.resize(
90
- size=(256, 512),
91
  resample=PIL.Image.NEAREST)
92
  seg_map = np.full(mask.shape[:-1], -1)
93
  for index, color in enumerate(COLOR_LIST):
 
87
  def process_mask(mask: np.ndarray) -> np.ndarray:
88
  if mask.shape != (512, 256, 3):
89
  mask = mask.resize(
90
+ size=(512, 256),
91
  resample=PIL.Image.NEAREST)
92
  seg_map = np.full(mask.shape[:-1], -1)
93
  for index, color in enumerate(COLOR_LIST):