process mask

model.py

@@ -86,7 +86,9 @@ class Model:
     @staticmethod
     def process_mask(mask: np.ndarray) -> np.ndarray:
         if mask.shape != (512, 256, 3):
-            return None
+            mask = image.resize(
+                size=(256, 512),
+                resample=PIL.Image.NEAREST)
         seg_map = np.full(mask.shape[:-1], -1)
         for index, color in enumerate(COLOR_LIST):
             seg_map[np.sum(mask == color, axis=2) == 3] = index