Spaces:

CVPR
/

Text2Human

Runtime error

yumingj commited on Jul 11, 2022

Commit

c5bdcdb

1 Parent(s): 2e9f191

process mask

Files changed (1) hide show

model.py CHANGED Viewed

@@ -86,7 +86,7 @@ class Model:
     @staticmethod
     def process_mask(mask: np.ndarray) -> np.ndarray:
         if mask.shape != (512, 256, 3):
-            mask = image.resize(
                 size=(256, 512),
                 resample=PIL.Image.NEAREST)
         seg_map = np.full(mask.shape[:-1], -1)

     @staticmethod
     def process_mask(mask: np.ndarray) -> np.ndarray:
         if mask.shape != (512, 256, 3):
+            mask = mask.resize(
                 size=(256, 512),
                 resample=PIL.Image.NEAREST)
         seg_map = np.full(mask.shape[:-1], -1)