Skip to content

Commit 22733f1

Browse files
committed
Added Delete Visualizations button to the training pipeline gui, exposed del_viz_predictions config option to the user
1 parent a9820ca commit 22733f1

22 files changed

+22
-1
lines changed

sleap/training_profiles/baseline.centroid.json

+1
Original file line numberDiff line numberDiff line change
@@ -116,6 +116,7 @@
116116
"runs_folder": "models",
117117
"tags": [],
118118
"view_visualizations": true,
119+
"delete_viz_images": true,
119120
"log_to_csv": true,
120121
"checkpointing": {
121122
"initial_model": false,

sleap/training_profiles/baseline_large_rf.bottomup.json

+1
Original file line numberDiff line numberDiff line change
@@ -125,6 +125,7 @@
125125
"runs_folder": "models",
126126
"tags": [],
127127
"view_visualizations": true,
128+
"delete_viz_images": true,
128129
"log_to_csv": true,
129130
"checkpointing": {
130131
"initial_model": false,

sleap/training_profiles/baseline_large_rf.single.json

+1
Original file line numberDiff line numberDiff line change
@@ -116,6 +116,7 @@
116116
"runs_folder": "models",
117117
"tags": [],
118118
"view_visualizations": true,
119+
"delete_viz_images": true,
119120
"log_to_csv": true,
120121
"checkpointing": {
121122
"initial_model": false,

sleap/training_profiles/baseline_large_rf.topdown.json

+1
Original file line numberDiff line numberDiff line change
@@ -117,6 +117,7 @@
117117
"runs_folder": "models",
118118
"tags": [],
119119
"view_visualizations": true,
120+
"delete_viz_images": true,
120121
"log_to_csv": true,
121122
"checkpointing": {
122123
"initial_model": false,

sleap/training_profiles/baseline_medium_rf.bottomup.json

+1
Original file line numberDiff line numberDiff line change
@@ -125,6 +125,7 @@
125125
"runs_folder": "models",
126126
"tags": [],
127127
"view_visualizations": true,
128+
"delete_viz_images": true,
128129
"log_to_csv": true,
129130
"checkpointing": {
130131
"initial_model": false,

sleap/training_profiles/baseline_medium_rf.single.json

+1
Original file line numberDiff line numberDiff line change
@@ -116,6 +116,7 @@
116116
"runs_folder": "models",
117117
"tags": [],
118118
"view_visualizations": true,
119+
"delete_viz_images": true,
119120
"log_to_csv": true,
120121
"checkpointing": {
121122
"initial_model": false,

sleap/training_profiles/baseline_medium_rf.topdown.json

+1
Original file line numberDiff line numberDiff line change
@@ -117,6 +117,7 @@
117117
"runs_folder": "models",
118118
"tags": [],
119119
"view_visualizations": true,
120+
"delete_viz_images": true,
120121
"log_to_csv": true,
121122
"checkpointing": {
122123
"initial_model": false,

sleap/training_profiles/pretrained.bottomup.json

+1
Original file line numberDiff line numberDiff line change
@@ -122,6 +122,7 @@
122122
"runs_folder": "models",
123123
"tags": [],
124124
"view_visualizations": true,
125+
"delete_viz_images": true,
125126
"log_to_csv": true,
126127
"checkpointing": {
127128
"initial_model": false,

sleap/training_profiles/pretrained.centroid.json

+1
Original file line numberDiff line numberDiff line change
@@ -113,6 +113,7 @@
113113
"runs_folder": "models",
114114
"tags": [],
115115
"view_visualizations": true,
116+
"delete_viz_images": true,
116117
"log_to_csv": true,
117118
"checkpointing": {
118119
"initial_model": false,

sleap/training_profiles/pretrained.single.json

+1
Original file line numberDiff line numberDiff line change
@@ -113,6 +113,7 @@
113113
"runs_folder": "models",
114114
"tags": [],
115115
"view_visualizations": true,
116+
"delete_viz_images": true,
116117
"log_to_csv": true,
117118
"checkpointing": {
118119
"initial_model": false,

sleap/training_profiles/pretrained.topdown.json

+1
Original file line numberDiff line numberDiff line change
@@ -114,6 +114,7 @@
114114
"runs_folder": "models",
115115
"tags": [],
116116
"view_visualizations": true,
117+
"delete_viz_images": true,
117118
"log_to_csv": true,
118119
"checkpointing": {
119120
"initial_model": false,

tests/data/models/min_tracks_2node.UNet.bottomup_multiclass/initial_config.json

+1
Original file line numberDiff line numberDiff line change
@@ -128,6 +128,7 @@
128128
"runs_folder": "models",
129129
"tags": [],
130130
"view_visualizations": false,
131+
"delete_viz_images": false,
131132
"log_to_csv": true,
132133
"checkpointing": {
133134
"initial_model": false,

tests/data/models/min_tracks_2node.UNet.bottomup_multiclass/training_config.json

+1
Original file line numberDiff line numberDiff line change
@@ -191,6 +191,7 @@
191191
"runs_folder": "models",
192192
"tags": [],
193193
"view_visualizations": false,
194+
"delete_viz_images": false,
194195
"log_to_csv": true,
195196
"checkpointing": {
196197
"initial_model": false,

tests/data/models/min_tracks_2node.UNet.topdown_multiclass/initial_config.json

+1-1
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,7 @@
141141
"runs_folder": "models",
142142
"tags": [],
143143
"view_visualizations": false,
144-
"delete_viz_images": true,
144+
"delete_viz_images": false,
145145
"zip_outputs": false,
146146
"log_to_csv": true,
147147
"checkpointing": {

tests/data/models/minimal_instance.UNet.bottomup/initial_config.json

+1
Original file line numberDiff line numberDiff line change
@@ -127,6 +127,7 @@
127127
"runs_folder": "models",
128128
"tags": [],
129129
"view_visualizations": false,
130+
"delete_viz_images": false,
130131
"log_to_csv": true,
131132
"checkpointing": {
132133
"initial_model": false,

tests/data/models/minimal_instance.UNet.bottomup/training_config.json

+1
Original file line numberDiff line numberDiff line change
@@ -192,6 +192,7 @@
192192
"runs_folder": "models",
193193
"tags": [],
194194
"view_visualizations": false,
195+
"delete_viz_images": false,
195196
"log_to_csv": true,
196197
"checkpointing": {
197198
"initial_model": false,

tests/data/models/minimal_instance.UNet.centered_instance/initial_config.json

+1
Original file line numberDiff line numberDiff line change
@@ -119,6 +119,7 @@
119119
"runs_folder": "models",
120120
"tags": [],
121121
"view_visualizations": false,
122+
"delete_viz_images": false,
122123
"log_to_csv": true,
123124
"checkpointing": {
124125
"initial_model": false,

tests/data/models/minimal_instance.UNet.centered_instance/training_config.json

+1
Original file line numberDiff line numberDiff line change
@@ -179,6 +179,7 @@
179179
"runs_folder": "models",
180180
"tags": [],
181181
"view_visualizations": false,
182+
"delete_viz_images": false,
182183
"log_to_csv": true,
183184
"checkpointing": {
184185
"initial_model": false,

tests/data/models/minimal_instance.UNet.centroid/initial_config.json

+1
Original file line numberDiff line numberDiff line change
@@ -118,6 +118,7 @@
118118
"runs_folder": "models",
119119
"tags": [],
120120
"view_visualizations": false,
121+
"delete_viz_images": false,
121122
"log_to_csv": true,
122123
"checkpointing": {
123124
"initial_model": false,

tests/data/models/minimal_instance.UNet.centroid/training_config.json

+1
Original file line numberDiff line numberDiff line change
@@ -175,6 +175,7 @@
175175
"runs_folder": "models",
176176
"tags": [],
177177
"view_visualizations": false,
178+
"delete_viz_images": false,
178179
"log_to_csv": true,
179180
"checkpointing": {
180181
"initial_model": false,

tests/data/models/minimal_robot.UNet.single_instance/initial_config.json

+1
Original file line numberDiff line numberDiff line change
@@ -120,6 +120,7 @@
120120
""
121121
],
122122
"view_visualizations": false,
123+
"delete_viz_images": false,
123124
"log_to_csv": true,
124125
"checkpointing": {
125126
"initial_model": false,

tests/data/models/minimal_robot.UNet.single_instance/training_config.json

+1
Original file line numberDiff line numberDiff line change
@@ -180,6 +180,7 @@
180180
""
181181
],
182182
"view_visualizations": false,
183+
"delete_viz_images": false,
183184
"log_to_csv": true,
184185
"checkpointing": {
185186
"initial_model": false,

0 commit comments

Comments
 (0)