add resize human masks option

supervisely-ecosystem · Oct 25, 2024 · 0bc2b26 · 0bc2b26
1 parent 9c53153
commit 0bc2b26
Show file tree

Hide file tree

Showing 4 changed files with 25 additions and 1 deletion.
diff --git a/config.json b/config.json
@@ -13,6 +13,8 @@
   "modal_template": "src/modal.html",
   "modal_template_state": {
     "humanMasks": true,
+    "resizeHumanMasks": false,
+    "resizePercentze": 100,
     "machineMasks": true,
     "instanceMasks": false,
     "thickness": 2

diff --git a/src/globals.py b/src/globals.py
@@ -27,6 +27,8 @@
 
 Image.MAX_IMAGE_PIXELS = None
 HUMAN_MASKS = bool(strtobool(os.environ["modal.state.humanMasks"]))
+RESIZE_HUMAN_MASKS = bool(strtobool(os.environ["modal.state.resizeHumanMasks"]))
+RESIZE_PERCENT = int(os.environ["modal.state.resizePercent"])
 MACHINE_MASKS = bool(strtobool(os.environ["modal.state.machineMasks"]))
 INSTANCE_MASKS = bool(strtobool(os.environ["modal.state.instanceMasks"]))
 THICKNESS = int(os.environ["modal.state.thickness"])

diff --git a/src/main.py b/src/main.py
@@ -84,9 +84,19 @@ def export_as_masks(api: sly.Api):
                             (overlay.astype(np.uint16) + temp_overlay.astype(np.uint16)) / 2
                         ).astype(np.uint8)
 
+                    array_to_write = np.concatenate([raw_img, overlay], axis=1)
+
+                    if g.RESIZE_HUMAN_MASKS:
+                        new_dimensions = (
+                            int(overlay.shape[0] * (g.RESIZE_PERCENT / 100)),
+                            int(overlay.shape[1] * (g.RESIZE_PERCENT / 100)),
+                            overlay.shape[2],
+                        )
+                        array_to_write = array_to_write.reshape(new_dimensions)
+
                     sly.image.write(
                         os.path.join(human_masks_dir, mask_img_name),
-                        np.concatenate([raw_img, overlay], axis=1),
+                        array_to_write,
                     )
 
                 if g.MACHINE_MASKS:

diff --git a/src/modal.html b/src/modal.html
@@ -7,6 +7,16 @@
                description="Semantic segmentation RGB masks with colors of classes. (0, 0, 0) is for background (unlabeled area)">
         <el-checkbox v-model="state.humanMasks">enable</el-checkbox>
     </sly-field>
+    <sly-field v-if="state.humanMasks"
+               title="Resize Human masks"
+               description="Whether to resize human masks or not. Input the number ">
+        <el-checkbox v-model="state.resizeHumanMasks">enable</el-checkbox>
+    </sly-field>
+    <sly-field v-if="state.humanMasks && state.resizeHumanMasks"
+               title="Resize percentage"
+               description="The percentage to resize human masks by">
+        <el-input-number v-model="state.resizePercent" :min="1" :max="100"></el-input-number>
+    </sly-field>
     <sly-field title="Machine masks"
                description="Additional RGB masks for NN where colors are the indices of classes">
         <el-checkbox v-model="state.machineMasks">enable</el-checkbox>