Added Delete Visualizations button to the training pipeline gui, expo…

…sed del_viz_predictions config option to the user
talmolab · Jun 23, 2024 · 22733f1 · 22733f1
1 parent a9820ca
commit 22733f1
Show file tree

Hide file tree

Showing 22 changed files with 22 additions and 1 deletion.
diff --git a/sleap/training_profiles/baseline.centroid.json b/sleap/training_profiles/baseline.centroid.json
@@ -116,6 +116,7 @@
         "runs_folder": "models",
         "tags": [],
         "view_visualizations": true,
+        "delete_viz_images": true,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,

diff --git a/sleap/training_profiles/baseline_large_rf.bottomup.json b/sleap/training_profiles/baseline_large_rf.bottomup.json
@@ -125,6 +125,7 @@
         "runs_folder": "models",
         "tags": [],
         "view_visualizations": true,
+        "delete_viz_images": true,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,

diff --git a/sleap/training_profiles/baseline_large_rf.single.json b/sleap/training_profiles/baseline_large_rf.single.json
@@ -116,6 +116,7 @@
         "runs_folder": "models",
         "tags": [],
         "view_visualizations": true,
+        "delete_viz_images": true,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,

diff --git a/sleap/training_profiles/baseline_large_rf.topdown.json b/sleap/training_profiles/baseline_large_rf.topdown.json
@@ -117,6 +117,7 @@
         "runs_folder": "models",
         "tags": [],
         "view_visualizations": true,
+        "delete_viz_images": true,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,

diff --git a/sleap/training_profiles/baseline_medium_rf.bottomup.json b/sleap/training_profiles/baseline_medium_rf.bottomup.json
@@ -125,6 +125,7 @@
         "runs_folder": "models",
         "tags": [],
         "view_visualizations": true,
+        "delete_viz_images": true,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,

diff --git a/sleap/training_profiles/baseline_medium_rf.single.json b/sleap/training_profiles/baseline_medium_rf.single.json
@@ -116,6 +116,7 @@
         "runs_folder": "models",
         "tags": [],
         "view_visualizations": true,
+        "delete_viz_images": true,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,

diff --git a/sleap/training_profiles/baseline_medium_rf.topdown.json b/sleap/training_profiles/baseline_medium_rf.topdown.json
@@ -117,6 +117,7 @@
         "runs_folder": "models",
         "tags": [],
         "view_visualizations": true,
+        "delete_viz_images": true,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,

diff --git a/sleap/training_profiles/pretrained.bottomup.json b/sleap/training_profiles/pretrained.bottomup.json
@@ -122,6 +122,7 @@
         "runs_folder": "models",
         "tags": [],
         "view_visualizations": true,
+        "delete_viz_images": true,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,

diff --git a/sleap/training_profiles/pretrained.centroid.json b/sleap/training_profiles/pretrained.centroid.json
@@ -113,6 +113,7 @@
         "runs_folder": "models",
         "tags": [],
         "view_visualizations": true,
+        "delete_viz_images": true,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,

diff --git a/sleap/training_profiles/pretrained.single.json b/sleap/training_profiles/pretrained.single.json
@@ -113,6 +113,7 @@
         "runs_folder": "models",
         "tags": [],
         "view_visualizations": true,
+        "delete_viz_images": true,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,

diff --git a/sleap/training_profiles/pretrained.topdown.json b/sleap/training_profiles/pretrained.topdown.json
@@ -114,6 +114,7 @@
         "runs_folder": "models",
         "tags": [],
         "view_visualizations": true,
+        "delete_viz_images": true,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,

diff --git a/tests/data/models/min_tracks_2node.UNet.bottomup_multiclass/initial_config.json b/tests/data/models/min_tracks_2node.UNet.bottomup_multiclass/initial_config.json
@@ -128,6 +128,7 @@
         "runs_folder": "models",
         "tags": [],
         "view_visualizations": false,
+        "delete_viz_images": false,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,

diff --git a/tests/data/models/min_tracks_2node.UNet.bottomup_multiclass/training_config.json b/tests/data/models/min_tracks_2node.UNet.bottomup_multiclass/training_config.json
@@ -191,6 +191,7 @@
         "runs_folder": "models",
         "tags": [],
         "view_visualizations": false,
+        "delete_viz_images": false,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,

diff --git a/tests/data/models/min_tracks_2node.UNet.topdown_multiclass/initial_config.json b/tests/data/models/min_tracks_2node.UNet.topdown_multiclass/initial_config.json
@@ -141,7 +141,7 @@
         "runs_folder": "models",
         "tags": [],
         "view_visualizations": false,
-        "delete_viz_images": true,
+        "delete_viz_images": false,
         "zip_outputs": false,
         "log_to_csv": true,
         "checkpointing": {

diff --git a/tests/data/models/minimal_instance.UNet.bottomup/initial_config.json b/tests/data/models/minimal_instance.UNet.bottomup/initial_config.json
@@ -127,6 +127,7 @@
         "runs_folder": "models",
         "tags": [],
         "view_visualizations": false,
+        "delete_viz_images": false,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,

diff --git a/tests/data/models/minimal_instance.UNet.bottomup/training_config.json b/tests/data/models/minimal_instance.UNet.bottomup/training_config.json
@@ -192,6 +192,7 @@
         "runs_folder": "models",
         "tags": [],
         "view_visualizations": false,
+        "delete_viz_images": false,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,

diff --git a/tests/data/models/minimal_instance.UNet.centered_instance/initial_config.json b/tests/data/models/minimal_instance.UNet.centered_instance/initial_config.json
@@ -119,6 +119,7 @@
         "runs_folder": "models",
         "tags": [],
         "view_visualizations": false,
+        "delete_viz_images": false,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,

diff --git a/tests/data/models/minimal_instance.UNet.centered_instance/training_config.json b/tests/data/models/minimal_instance.UNet.centered_instance/training_config.json
@@ -179,6 +179,7 @@
         "runs_folder": "models",
         "tags": [],
         "view_visualizations": false,
+        "delete_viz_images": false,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,

diff --git a/tests/data/models/minimal_instance.UNet.centroid/initial_config.json b/tests/data/models/minimal_instance.UNet.centroid/initial_config.json
@@ -118,6 +118,7 @@
         "runs_folder": "models",
         "tags": [],
         "view_visualizations": false,
+        "delete_viz_images": false,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,

diff --git a/tests/data/models/minimal_instance.UNet.centroid/training_config.json b/tests/data/models/minimal_instance.UNet.centroid/training_config.json
@@ -175,6 +175,7 @@
         "runs_folder": "models",
         "tags": [],
         "view_visualizations": false,
+        "delete_viz_images": false,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,

diff --git a/tests/data/models/minimal_robot.UNet.single_instance/initial_config.json b/tests/data/models/minimal_robot.UNet.single_instance/initial_config.json
@@ -120,6 +120,7 @@
             ""
         ],
         "view_visualizations": false,
+        "delete_viz_images": false,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,

diff --git a/tests/data/models/minimal_robot.UNet.single_instance/training_config.json b/tests/data/models/minimal_robot.UNet.single_instance/training_config.json
@@ -180,6 +180,7 @@
             ""
         ],
         "view_visualizations": false,
+        "delete_viz_images": false,
         "log_to_csv": true,
         "checkpointing": {
             "initial_model": false,