Do not show HeatMap generation UI setting by default

De-clutter the UI

Author: semjon00

README.md

@@ -23,6 +23,10 @@ images generated by [@semjon00](https://github.com/semjon00) from CC0 photos, mo
 ## Changelog
 * v0.4.2
     * rembg update
+    * some default settings were changed - if the results do not match with what you expect, set settings to the old defaults
+    * heatmap generation UI option is hidden - if you want to use it, please activate it in the extension settings
+    * fixed regression - batch processing now saves images while the generation is happening
+    * other bugfixes
 * v0.4.1 standalone mode
     * ability to run DepthMap without WebUI (Use main.py. Make sure all the dependencies are installed. The support is not feature-complete.)
     * NormalMap generation

scripts/depthmap.py

@@ -71,26 +71,20 @@ def run(self, p, *inputs):
 
 
 # TODO: some of them may be put into the main ui pane
+# TODO: allow in standalone mode
 def on_ui_settings():
     section = ('depthmap-script', "Depthmap extension")
-    shared.opts.add_option("depthmap_script_keepmodels",
-                           shared.OptionInfo(False, "Do not unload depth and pix2pix models.",
-                                             section=section))
-    shared.opts.add_option("depthmap_script_boost_rmax",
-                           shared.OptionInfo(1600, "Maximum wholesize for boost (Rmax)",
-                                             section=section))
-    shared.opts.add_option("depthmap_script_save_ply",
-                           shared.OptionInfo(False, "Save additional PLY file with 3D inpainted mesh.",
-                                             section=section))
-    shared.opts.add_option("depthmap_script_show_3d",
-                           shared.OptionInfo(True, "Enable showing 3D Meshes in output tab. (Experimental)",
-                                             section=section))
-    shared.opts.add_option("depthmap_script_show_3d_inpaint",
-                           shared.OptionInfo(True, "Also show 3D Inpainted Mesh in 3D Mesh output tab. (Experimental)",
-                                             section=section))
-    shared.opts.add_option("depthmap_script_mesh_maxsize",
-                           shared.OptionInfo(2048, "Max size for generating simple mesh.",
-                                             section=section))
+
+    def add_option(name, default_value, description, name_prefix='depthmap_script'):
+        shared.opts.add_option(f"{name_prefix}_{name}", shared.OptionInfo(default_value, description, section=section))
+
+    add_option('keepmodels', False, "Do not unload depth and pix2pix models.")
+    add_option('boost_rmax', 1600, "Maximum wholesize for boost (Rmax)")
+    add_option('save_ply', False, "Save additional PLY file with 3D inpainted mesh.")
+    add_option('show_3d', True, "Enable showing 3D Meshes in output tab. (Experimental)")
+    add_option('show_3d_inpaint', True, "Also show 3D Inpainted Mesh in 3D Mesh output tab. (Experimental)")
+    add_option('mesh_maxsize', 2048, "Max size for generating simple mesh.")
+    add_option('gen_heatmap_from_ui', False, "Show an option to generate HeatMap in the UI")
 
 
 from modules import script_callbacks

src/common_ui.py

@@ -102,9 +102,10 @@ def main_ui_panel(is_depth_tab):
                 with gr.Row():
                     inp += go.NORMALMAP_INVERT, gr.Checkbox(label="Invert")
 
-        with gr.Box():
-            with gr.Row():
-                inp += go.GEN_HEATMAP, gr.Checkbox(label="Generate HeatMap")
+        if backbone.get_opt('depthmap_script_gen_heatmap_from_ui', False):
+            with gr.Box():
+                with gr.Row():
+                    inp += go.GEN_HEATMAP, gr.Checkbox(label="Generate HeatMap")
 
         with gr.Box():
             with gr.Column():