Browse Source

added mouse-EPM project

niksirbi 10 months ago
parent
commit
10c24f5113
100 changed files with 78238 additions and 0 deletions
  1. BIN
      mouse-EPM/.DS_Store
  2. BIN
      mouse-EPM/derivatives/.DS_Store
  3. BIN
      mouse-EPM/derivatives/behav/.DS_Store
  4. BIN
      mouse-EPM/derivatives/behav/software-DLC_predictions/.DS_Store
  5. 158 0
      mouse-EPM/derivatives/behav/software-DLC_predictions/software-DLC_config.yaml
  6. 18488 0
      mouse-EPM/derivatives/behav/software-DLC_predictions/sub-01_ses-01_task-EPM_time-165049_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000.csv
  7. BIN
      mouse-EPM/derivatives/behav/software-DLC_predictions/sub-01_ses-01_task-EPM_time-165049_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000.h5
  8. 18488 0
      mouse-EPM/derivatives/behav/software-DLC_predictions/sub-01_ses-01_task-EPM_time-165049_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_filtered.csv
  9. BIN
      mouse-EPM/derivatives/behav/software-DLC_predictions/sub-01_ses-01_task-EPM_time-165049_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_filtered.h5
  10. 1 0
      mouse-EPM/derivatives/behav/software-DLC_predictions/sub-01_ses-01_task-EPM_time-165049_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_filtered_labeled.mp4
  11. BIN
      mouse-EPM/derivatives/behav/software-DLC_predictions/sub-01_ses-01_task-EPM_time-165049_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_meta.pickle
  12. BIN
      mouse-EPM/derivatives/behav/software-DLC_predictions/sub-01_ses-01_task-EPM_time-165049_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_plot-poses/hist.png
  13. BIN
      mouse-EPM/derivatives/behav/software-DLC_predictions/sub-01_ses-01_task-EPM_time-165049_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_plot-poses/plot-likelihood.png
  14. BIN
      mouse-EPM/derivatives/behav/software-DLC_predictions/sub-01_ses-01_task-EPM_time-165049_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_plot-poses/plot.png
  15. BIN
      mouse-EPM/derivatives/behav/software-DLC_predictions/sub-01_ses-01_task-EPM_time-165049_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_plot-poses/trajectory.png
  16. BIN
      mouse-EPM/derivatives/behav/software-DLC_predictions/sub-01_ses-01_task-EPM_time-170049_frame.png
  17. 18488 0
      mouse-EPM/derivatives/behav/software-DLC_predictions/sub-02_ses-01_task-EPM_time-185651_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000.csv
  18. BIN
      mouse-EPM/derivatives/behav/software-DLC_predictions/sub-02_ses-01_task-EPM_time-185651_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000.h5
  19. 18488 0
      mouse-EPM/derivatives/behav/software-DLC_predictions/sub-02_ses-01_task-EPM_time-185651_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_filtered.csv
  20. BIN
      mouse-EPM/derivatives/behav/software-DLC_predictions/sub-02_ses-01_task-EPM_time-185651_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_filtered.h5
  21. 1 0
      mouse-EPM/derivatives/behav/software-DLC_predictions/sub-02_ses-01_task-EPM_time-185651_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_filtered_labeled.mp4
  22. BIN
      mouse-EPM/derivatives/behav/software-DLC_predictions/sub-02_ses-01_task-EPM_time-185651_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_meta.pickle
  23. BIN
      mouse-EPM/derivatives/behav/software-DLC_predictions/sub-02_ses-01_task-EPM_time-185651_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_plot_poses/hist.png
  24. BIN
      mouse-EPM/derivatives/behav/software-DLC_predictions/sub-02_ses-01_task-EPM_time-185651_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_plot_poses/plot-likelihood.png
  25. BIN
      mouse-EPM/derivatives/behav/software-DLC_predictions/sub-02_ses-01_task-EPM_time-185651_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_plot_poses/plot.png
  26. BIN
      mouse-EPM/derivatives/behav/software-DLC_predictions/sub-02_ses-01_task-EPM_time-185651_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_plot_poses/trajectory.png
  27. BIN
      mouse-EPM/derivatives/behav/software-DLC_predictions/sub-02_ses-01_task-EPM_time-190651_frame.png
  28. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/.DS_Store
  29. 166 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/centered_instance.json
  30. 165 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/centroid.json
  31. 1 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/inference-script.sh
  32. 8 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/jobs.yaml
  33. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/labels.v001.slp
  34. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/labels.v002.slp
  35. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/.DS_Store
  36. 1 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_173835.centroid.n=20/best_model.h5
  37. 165 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_173835.centroid.n=20/initial_config.json
  38. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_173835.centroid.n=20/labels_gt.train.slp
  39. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_173835.centroid.n=20/labels_gt.val.slp
  40. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_173835.centroid.n=20/labels_pr.train.slp
  41. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_173835.centroid.n=20/labels_pr.val.slp
  42. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_173835.centroid.n=20/metrics.train.npz
  43. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_173835.centroid.n=20/metrics.val.npz
  44. 356 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_173835.centroid.n=20/training_config.json
  45. 13 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_173835.centroid.n=20/training_log.csv
  46. 1 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_175736.centered_instance.n=20/best_model.h5
  47. 166 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_175736.centered_instance.n=20/initial_config.json
  48. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_175736.centered_instance.n=20/labels_gt.train.slp
  49. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_175736.centered_instance.n=20/labels_gt.val.slp
  50. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_175736.centered_instance.n=20/labels_pr.train.slp
  51. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_175736.centered_instance.n=20/labels_pr.val.slp
  52. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_175736.centered_instance.n=20/metrics.train.npz
  53. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_175736.centered_instance.n=20/metrics.val.npz
  54. 364 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_175736.centered_instance.n=20/training_config.json
  55. 16 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_175736.centered_instance.n=20/training_log.csv
  56. 1 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_123927.centroid.n=40/best_model.h5
  57. 189 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_123927.centroid.n=40/initial_config.json
  58. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_123927.centroid.n=40/labels_gt.train.slp
  59. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_123927.centroid.n=40/labels_gt.val.slp
  60. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_123927.centroid.n=40/labels_pr.train.slp
  61. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_123927.centroid.n=40/labels_pr.val.slp
  62. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_123927.centroid.n=40/metrics.train.npz
  63. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_123927.centroid.n=40/metrics.val.npz
  64. 377 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_123927.centroid.n=40/training_config.json
  65. 9 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_123927.centroid.n=40/training_log.csv
  66. 1 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_130255.centered_instance.n=40/best_model.h5
  67. 197 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_130255.centered_instance.n=40/initial_config.json
  68. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_130255.centered_instance.n=40/labels_gt.train.slp
  69. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_130255.centered_instance.n=40/labels_gt.val.slp
  70. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_130255.centered_instance.n=40/labels_pr.train.slp
  71. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_130255.centered_instance.n=40/labels_pr.val.slp
  72. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_130255.centered_instance.n=40/metrics.train.npz
  73. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_130255.centered_instance.n=40/metrics.val.npz
  74. 385 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_130255.centered_instance.n=40/training_config.json
  75. 19 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_130255.centered_instance.n=40/training_log.csv
  76. 1 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_101726.centroid.n=60/best_model.h5
  77. 210 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_101726.centroid.n=60/initial_config.json
  78. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_101726.centroid.n=60/labels_gt.train.slp
  79. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_101726.centroid.n=60/labels_gt.val.slp
  80. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_101726.centroid.n=60/labels_pr.train.slp
  81. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_101726.centroid.n=60/labels_pr.val.slp
  82. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_101726.centroid.n=60/metrics.train.npz
  83. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_101726.centroid.n=60/metrics.val.npz
  84. 398 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_101726.centroid.n=60/training_config.json
  85. 22 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_101726.centroid.n=60/training_log.csv
  86. 1 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_105044.centered_instance.n=60/best_model.h5
  87. 218 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_105044.centered_instance.n=60/initial_config.json
  88. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_105044.centered_instance.n=60/labels_gt.train.slp
  89. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_105044.centered_instance.n=60/labels_gt.val.slp
  90. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_105044.centered_instance.n=60/labels_pr.train.slp
  91. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_105044.centered_instance.n=60/labels_pr.val.slp
  92. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_105044.centered_instance.n=60/metrics.train.npz
  93. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_105044.centered_instance.n=60/metrics.val.npz
  94. 406 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_105044.centered_instance.n=60/training_config.json
  95. 38 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_105044.centered_instance.n=60/training_log.csv
  96. 1 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231121_151642.centroid.n=60/best_model.h5
  97. 231 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231121_151642.centroid.n=60/initial_config.json
  98. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231121_151642.centroid.n=60/labels_gt.train.slp
  99. BIN
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231121_151642.centroid.n=60/labels_gt.val.slp
  100. 0 0
      mouse-EPM/derivatives/behav/software-SLEAP_project/models/231121_151642.centroid.n=60/labels_pr.train.slp

BIN
mouse-EPM/.DS_Store


BIN
mouse-EPM/derivatives/.DS_Store


BIN
mouse-EPM/derivatives/behav/.DS_Store


BIN
mouse-EPM/derivatives/behav/software-DLC_predictions/.DS_Store


+ 158 - 0
mouse-EPM/derivatives/behav/software-DLC_predictions/software-DLC_config.yaml

@@ -0,0 +1,158 @@
+    # Project definitions (do not edit)
+Task: MouseTopDown
+scorer: Loukia
+date: Sep13
+multianimalproject: false
+identity:
+
+    # Project path (change when moving around)
+project_path: D:\DlcTrainedNetworks\MASTER_DoNotModify\MouseTopDown-Loukia-2022-09-13
+
+    # Annotation data set configuration (and individual video cropping parameters)
+video_sets:
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M708149_EPM_20200317_165049331-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M708149_NOR_TRAINING_20200319_185852150-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M723400_EPM_20200317_152959980-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M724000_NOR_TRAINING_20200319_151412092-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M727755_20200316_171303023cropped-converted.mp4:
+    crop: 0, 1300, 0, 1318
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M727755_LightDarkBox_20200320_171136828-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M727755_NOR_TESTING_20200319_183309841-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M727781_EPM_20200317_175421340-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M727781_LightDarkBox_20200320_175142158-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetwork\MouseTopDown-Loukia-2022-09-13\videos\M727781_NOR_TESTING_20200319_202705887-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M727781_YMAZE_20200318_173707160-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M729733_LightDarkBox_20200320_133332476-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M729734_LightDarkBox_20200320_134630707-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M733781_20200316_183445993cropped-converted.mp4:
+    crop: 0, 1300, 0, 1280
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M808417_YMAZE_20200318_164311653-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M908825L_EPM_20201201_153441298downsampled.mp4:
+    crop: 0, 1266, 0, 1000
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M908825R_YMAZE_20201202_163753002downsampled.mp4:
+    crop: 0, 1266, 0, 1000
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M908825_LIGHTDARKBOX_20201204_153816879downsampled.mp4:
+    crop: 0, 1266, 0, 1000
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M908825_NOR_TEST_20201203_123740112downsampled.mp4:
+    crop: 0, 1266, 0, 1000
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M908825_NOR_TRAIN_20201203_113703500downsampled.mp4:
+    crop: 0, 1266, 0, 1000
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M908825_OF_20201130_155056660downsampled.mp4:
+    crop: 0, 1266, 0, 1000
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M910103L_EPM_20201201_160638671downsampled.mp4:
+    crop: 0, 1266, 0, 1000
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M910103L_YMAZE_20201202_161319135downsampled.mp4:
+    crop: 0, 1266, 0, 1000
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M910103_LIGHTDARKBOX_20201204_144551493downsampled.mp4:
+    crop: 0, 1266, 0, 1000
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M910103_NOR_TEST_20201203_152135944downsampled.mp4:
+    crop: 0, 1266, 0, 1000
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M910103_NOR_TRAIN_20201203_142151678downsampled.mp4:
+    crop: 0, 1266, 0, 1000
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M910103_OF_20201130_162822037downsampled.mp4:
+    crop: 0, 1266, 0, 1000
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M898623_NOR_TRAIN_20201108_114816752-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M898624_NOR_TEST_20201108_191243240-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M898633_NOR_TRAIN_20201108_183716376-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M898634_NOR_TEST_20201108_194857760-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M708154_EPM_20200317_185651629-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M729731_EPM_20200317_160124105-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M723998_EPM_20200317_145552459-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M728731_EPM_20200317_154549993-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M729733_EPM_20200317_180936869-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M729734_EPM_20200317_182308312-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M727781_NOR_TESTING_20200319_202705887-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M727755_EPM_20200317_170544999-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M727756_EPM_20200317_172001079-converted.mp4:
+    crop: 0, 1300, 0, 1028
+  D:\DlcTrainedNetworks\MouseTopDown-Loukia-2022-09-13\videos\M727780_EPM_20200317_174009609-converted.mp4:
+    crop: 0, 1300, 0, 1028
+bodyparts:
+- snout
+- left_ear
+- right_ear
+- centre
+- lateral_left
+- lateral_right
+- tailbase
+- tail_end
+
+
+    # Fraction of video to start/stop when extracting frames for labeling/refinement
+
+    # Fraction of video to start/stop when extracting frames for labeling/refinement
+
+    # Fraction of video to start/stop when extracting frames for labeling/refinement
+
+    # Fraction of video to start/stop when extracting frames for labeling/refinement
+
+    # Fraction of video to start/stop when extracting frames for labeling/refinement
+
+    # Fraction of video to start/stop when extracting frames for labeling/refinement
+
+    # Fraction of video to start/stop when extracting frames for labeling/refinement
+
+    # Fraction of video to start/stop when extracting frames for labeling/refinement
+
+    # Fraction of video to start/stop when extracting frames for labeling/refinement
+
+    # Fraction of video to start/stop when extracting frames for labeling/refinement
+start: 0.0
+stop: 1.0
+numframes2pick: 5
+
+    # Plotting configuration
+skeleton:
+skeleton_color: black
+pcutoff: 0.6
+dotsize: 4
+alphavalue: 0.4
+colormap: gist_rainbow
+
+    # Training,Evaluation and Analysis configuration
+TrainingFraction:
+- 0.90
+iteration: 1
+default_net_type: resnet_50
+default_augmenter: default
+snapshotindex: 3
+batch_size: 8
+
+    # Cropping Parameters (for analysis and outlier frame detection)
+cropping: false
+    #if cropping is true for analysis, then set the values here:
+x1: 0
+x2: 640
+y1: 277
+y2: 624
+
+    # Refinement configuration (parameters from annotation dataset configuration also relevant in this stage)
+corner2move2:
+- 50
+- 50
+move2corner: true

File diff suppressed because it is too large
+ 18488 - 0
mouse-EPM/derivatives/behav/software-DLC_predictions/sub-01_ses-01_task-EPM_time-165049_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000.csv


BIN
mouse-EPM/derivatives/behav/software-DLC_predictions/sub-01_ses-01_task-EPM_time-165049_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000.h5


File diff suppressed because it is too large
+ 18488 - 0
mouse-EPM/derivatives/behav/software-DLC_predictions/sub-01_ses-01_task-EPM_time-165049_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_filtered.csv


BIN
mouse-EPM/derivatives/behav/software-DLC_predictions/sub-01_ses-01_task-EPM_time-165049_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_filtered.h5


+ 1 - 0
mouse-EPM/derivatives/behav/software-DLC_predictions/sub-01_ses-01_task-EPM_time-165049_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_filtered_labeled.mp4

@@ -0,0 +1 @@
+/annex/objects/MD5-s114212259--10c0aab800bd8d680fde36bd4708f9f1

BIN
mouse-EPM/derivatives/behav/software-DLC_predictions/sub-01_ses-01_task-EPM_time-165049_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_meta.pickle


BIN
mouse-EPM/derivatives/behav/software-DLC_predictions/sub-01_ses-01_task-EPM_time-165049_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_plot-poses/hist.png


BIN
mouse-EPM/derivatives/behav/software-DLC_predictions/sub-01_ses-01_task-EPM_time-165049_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_plot-poses/plot-likelihood.png


BIN
mouse-EPM/derivatives/behav/software-DLC_predictions/sub-01_ses-01_task-EPM_time-165049_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_plot-poses/plot.png


BIN
mouse-EPM/derivatives/behav/software-DLC_predictions/sub-01_ses-01_task-EPM_time-165049_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_plot-poses/trajectory.png


BIN
mouse-EPM/derivatives/behav/software-DLC_predictions/sub-01_ses-01_task-EPM_time-170049_frame.png


File diff suppressed because it is too large
+ 18488 - 0
mouse-EPM/derivatives/behav/software-DLC_predictions/sub-02_ses-01_task-EPM_time-185651_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000.csv


BIN
mouse-EPM/derivatives/behav/software-DLC_predictions/sub-02_ses-01_task-EPM_time-185651_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000.h5


File diff suppressed because it is too large
+ 18488 - 0
mouse-EPM/derivatives/behav/software-DLC_predictions/sub-02_ses-01_task-EPM_time-185651_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_filtered.csv


BIN
mouse-EPM/derivatives/behav/software-DLC_predictions/sub-02_ses-01_task-EPM_time-185651_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_filtered.h5


+ 1 - 0
mouse-EPM/derivatives/behav/software-DLC_predictions/sub-02_ses-01_task-EPM_time-185651_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_filtered_labeled.mp4

@@ -0,0 +1 @@
+/annex/objects/MD5-s114048722--5215b4b6cebf8c70867eb58bf11e2368

BIN
mouse-EPM/derivatives/behav/software-DLC_predictions/sub-02_ses-01_task-EPM_time-185651_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_meta.pickle


BIN
mouse-EPM/derivatives/behav/software-DLC_predictions/sub-02_ses-01_task-EPM_time-185651_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_plot_poses/hist.png


BIN
mouse-EPM/derivatives/behav/software-DLC_predictions/sub-02_ses-01_task-EPM_time-185651_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_plot_poses/plot-likelihood.png


BIN
mouse-EPM/derivatives/behav/software-DLC_predictions/sub-02_ses-01_task-EPM_time-185651_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_plot_poses/plot.png


BIN
mouse-EPM/derivatives/behav/software-DLC_predictions/sub-02_ses-01_task-EPM_time-185651_video_DLC_resnet50_MouseTopDownSep13shuffle1_340000_plot_poses/trajectory.png


BIN
mouse-EPM/derivatives/behav/software-DLC_predictions/sub-02_ses-01_task-EPM_time-190651_frame.png


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/.DS_Store


+ 166 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/centered_instance.json

@@ -0,0 +1,166 @@
+{
+    "data": {
+        "labels": {
+            "training_labels": null,
+            "validation_labels": null,
+            "validation_fraction": 0.2,
+            "test_labels": null,
+            "split_by_inds": false,
+            "training_inds": null,
+            "validation_inds": null,
+            "test_inds": null,
+            "search_path_hints": [],
+            "skeletons": []
+        },
+        "preprocessing": {
+            "ensure_rgb": false,
+            "ensure_grayscale": false,
+            "imagenet_mode": null,
+            "input_scaling": 1.0,
+            "pad_to_stride": null,
+            "resize_and_pad_to_target": true,
+            "target_height": null,
+            "target_width": null
+        },
+        "instance_cropping": {
+            "center_on_part": null,
+            "crop_size": null,
+            "crop_size_detection_padding": 16
+        }
+    },
+    "model": {
+        "backbone": {
+            "leap": null,
+            "unet": {
+                "stem_stride": null,
+                "max_stride": 32,
+                "output_stride": 4,
+                "filters": 24,
+                "filters_rate": 2.0,
+                "middle_block": true,
+                "up_interpolate": true,
+                "stacks": 1
+            },
+            "hourglass": null,
+            "resnet": null,
+            "pretrained_encoder": null
+        },
+        "heads": {
+            "single_instance": null,
+            "centroid": null,
+            "centered_instance": {
+                "anchor_part": null,
+                "part_names": null,
+                "sigma": 2.5,
+                "output_stride": 4,
+                "loss_weight": 1.0,
+                "offset_refinement": false
+            },
+            "multi_instance": null,
+            "multi_class_bottomup": null,
+            "multi_class_topdown": null
+        },
+        "base_checkpoint": null
+    },
+    "optimization": {
+        "preload_data": true,
+        "augmentation_config": {
+            "rotate": true,
+            "rotation_min_angle": -180.0,
+            "rotation_max_angle": 180.0,
+            "translate": false,
+            "translate_min": -5,
+            "translate_max": 5,
+            "scale": false,
+            "scale_min": 0.9,
+            "scale_max": 1.1,
+            "uniform_noise": false,
+            "uniform_noise_min_val": 0.0,
+            "uniform_noise_max_val": 10.0,
+            "gaussian_noise": false,
+            "gaussian_noise_mean": 5.0,
+            "gaussian_noise_stddev": 1.0,
+            "contrast": false,
+            "contrast_min_gamma": 0.5,
+            "contrast_max_gamma": 2.0,
+            "brightness": false,
+            "brightness_min_val": 0.0,
+            "brightness_max_val": 10.0,
+            "random_crop": false,
+            "random_crop_height": 256,
+            "random_crop_width": 256,
+            "random_flip": true,
+            "flip_horizontal": false
+        },
+        "online_shuffling": true,
+        "shuffle_buffer_size": 128,
+        "prefetch": true,
+        "batch_size": 4,
+        "batches_per_epoch": null,
+        "min_batches_per_epoch": 200,
+        "val_batches_per_epoch": null,
+        "min_val_batches_per_epoch": 10,
+        "epochs": 50,
+        "optimizer": "adam",
+        "initial_learning_rate": 0.0001,
+        "learning_rate_schedule": {
+            "reduce_on_plateau": true,
+            "reduction_factor": 0.5,
+            "plateau_min_delta": 1e-06,
+            "plateau_patience": 5,
+            "plateau_cooldown": 3,
+            "min_learning_rate": 1e-08
+        },
+        "hard_keypoint_mining": {
+            "online_mining": false,
+            "hard_to_easy_ratio": 2.0,
+            "min_hard_keypoints": 2,
+            "max_hard_keypoints": null,
+            "loss_scale": 5.0
+        },
+        "early_stopping": {
+            "stop_training_on_plateau": true,
+            "plateau_min_delta": 1e-08,
+            "plateau_patience": 10
+        }
+    },
+    "outputs": {
+        "save_outputs": true,
+        "run_name": "231109_173631",
+        "run_name_prefix": "",
+        "run_name_suffix": ".centered_instance",
+        "runs_folder": "models",
+        "tags": [
+            ""
+        ],
+        "save_visualizations": true,
+        "delete_viz_images": true,
+        "zip_outputs": false,
+        "log_to_csv": true,
+        "checkpointing": {
+            "initial_model": false,
+            "best_model": true,
+            "every_epoch": false,
+            "latest_model": false,
+            "final_model": false
+        },
+        "tensorboard": {
+            "write_logs": false,
+            "loss_frequency": "epoch",
+            "architecture_graph": false,
+            "profile_graph": false,
+            "visualizations": true
+        },
+        "zmq": {
+            "subscribe_to_controller": false,
+            "controller_address": "tcp://127.0.0.1:9000",
+            "controller_polling_timeout": 10,
+            "publish_updates": false,
+            "publish_address": "tcp://127.0.0.1:9001"
+        }
+    },
+    "name": "",
+    "description": "",
+    "sleap_version": "1.3.1",
+    "filename": "centered_instance.json"
+}

+ 165 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/centroid.json

@@ -0,0 +1,165 @@
+{
+    "data": {
+        "labels": {
+            "training_labels": null,
+            "validation_labels": null,
+            "validation_fraction": 0.2,
+            "test_labels": null,
+            "split_by_inds": false,
+            "training_inds": null,
+            "validation_inds": null,
+            "test_inds": null,
+            "search_path_hints": [],
+            "skeletons": []
+        },
+        "preprocessing": {
+            "ensure_rgb": false,
+            "ensure_grayscale": false,
+            "imagenet_mode": null,
+            "input_scaling": 0.5,
+            "pad_to_stride": null,
+            "resize_and_pad_to_target": true,
+            "target_height": null,
+            "target_width": null
+        },
+        "instance_cropping": {
+            "center_on_part": null,
+            "crop_size": null,
+            "crop_size_detection_padding": 16
+        }
+    },
+    "model": {
+        "backbone": {
+            "leap": null,
+            "unet": {
+                "stem_stride": null,
+                "max_stride": 32,
+                "output_stride": 2,
+                "filters": 16,
+                "filters_rate": 2.0,
+                "middle_block": true,
+                "up_interpolate": true,
+                "stacks": 1
+            },
+            "hourglass": null,
+            "resnet": null,
+            "pretrained_encoder": null
+        },
+        "heads": {
+            "single_instance": null,
+            "centroid": {
+                "anchor_part": null,
+                "sigma": 2.5,
+                "output_stride": 2,
+                "loss_weight": 1.0,
+                "offset_refinement": false
+            },
+            "centered_instance": null,
+            "multi_instance": null,
+            "multi_class_bottomup": null,
+            "multi_class_topdown": null
+        },
+        "base_checkpoint": null
+    },
+    "optimization": {
+        "preload_data": true,
+        "augmentation_config": {
+            "rotate": true,
+            "rotation_min_angle": -180.0,
+            "rotation_max_angle": 180.0,
+            "translate": false,
+            "translate_min": -5,
+            "translate_max": 5,
+            "scale": false,
+            "scale_min": 0.9,
+            "scale_max": 1.1,
+            "uniform_noise": false,
+            "uniform_noise_min_val": 0.0,
+            "uniform_noise_max_val": 10.0,
+            "gaussian_noise": false,
+            "gaussian_noise_mean": 5.0,
+            "gaussian_noise_stddev": 1.0,
+            "contrast": false,
+            "contrast_min_gamma": 0.5,
+            "contrast_max_gamma": 2.0,
+            "brightness": false,
+            "brightness_min_val": 0.0,
+            "brightness_max_val": 10.0,
+            "random_crop": false,
+            "random_crop_height": 256,
+            "random_crop_width": 256,
+            "random_flip": true,
+            "flip_horizontal": false
+        },
+        "online_shuffling": true,
+        "shuffle_buffer_size": 128,
+        "prefetch": true,
+        "batch_size": 4,
+        "batches_per_epoch": null,
+        "min_batches_per_epoch": 200,
+        "val_batches_per_epoch": null,
+        "min_val_batches_per_epoch": 10,
+        "epochs": 50,
+        "optimizer": "adam",
+        "initial_learning_rate": 0.0001,
+        "learning_rate_schedule": {
+            "reduce_on_plateau": true,
+            "reduction_factor": 0.5,
+            "plateau_min_delta": 1e-06,
+            "plateau_patience": 5,
+            "plateau_cooldown": 3,
+            "min_learning_rate": 1e-08
+        },
+        "hard_keypoint_mining": {
+            "online_mining": false,
+            "hard_to_easy_ratio": 2.0,
+            "min_hard_keypoints": 2,
+            "max_hard_keypoints": null,
+            "loss_scale": 5.0
+        },
+        "early_stopping": {
+            "stop_training_on_plateau": true,
+            "plateau_min_delta": 1e-08,
+            "plateau_patience": 20
+        }
+    },
+    "outputs": {
+        "save_outputs": true,
+        "run_name": "231109_173631",
+        "run_name_prefix": "",
+        "run_name_suffix": ".centroid",
+        "runs_folder": "models",
+        "tags": [
+            ""
+        ],
+        "save_visualizations": true,
+        "delete_viz_images": true,
+        "zip_outputs": false,
+        "log_to_csv": true,
+        "checkpointing": {
+            "initial_model": false,
+            "best_model": true,
+            "every_epoch": false,
+            "latest_model": false,
+            "final_model": false
+        },
+        "tensorboard": {
+            "write_logs": false,
+            "loss_frequency": "epoch",
+            "architecture_graph": false,
+            "profile_graph": false,
+            "visualizations": true
+        },
+        "zmq": {
+            "subscribe_to_controller": false,
+            "controller_address": "tcp://127.0.0.1:9000",
+            "controller_polling_timeout": 10,
+            "publish_updates": false,
+            "publish_address": "tcp://127.0.0.1:9001"
+        }
+    },
+    "name": "",
+    "description": "",
+    "sleap_version": "1.3.1",
+    "filename": "centroid.json"
+}

+ 1 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/inference-script.sh

@@ -0,0 +1 @@
+#!/bin/bash

+ 8 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/jobs.yaml

@@ -0,0 +1,8 @@
+inference: []
+training:
+- cfg: centroid.json
+  run_path: models/231109_173631.centroid
+  train_labels: labels.v001.slp
+- cfg: centered_instance.json
+  run_path: models/231109_173631.centered_instance
+  train_labels: labels.v001.slp

BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/labels.v001.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/labels.v002.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/.DS_Store


+ 1 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_173835.centroid.n=20/best_model.h5

@@ -0,0 +1 @@
+/annex/objects/MD5-s94476560--19286015ae569bc9d9d4a54c9a67984e

+ 165 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_173835.centroid.n=20/initial_config.json

@@ -0,0 +1,165 @@
+{
+    "data": {
+        "labels": {
+            "training_labels": null,
+            "validation_labels": null,
+            "validation_fraction": 0.2,
+            "test_labels": null,
+            "split_by_inds": false,
+            "training_inds": null,
+            "validation_inds": null,
+            "test_inds": null,
+            "search_path_hints": [],
+            "skeletons": []
+        },
+        "preprocessing": {
+            "ensure_rgb": false,
+            "ensure_grayscale": false,
+            "imagenet_mode": null,
+            "input_scaling": 0.5,
+            "pad_to_stride": null,
+            "resize_and_pad_to_target": true,
+            "target_height": null,
+            "target_width": null
+        },
+        "instance_cropping": {
+            "center_on_part": null,
+            "crop_size": null,
+            "crop_size_detection_padding": 16
+        }
+    },
+    "model": {
+        "backbone": {
+            "leap": null,
+            "unet": {
+                "stem_stride": null,
+                "max_stride": 32,
+                "output_stride": 2,
+                "filters": 16,
+                "filters_rate": 2.0,
+                "middle_block": true,
+                "up_interpolate": true,
+                "stacks": 1
+            },
+            "hourglass": null,
+            "resnet": null,
+            "pretrained_encoder": null
+        },
+        "heads": {
+            "single_instance": null,
+            "centroid": {
+                "anchor_part": null,
+                "sigma": 2.5,
+                "output_stride": 2,
+                "loss_weight": 1.0,
+                "offset_refinement": false
+            },
+            "centered_instance": null,
+            "multi_instance": null,
+            "multi_class_bottomup": null,
+            "multi_class_topdown": null
+        },
+        "base_checkpoint": null
+    },
+    "optimization": {
+        "preload_data": true,
+        "augmentation_config": {
+            "rotate": true,
+            "rotation_min_angle": -180.0,
+            "rotation_max_angle": 180.0,
+            "translate": false,
+            "translate_min": -5,
+            "translate_max": 5,
+            "scale": false,
+            "scale_min": 0.9,
+            "scale_max": 1.1,
+            "uniform_noise": false,
+            "uniform_noise_min_val": 0.0,
+            "uniform_noise_max_val": 10.0,
+            "gaussian_noise": false,
+            "gaussian_noise_mean": 5.0,
+            "gaussian_noise_stddev": 1.0,
+            "contrast": false,
+            "contrast_min_gamma": 0.5,
+            "contrast_max_gamma": 2.0,
+            "brightness": false,
+            "brightness_min_val": 0.0,
+            "brightness_max_val": 10.0,
+            "random_crop": false,
+            "random_crop_height": 256,
+            "random_crop_width": 256,
+            "random_flip": true,
+            "flip_horizontal": false
+        },
+        "online_shuffling": true,
+        "shuffle_buffer_size": 128,
+        "prefetch": true,
+        "batch_size": 4,
+        "batches_per_epoch": null,
+        "min_batches_per_epoch": 200,
+        "val_batches_per_epoch": null,
+        "min_val_batches_per_epoch": 10,
+        "epochs": 50,
+        "optimizer": "adam",
+        "initial_learning_rate": 0.0001,
+        "learning_rate_schedule": {
+            "reduce_on_plateau": true,
+            "reduction_factor": 0.5,
+            "plateau_min_delta": 1e-06,
+            "plateau_patience": 5,
+            "plateau_cooldown": 3,
+            "min_learning_rate": 1e-08
+        },
+        "hard_keypoint_mining": {
+            "online_mining": false,
+            "hard_to_easy_ratio": 2.0,
+            "min_hard_keypoints": 2,
+            "max_hard_keypoints": null,
+            "loss_scale": 5.0
+        },
+        "early_stopping": {
+            "stop_training_on_plateau": true,
+            "plateau_min_delta": 1e-08,
+            "plateau_patience": 20
+        }
+    },
+    "outputs": {
+        "save_outputs": true,
+        "run_name": "231109_173835.centroid.n=20",
+        "run_name_prefix": "",
+        "run_name_suffix": "",
+        "runs_folder": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/models",
+        "tags": [
+            ""
+        ],
+        "save_visualizations": true,
+        "delete_viz_images": true,
+        "zip_outputs": false,
+        "log_to_csv": true,
+        "checkpointing": {
+            "initial_model": false,
+            "best_model": true,
+            "every_epoch": false,
+            "latest_model": false,
+            "final_model": false
+        },
+        "tensorboard": {
+            "write_logs": false,
+            "loss_frequency": "epoch",
+            "architecture_graph": false,
+            "profile_graph": false,
+            "visualizations": true
+        },
+        "zmq": {
+            "subscribe_to_controller": true,
+            "controller_address": "tcp://127.0.0.1:9000",
+            "controller_polling_timeout": 10,
+            "publish_updates": true,
+            "publish_address": "tcp://127.0.0.1:9001"
+        }
+    },
+    "name": "",
+    "description": "",
+    "sleap_version": "1.3.1",
+    "filename": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/models/231109_173835.centroid.n=20/initial_config.json"
+}

BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_173835.centroid.n=20/labels_gt.train.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_173835.centroid.n=20/labels_gt.val.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_173835.centroid.n=20/labels_pr.train.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_173835.centroid.n=20/labels_pr.val.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_173835.centroid.n=20/metrics.train.npz


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_173835.centroid.n=20/metrics.val.npz


+ 356 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_173835.centroid.n=20/training_config.json

@@ -0,0 +1,356 @@
+{
+    "data": {
+        "labels": {
+            "training_labels": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/labels.v001.slp",
+            "validation_labels": null,
+            "validation_fraction": 0.2,
+            "test_labels": null,
+            "split_by_inds": false,
+            "training_inds": [
+                14,
+                16,
+                11,
+                17,
+                13,
+                8,
+                1,
+                3,
+                15,
+                2,
+                10,
+                4,
+                5,
+                6,
+                7,
+                19
+            ],
+            "validation_inds": [
+                18,
+                9,
+                12,
+                0
+            ],
+            "test_inds": null,
+            "search_path_hints": [
+                ""
+            ],
+            "skeletons": [
+                {
+                    "directed": true,
+                    "graph": {
+                        "name": "Skeleton-3",
+                        "num_edges_inserted": 15
+                    },
+                    "links": [
+                        {
+                            "edge_insert_idx": 5,
+                            "key": 0,
+                            "source": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "snout",
+                                    "weight": 1.0
+                                }
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "left_ear",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/reduce": [
+                                    {
+                                        "py/type": "sleap.skeleton.EdgeType"
+                                    },
+                                    {
+                                        "py/tuple": [
+                                            1
+                                        ]
+                                    }
+                                ]
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 6,
+                            "key": 0,
+                            "source": {
+                                "py/id": 1
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "right_ear",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 7,
+                            "key": 0,
+                            "source": {
+                                "py/id": 1
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "centre",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 8,
+                            "key": 0,
+                            "source": {
+                                "py/id": 2
+                            },
+                            "target": {
+                                "py/id": 5
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 9,
+                            "key": 0,
+                            "source": {
+                                "py/id": 4
+                            },
+                            "target": {
+                                "py/id": 5
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 10,
+                            "key": 0,
+                            "source": {
+                                "py/id": 5
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "tail_base",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 14,
+                            "key": 0,
+                            "source": {
+                                "py/id": 6
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "tail_end",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        }
+                    ],
+                    "multigraph": true,
+                    "nodes": [
+                        {
+                            "id": {
+                                "py/id": 1
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 2
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 4
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 5
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 6
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 7
+                            }
+                        }
+                    ]
+                }
+            ]
+        },
+        "preprocessing": {
+            "ensure_rgb": false,
+            "ensure_grayscale": false,
+            "imagenet_mode": null,
+            "input_scaling": 0.5,
+            "pad_to_stride": 32,
+            "resize_and_pad_to_target": true,
+            "target_height": 1028,
+            "target_width": 1300
+        },
+        "instance_cropping": {
+            "center_on_part": null,
+            "crop_size": null,
+            "crop_size_detection_padding": 16
+        }
+    },
+    "model": {
+        "backbone": {
+            "leap": null,
+            "unet": {
+                "stem_stride": null,
+                "max_stride": 32,
+                "output_stride": 2,
+                "filters": 16,
+                "filters_rate": 2.0,
+                "middle_block": true,
+                "up_interpolate": true,
+                "stacks": 1
+            },
+            "hourglass": null,
+            "resnet": null,
+            "pretrained_encoder": null
+        },
+        "heads": {
+            "single_instance": null,
+            "centroid": {
+                "anchor_part": null,
+                "sigma": 2.5,
+                "output_stride": 2,
+                "loss_weight": 1.0,
+                "offset_refinement": false
+            },
+            "centered_instance": null,
+            "multi_instance": null,
+            "multi_class_bottomup": null,
+            "multi_class_topdown": null
+        },
+        "base_checkpoint": null
+    },
+    "optimization": {
+        "preload_data": true,
+        "augmentation_config": {
+            "rotate": true,
+            "rotation_min_angle": -180.0,
+            "rotation_max_angle": 180.0,
+            "translate": false,
+            "translate_min": -5,
+            "translate_max": 5,
+            "scale": false,
+            "scale_min": 0.9,
+            "scale_max": 1.1,
+            "uniform_noise": false,
+            "uniform_noise_min_val": 0.0,
+            "uniform_noise_max_val": 10.0,
+            "gaussian_noise": false,
+            "gaussian_noise_mean": 5.0,
+            "gaussian_noise_stddev": 1.0,
+            "contrast": false,
+            "contrast_min_gamma": 0.5,
+            "contrast_max_gamma": 2.0,
+            "brightness": false,
+            "brightness_min_val": 0.0,
+            "brightness_max_val": 10.0,
+            "random_crop": false,
+            "random_crop_height": 256,
+            "random_crop_width": 256,
+            "random_flip": true,
+            "flip_horizontal": false
+        },
+        "online_shuffling": true,
+        "shuffle_buffer_size": 128,
+        "prefetch": true,
+        "batch_size": 4,
+        "batches_per_epoch": 200,
+        "min_batches_per_epoch": 200,
+        "val_batches_per_epoch": 10,
+        "min_val_batches_per_epoch": 10,
+        "epochs": 50,
+        "optimizer": "adam",
+        "initial_learning_rate": 0.0001,
+        "learning_rate_schedule": {
+            "reduce_on_plateau": true,
+            "reduction_factor": 0.5,
+            "plateau_min_delta": 1e-06,
+            "plateau_patience": 5,
+            "plateau_cooldown": 3,
+            "min_learning_rate": 1e-08
+        },
+        "hard_keypoint_mining": {
+            "online_mining": false,
+            "hard_to_easy_ratio": 2.0,
+            "min_hard_keypoints": 2,
+            "max_hard_keypoints": null,
+            "loss_scale": 5.0
+        },
+        "early_stopping": {
+            "stop_training_on_plateau": true,
+            "plateau_min_delta": 1e-08,
+            "plateau_patience": 20
+        }
+    },
+    "outputs": {
+        "save_outputs": true,
+        "run_name": "231109_173835.centroid.n=20",
+        "run_name_prefix": "",
+        "run_name_suffix": "",
+        "runs_folder": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/models",
+        "tags": [
+            ""
+        ],
+        "save_visualizations": true,
+        "delete_viz_images": true,
+        "zip_outputs": false,
+        "log_to_csv": true,
+        "checkpointing": {
+            "initial_model": false,
+            "best_model": true,
+            "every_epoch": false,
+            "latest_model": false,
+            "final_model": false
+        },
+        "tensorboard": {
+            "write_logs": false,
+            "loss_frequency": "epoch",
+            "architecture_graph": false,
+            "profile_graph": false,
+            "visualizations": true
+        },
+        "zmq": {
+            "subscribe_to_controller": true,
+            "controller_address": "tcp://127.0.0.1:9000",
+            "controller_polling_timeout": 10,
+            "publish_updates": true,
+            "publish_address": "tcp://127.0.0.1:9001"
+        }
+    },
+    "name": "",
+    "description": "",
+    "sleap_version": "1.3.1",
+    "filename": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/models/231109_173835.centroid.n=20/training_config.json"
+}

+ 13 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_173835.centroid.n=20/training_log.csv

@@ -0,0 +1,13 @@
+epoch,loss,lr,val_loss
+0,0.00021083926549181342,1e-04,0.00021268916316330433
+1,0.00020927441073581576,1e-04,0.0002118827251251787
+2,0.00020380962814670056,1e-04,0.00020297695300541818
+3,0.00019226822769269347,1e-04,0.00019314710516482592
+4,0.00018521152378525585,1e-04,0.0001944211107911542
+5,0.00017891399329528213,1e-04,0.0002003508125199005
+6,0.00017163110896945,1e-04,0.00020616284746211022
+7,0.00016339478315785527,1e-04,0.00019474081636872143
+8,0.0001610323233762756,1e-04,0.00025789038045331836
+9,0.0001456212776247412,5e-05,0.00021338823717087507
+10,0.00013246058369986713,5e-05,0.00021206261590123177
+11,0.00012467765191104263,5e-05,0.00021831318736076355

+ 1 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_175736.centered_instance.n=20/best_model.h5

@@ -0,0 +1 @@
+/annex/objects/MD5-s211231488--4455a8bdb023fa416116455ba5e10953

+ 166 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_175736.centered_instance.n=20/initial_config.json

@@ -0,0 +1,166 @@
+{
+    "data": {
+        "labels": {
+            "training_labels": null,
+            "validation_labels": null,
+            "validation_fraction": 0.2,
+            "test_labels": null,
+            "split_by_inds": false,
+            "training_inds": null,
+            "validation_inds": null,
+            "test_inds": null,
+            "search_path_hints": [],
+            "skeletons": []
+        },
+        "preprocessing": {
+            "ensure_rgb": false,
+            "ensure_grayscale": false,
+            "imagenet_mode": null,
+            "input_scaling": 1.0,
+            "pad_to_stride": null,
+            "resize_and_pad_to_target": true,
+            "target_height": null,
+            "target_width": null
+        },
+        "instance_cropping": {
+            "center_on_part": null,
+            "crop_size": null,
+            "crop_size_detection_padding": 16
+        }
+    },
+    "model": {
+        "backbone": {
+            "leap": null,
+            "unet": {
+                "stem_stride": null,
+                "max_stride": 32,
+                "output_stride": 4,
+                "filters": 24,
+                "filters_rate": 2.0,
+                "middle_block": true,
+                "up_interpolate": true,
+                "stacks": 1
+            },
+            "hourglass": null,
+            "resnet": null,
+            "pretrained_encoder": null
+        },
+        "heads": {
+            "single_instance": null,
+            "centroid": null,
+            "centered_instance": {
+                "anchor_part": null,
+                "part_names": null,
+                "sigma": 2.5,
+                "output_stride": 4,
+                "loss_weight": 1.0,
+                "offset_refinement": false
+            },
+            "multi_instance": null,
+            "multi_class_bottomup": null,
+            "multi_class_topdown": null
+        },
+        "base_checkpoint": null
+    },
+    "optimization": {
+        "preload_data": true,
+        "augmentation_config": {
+            "rotate": true,
+            "rotation_min_angle": -180.0,
+            "rotation_max_angle": 180.0,
+            "translate": false,
+            "translate_min": -5,
+            "translate_max": 5,
+            "scale": false,
+            "scale_min": 0.9,
+            "scale_max": 1.1,
+            "uniform_noise": false,
+            "uniform_noise_min_val": 0.0,
+            "uniform_noise_max_val": 10.0,
+            "gaussian_noise": false,
+            "gaussian_noise_mean": 5.0,
+            "gaussian_noise_stddev": 1.0,
+            "contrast": false,
+            "contrast_min_gamma": 0.5,
+            "contrast_max_gamma": 2.0,
+            "brightness": false,
+            "brightness_min_val": 0.0,
+            "brightness_max_val": 10.0,
+            "random_crop": false,
+            "random_crop_height": 256,
+            "random_crop_width": 256,
+            "random_flip": true,
+            "flip_horizontal": false
+        },
+        "online_shuffling": true,
+        "shuffle_buffer_size": 128,
+        "prefetch": true,
+        "batch_size": 4,
+        "batches_per_epoch": null,
+        "min_batches_per_epoch": 200,
+        "val_batches_per_epoch": null,
+        "min_val_batches_per_epoch": 10,
+        "epochs": 50,
+        "optimizer": "adam",
+        "initial_learning_rate": 0.0001,
+        "learning_rate_schedule": {
+            "reduce_on_plateau": true,
+            "reduction_factor": 0.5,
+            "plateau_min_delta": 1e-06,
+            "plateau_patience": 5,
+            "plateau_cooldown": 3,
+            "min_learning_rate": 1e-08
+        },
+        "hard_keypoint_mining": {
+            "online_mining": false,
+            "hard_to_easy_ratio": 2.0,
+            "min_hard_keypoints": 2,
+            "max_hard_keypoints": null,
+            "loss_scale": 5.0
+        },
+        "early_stopping": {
+            "stop_training_on_plateau": true,
+            "plateau_min_delta": 1e-08,
+            "plateau_patience": 10
+        }
+    },
+    "outputs": {
+        "save_outputs": true,
+        "run_name": "231109_175736.centered_instance.n=20",
+        "run_name_prefix": "",
+        "run_name_suffix": "",
+        "runs_folder": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/models",
+        "tags": [
+            ""
+        ],
+        "save_visualizations": true,
+        "delete_viz_images": true,
+        "zip_outputs": false,
+        "log_to_csv": true,
+        "checkpointing": {
+            "initial_model": false,
+            "best_model": true,
+            "every_epoch": false,
+            "latest_model": false,
+            "final_model": false
+        },
+        "tensorboard": {
+            "write_logs": false,
+            "loss_frequency": "epoch",
+            "architecture_graph": false,
+            "profile_graph": false,
+            "visualizations": true
+        },
+        "zmq": {
+            "subscribe_to_controller": true,
+            "controller_address": "tcp://127.0.0.1:9000",
+            "controller_polling_timeout": 10,
+            "publish_updates": true,
+            "publish_address": "tcp://127.0.0.1:9001"
+        }
+    },
+    "name": "",
+    "description": "",
+    "sleap_version": "1.3.1",
+    "filename": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/models/231109_175736.centered_instance.n=20/initial_config.json"
+}

BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_175736.centered_instance.n=20/labels_gt.train.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_175736.centered_instance.n=20/labels_gt.val.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_175736.centered_instance.n=20/labels_pr.train.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_175736.centered_instance.n=20/labels_pr.val.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_175736.centered_instance.n=20/metrics.train.npz


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_175736.centered_instance.n=20/metrics.val.npz


+ 364 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_175736.centered_instance.n=20/training_config.json

@@ -0,0 +1,364 @@
+{
+    "data": {
+        "labels": {
+            "training_labels": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/labels.v001.slp",
+            "validation_labels": null,
+            "validation_fraction": 0.2,
+            "test_labels": null,
+            "split_by_inds": false,
+            "training_inds": [
+                11,
+                13,
+                8,
+                16,
+                2,
+                1,
+                12,
+                17,
+                6,
+                4,
+                14,
+                10,
+                3,
+                15,
+                7,
+                5
+            ],
+            "validation_inds": [
+                19,
+                9,
+                18,
+                0
+            ],
+            "test_inds": null,
+            "search_path_hints": [
+                ""
+            ],
+            "skeletons": [
+                {
+                    "directed": true,
+                    "graph": {
+                        "name": "Skeleton-3",
+                        "num_edges_inserted": 15
+                    },
+                    "links": [
+                        {
+                            "edge_insert_idx": 5,
+                            "key": 0,
+                            "source": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "snout",
+                                    "weight": 1.0
+                                }
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "left_ear",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/reduce": [
+                                    {
+                                        "py/type": "sleap.skeleton.EdgeType"
+                                    },
+                                    {
+                                        "py/tuple": [
+                                            1
+                                        ]
+                                    }
+                                ]
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 6,
+                            "key": 0,
+                            "source": {
+                                "py/id": 1
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "right_ear",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 7,
+                            "key": 0,
+                            "source": {
+                                "py/id": 1
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "centre",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 8,
+                            "key": 0,
+                            "source": {
+                                "py/id": 2
+                            },
+                            "target": {
+                                "py/id": 5
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 9,
+                            "key": 0,
+                            "source": {
+                                "py/id": 4
+                            },
+                            "target": {
+                                "py/id": 5
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 10,
+                            "key": 0,
+                            "source": {
+                                "py/id": 5
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "tail_base",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 14,
+                            "key": 0,
+                            "source": {
+                                "py/id": 6
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "tail_end",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        }
+                    ],
+                    "multigraph": true,
+                    "nodes": [
+                        {
+                            "id": {
+                                "py/id": 1
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 2
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 4
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 5
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 6
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 7
+                            }
+                        }
+                    ]
+                }
+            ]
+        },
+        "preprocessing": {
+            "ensure_rgb": false,
+            "ensure_grayscale": false,
+            "imagenet_mode": null,
+            "input_scaling": 1.0,
+            "pad_to_stride": 1,
+            "resize_and_pad_to_target": true,
+            "target_height": 1028,
+            "target_width": 1300
+        },
+        "instance_cropping": {
+            "center_on_part": null,
+            "crop_size": 256,
+            "crop_size_detection_padding": 16
+        }
+    },
+    "model": {
+        "backbone": {
+            "leap": null,
+            "unet": {
+                "stem_stride": null,
+                "max_stride": 32,
+                "output_stride": 4,
+                "filters": 24,
+                "filters_rate": 2.0,
+                "middle_block": true,
+                "up_interpolate": true,
+                "stacks": 1
+            },
+            "hourglass": null,
+            "resnet": null,
+            "pretrained_encoder": null
+        },
+        "heads": {
+            "single_instance": null,
+            "centroid": null,
+            "centered_instance": {
+                "anchor_part": null,
+                "part_names": [
+                    "snout",
+                    "left_ear",
+                    "right_ear",
+                    "centre",
+                    "tail_base",
+                    "tail_end"
+                ],
+                "sigma": 2.5,
+                "output_stride": 4,
+                "loss_weight": 1.0,
+                "offset_refinement": false
+            },
+            "multi_instance": null,
+            "multi_class_bottomup": null,
+            "multi_class_topdown": null
+        },
+        "base_checkpoint": null
+    },
+    "optimization": {
+        "preload_data": true,
+        "augmentation_config": {
+            "rotate": true,
+            "rotation_min_angle": -180.0,
+            "rotation_max_angle": 180.0,
+            "translate": false,
+            "translate_min": -5,
+            "translate_max": 5,
+            "scale": false,
+            "scale_min": 0.9,
+            "scale_max": 1.1,
+            "uniform_noise": false,
+            "uniform_noise_min_val": 0.0,
+            "uniform_noise_max_val": 10.0,
+            "gaussian_noise": false,
+            "gaussian_noise_mean": 5.0,
+            "gaussian_noise_stddev": 1.0,
+            "contrast": false,
+            "contrast_min_gamma": 0.5,
+            "contrast_max_gamma": 2.0,
+            "brightness": false,
+            "brightness_min_val": 0.0,
+            "brightness_max_val": 10.0,
+            "random_crop": false,
+            "random_crop_height": 256,
+            "random_crop_width": 256,
+            "random_flip": true,
+            "flip_horizontal": false
+        },
+        "online_shuffling": true,
+        "shuffle_buffer_size": 128,
+        "prefetch": true,
+        "batch_size": 4,
+        "batches_per_epoch": 200,
+        "min_batches_per_epoch": 200,
+        "val_batches_per_epoch": 10,
+        "min_val_batches_per_epoch": 10,
+        "epochs": 50,
+        "optimizer": "adam",
+        "initial_learning_rate": 0.0001,
+        "learning_rate_schedule": {
+            "reduce_on_plateau": true,
+            "reduction_factor": 0.5,
+            "plateau_min_delta": 1e-06,
+            "plateau_patience": 5,
+            "plateau_cooldown": 3,
+            "min_learning_rate": 1e-08
+        },
+        "hard_keypoint_mining": {
+            "online_mining": false,
+            "hard_to_easy_ratio": 2.0,
+            "min_hard_keypoints": 2,
+            "max_hard_keypoints": null,
+            "loss_scale": 5.0
+        },
+        "early_stopping": {
+            "stop_training_on_plateau": true,
+            "plateau_min_delta": 1e-08,
+            "plateau_patience": 10
+        }
+    },
+    "outputs": {
+        "save_outputs": true,
+        "run_name": "231109_175736.centered_instance.n=20",
+        "run_name_prefix": "",
+        "run_name_suffix": "",
+        "runs_folder": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/models",
+        "tags": [
+            ""
+        ],
+        "save_visualizations": true,
+        "delete_viz_images": true,
+        "zip_outputs": false,
+        "log_to_csv": true,
+        "checkpointing": {
+            "initial_model": false,
+            "best_model": true,
+            "every_epoch": false,
+            "latest_model": false,
+            "final_model": false
+        },
+        "tensorboard": {
+            "write_logs": false,
+            "loss_frequency": "epoch",
+            "architecture_graph": false,
+            "profile_graph": false,
+            "visualizations": true
+        },
+        "zmq": {
+            "subscribe_to_controller": true,
+            "controller_address": "tcp://127.0.0.1:9000",
+            "controller_polling_timeout": 10,
+            "publish_updates": true,
+            "publish_address": "tcp://127.0.0.1:9001"
+        }
+    },
+    "name": "",
+    "description": "",
+    "sleap_version": "1.3.1",
+    "filename": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/models/231109_175736.centered_instance.n=20/training_config.json"
+}

+ 16 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231109_175736.centered_instance.n=20/training_log.csv

@@ -0,0 +1,16 @@
+epoch,centre,left_ear,loss,lr,right_ear,snout,tail_base,tail_end,val_centre,val_left_ear,val_loss,val_right_ear,val_snout,val_tail_base,val_tail_end
+0,0.002975167939439416,0.0044427430257201195,0.004062215331941843,1e-04,0.004400129895657301,0.0039957608096301556,0.0042338864877820015,0.004325599875301123,0.002151086227968335,0.0040987092070281506,0.0038608938921242952,0.00399946141988039,0.0042409272864460945,0.0041845375671982765,0.0044906409457325935
+1,0.002036958234384656,0.0037343264557421207,0.003452027216553688,1e-04,0.003484944812953472,0.003589258762076497,0.003565155901014805,0.004301523789763451,0.0017771354177966714,0.0030613900162279606,0.0032516312785446644,0.003229345427826047,0.003393381368368864,0.003615848021581769,0.00443268520757556
+2,0.0015177408931776881,0.002993342000991106,0.0027133419644087553,1e-04,0.00247642258182168,0.0026552046183496714,0.0023807210382074118,0.004256619140505791,0.0013319492572918534,0.0024787031579762697,0.0024812130723148584,0.0028147336561232805,0.0020691656973212957,0.0017632136587053537,0.004429512657225132
+3,0.001060483162291348,0.0027302054222673178,0.0022362638264894485,1e-04,0.0021447280887514353,0.0017844740068539977,0.001508807297796011,0.004188882652670145,0.0014947112649679184,0.002505603479221463,0.0022650992032140493,0.002519984031096101,0.0012749942252412438,0.001393066719174385,0.004402234684675932
+4,0.0008644217741675675,0.00236140051856637,0.0018557378789409995,1e-04,0.0018127203220501542,0.001062334282323718,0.0009542381158098578,0.004079308360815048,0.0015006187604740262,0.002503448398783803,0.0021723746322095394,0.002736056922003627,0.0010273961815983057,0.0008500386029481888,0.0044166878797113895
+5,0.0007275349926203489,0.0018218164332211018,0.0015934775583446026,1e-04,0.0014281354378908873,0.0007874799775891006,0.0008261989569291472,0.003969697747379541,0.0012084281770512462,0.0034446336794644594,0.002257307292893529,0.0029235580004751682,0.001062246272340417,0.0006242402596399188,0.004280734807252884
+6,0.0005153779638931155,0.0010349468793720007,0.0012254455359652638,1e-04,0.0008332280558533967,0.0006081011379137635,0.0006065416964702308,0.0037544756196439266,0.0010904066730290651,0.0033260241616517305,0.0022684913128614426,0.0035296943970024586,0.000998584902845323,0.0004935756442137063,0.004172661807388067
+7,0.0003683237300720066,0.0006774009671062231,0.000981622957624495,1e-04,0.0005414211191236973,0.0004333715478423983,0.0004685491439886391,0.0034006712958216667,0.001226537860929966,0.004476710222661495,0.002570545766502619,0.004144326783716679,0.0007938244380056858,0.0005162546876817942,0.0042656185105443
+8,0.0002709372784011066,0.00041227543260902166,0.0007886385428719223,1e-04,0.0003291151951998472,0.0003661088994704187,0.00035156498779542744,0.0030018279794603586,0.001452430384233594,0.003981937654316425,0.0025286648888140917,0.004297957755625248,0.0006902028108015656,0.00046721691614948213,0.00428224541246891
+9,0.00021474555251188576,0.0002854090998880565,0.0006318267551250756,1e-04,0.00024002778809517622,0.00023444881662726402,0.0002756599278654903,0.002540669171139598,0.0011996440589427948,0.003970513120293617,0.002482522279024124,0.003868511412292719,0.0006352971540763974,0.00048222276382148266,0.004738945513963699
+10,0.00011570048809517175,0.00014386391558218747,0.0003698522341437638,5e-05,0.00013706031313631684,0.00014529324835166335,0.00015428662300109863,0.001522909151390195,0.001392959849908948,0.003575350623577833,0.0024379207752645016,0.003936447203159332,0.0004637093807104975,0.00046858322457410395,0.004790475592017174
+11,8.882061229087412e-05,0.00011418702342780307,0.0002814768813550472,5e-05,0.00011190932855242863,0.00011502973939059302,0.0001160706378868781,0.0011428433936089277,0.001439206418581307,0.004203219898045063,0.0027024096343666315,0.004152849316596985,0.0005585373728536069,0.0005686314543709159,0.005292013753205538
+12,7.326801278395578e-05,0.00010244335135212168,0.00024102958559524268,5e-05,0.00010120250226464123,0.00010582867980701849,9.984723874367774e-05,0.0009635876631364226,0.001376403495669365,0.004064911976456642,0.0025864136405289173,0.004194085486233234,0.0005844272091053426,0.0004420395416673273,0.004856613464653492
+13,7.278163684532046e-05,8.986399188870564e-05,0.00020969360775779933,5e-05,8.68533825268969e-05,9.67364467214793e-05,9.667254926171154e-05,0.0008152539376169443,0.001352245919406414,0.004099206067621708,0.002499791793525219,0.00411578081548214,0.0005689889658242464,0.00037656803033314645,0.004485961049795151
+14,6.378271791618317e-05,8.392972813453525e-05,0.00016302017320413142,5e-05,7.973214815137908e-05,8.878341759555042e-05,8.815217006485909e-05,0.0005737408646382391,0.0013924259692430496,0.0042595025151968,0.0027142788749188185,0.004331274423748255,0.0004821028560400009,0.00038181975833140314,0.0054385485127568245

+ 1 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_123927.centroid.n=40/best_model.h5

@@ -0,0 +1 @@
+/annex/objects/MD5-s94476560--5b8e884575333053af1d0ad29aa273fb

+ 189 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_123927.centroid.n=40/initial_config.json

@@ -0,0 +1,189 @@
+{
+    "data": {
+        "labels": {
+            "training_labels": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/labels.v001.slp",
+            "validation_labels": null,
+            "validation_fraction": 0.2,
+            "test_labels": null,
+            "split_by_inds": false,
+            "training_inds": [
+                14,
+                16,
+                11,
+                17,
+                13,
+                8,
+                1,
+                3,
+                15,
+                2,
+                10,
+                4,
+                5,
+                6,
+                7,
+                19
+            ],
+            "validation_inds": [
+                18,
+                9,
+                12,
+                0
+            ],
+            "test_inds": null,
+            "search_path_hints": [
+                ""
+            ],
+            "skeletons": []
+        },
+        "preprocessing": {
+            "ensure_rgb": false,
+            "ensure_grayscale": false,
+            "imagenet_mode": null,
+            "input_scaling": 0.5,
+            "pad_to_stride": 32,
+            "resize_and_pad_to_target": true,
+            "target_height": 1028,
+            "target_width": 1300
+        },
+        "instance_cropping": {
+            "center_on_part": null,
+            "crop_size": null,
+            "crop_size_detection_padding": 16
+        }
+    },
+    "model": {
+        "backbone": {
+            "leap": null,
+            "unet": {
+                "stem_stride": null,
+                "max_stride": 32,
+                "output_stride": 2,
+                "filters": 16,
+                "filters_rate": 2.0,
+                "middle_block": true,
+                "up_interpolate": true,
+                "stacks": 1
+            },
+            "hourglass": null,
+            "resnet": null,
+            "pretrained_encoder": null
+        },
+        "heads": {
+            "single_instance": null,
+            "centroid": {
+                "anchor_part": null,
+                "sigma": 2.5,
+                "output_stride": 2,
+                "loss_weight": 1.0,
+                "offset_refinement": false
+            },
+            "centered_instance": null,
+            "multi_instance": null,
+            "multi_class_bottomup": null,
+            "multi_class_topdown": null
+        },
+        "base_checkpoint": null
+    },
+    "optimization": {
+        "preload_data": true,
+        "augmentation_config": {
+            "rotate": true,
+            "rotation_min_angle": -180.0,
+            "rotation_max_angle": 180.0,
+            "translate": false,
+            "translate_min": -5,
+            "translate_max": 5,
+            "scale": false,
+            "scale_min": 0.9,
+            "scale_max": 1.1,
+            "uniform_noise": false,
+            "uniform_noise_min_val": 0.0,
+            "uniform_noise_max_val": 10.0,
+            "gaussian_noise": false,
+            "gaussian_noise_mean": 5.0,
+            "gaussian_noise_stddev": 1.0,
+            "contrast": false,
+            "contrast_min_gamma": 0.5,
+            "contrast_max_gamma": 2.0,
+            "brightness": false,
+            "brightness_min_val": 0.0,
+            "brightness_max_val": 10.0,
+            "random_crop": false,
+            "random_crop_height": 256,
+            "random_crop_width": 256,
+            "random_flip": false,
+            "flip_horizontal": false
+        },
+        "online_shuffling": true,
+        "shuffle_buffer_size": 128,
+        "prefetch": true,
+        "batch_size": 4,
+        "batches_per_epoch": 200,
+        "min_batches_per_epoch": 200,
+        "val_batches_per_epoch": 10,
+        "min_val_batches_per_epoch": 10,
+        "epochs": 50,
+        "optimizer": "adam",
+        "initial_learning_rate": 0.0001,
+        "learning_rate_schedule": {
+            "reduce_on_plateau": true,
+            "reduction_factor": 0.5,
+            "plateau_min_delta": 1e-06,
+            "plateau_patience": 5,
+            "plateau_cooldown": 3,
+            "min_learning_rate": 1e-08
+        },
+        "hard_keypoint_mining": {
+            "online_mining": false,
+            "hard_to_easy_ratio": 2.0,
+            "min_hard_keypoints": 2,
+            "max_hard_keypoints": null,
+            "loss_scale": 5.0
+        },
+        "early_stopping": {
+            "stop_training_on_plateau": true,
+            "plateau_min_delta": 1e-08,
+            "plateau_patience": 20
+        }
+    },
+    "outputs": {
+        "save_outputs": true,
+        "run_name": "231110_123927.centroid.n=40",
+        "run_name_prefix": "",
+        "run_name_suffix": "",
+        "runs_folder": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/models",
+        "tags": [
+            ""
+        ],
+        "save_visualizations": true,
+        "delete_viz_images": true,
+        "zip_outputs": false,
+        "log_to_csv": true,
+        "checkpointing": {
+            "initial_model": false,
+            "best_model": true,
+            "every_epoch": false,
+            "latest_model": false,
+            "final_model": false
+        },
+        "tensorboard": {
+            "write_logs": false,
+            "loss_frequency": "epoch",
+            "architecture_graph": false,
+            "profile_graph": false,
+            "visualizations": true
+        },
+        "zmq": {
+            "subscribe_to_controller": true,
+            "controller_address": "tcp://127.0.0.1:9000",
+            "controller_polling_timeout": 10,
+            "publish_updates": true,
+            "publish_address": "tcp://127.0.0.1:9001"
+        }
+    },
+    "name": "",
+    "description": "",
+    "sleap_version": "1.3.1",
+    "filename": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/models/231110_123927.centroid.n=40/initial_config.json"
+}

BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_123927.centroid.n=40/labels_gt.train.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_123927.centroid.n=40/labels_gt.val.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_123927.centroid.n=40/labels_pr.train.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_123927.centroid.n=40/labels_pr.val.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_123927.centroid.n=40/metrics.train.npz


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_123927.centroid.n=40/metrics.val.npz


+ 377 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_123927.centroid.n=40/training_config.json

@@ -0,0 +1,377 @@
+{
+    "data": {
+        "labels": {
+            "training_labels": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/labels.v002.slp",
+            "validation_labels": null,
+            "validation_fraction": 0.2,
+            "test_labels": null,
+            "split_by_inds": false,
+            "training_inds": [
+                37,
+                3,
+                21,
+                33,
+                18,
+                7,
+                16,
+                34,
+                6,
+                13,
+                24,
+                25,
+                2,
+                5,
+                0,
+                38,
+                39,
+                30,
+                17,
+                4,
+                15,
+                9,
+                11,
+                1,
+                23,
+                29,
+                22,
+                19,
+                28,
+                32,
+                8,
+                10
+            ],
+            "validation_inds": [
+                31,
+                12,
+                20,
+                27,
+                14,
+                36,
+                35,
+                26
+            ],
+            "test_inds": null,
+            "search_path_hints": [
+                "",
+                ""
+            ],
+            "skeletons": [
+                {
+                    "directed": true,
+                    "graph": {
+                        "name": "Skeleton-3",
+                        "num_edges_inserted": 15
+                    },
+                    "links": [
+                        {
+                            "edge_insert_idx": 5,
+                            "key": 0,
+                            "source": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "snout",
+                                    "weight": 1.0
+                                }
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "left_ear",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/reduce": [
+                                    {
+                                        "py/type": "sleap.skeleton.EdgeType"
+                                    },
+                                    {
+                                        "py/tuple": [
+                                            1
+                                        ]
+                                    }
+                                ]
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 6,
+                            "key": 0,
+                            "source": {
+                                "py/id": 1
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "right_ear",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 7,
+                            "key": 0,
+                            "source": {
+                                "py/id": 1
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "centre",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 8,
+                            "key": 0,
+                            "source": {
+                                "py/id": 2
+                            },
+                            "target": {
+                                "py/id": 5
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 9,
+                            "key": 0,
+                            "source": {
+                                "py/id": 4
+                            },
+                            "target": {
+                                "py/id": 5
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 10,
+                            "key": 0,
+                            "source": {
+                                "py/id": 5
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "tail_base",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 14,
+                            "key": 0,
+                            "source": {
+                                "py/id": 6
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "tail_end",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        }
+                    ],
+                    "multigraph": true,
+                    "nodes": [
+                        {
+                            "id": {
+                                "py/id": 1
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 2
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 4
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 5
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 6
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 7
+                            }
+                        }
+                    ]
+                }
+            ]
+        },
+        "preprocessing": {
+            "ensure_rgb": false,
+            "ensure_grayscale": false,
+            "imagenet_mode": null,
+            "input_scaling": 0.5,
+            "pad_to_stride": 32,
+            "resize_and_pad_to_target": true,
+            "target_height": 1028,
+            "target_width": 1300
+        },
+        "instance_cropping": {
+            "center_on_part": null,
+            "crop_size": null,
+            "crop_size_detection_padding": 16
+        }
+    },
+    "model": {
+        "backbone": {
+            "leap": null,
+            "unet": {
+                "stem_stride": null,
+                "max_stride": 32,
+                "output_stride": 2,
+                "filters": 16,
+                "filters_rate": 2.0,
+                "middle_block": true,
+                "up_interpolate": true,
+                "stacks": 1
+            },
+            "hourglass": null,
+            "resnet": null,
+            "pretrained_encoder": null
+        },
+        "heads": {
+            "single_instance": null,
+            "centroid": {
+                "anchor_part": null,
+                "sigma": 2.5,
+                "output_stride": 2,
+                "loss_weight": 1.0,
+                "offset_refinement": false
+            },
+            "centered_instance": null,
+            "multi_instance": null,
+            "multi_class_bottomup": null,
+            "multi_class_topdown": null
+        },
+        "base_checkpoint": null
+    },
+    "optimization": {
+        "preload_data": true,
+        "augmentation_config": {
+            "rotate": true,
+            "rotation_min_angle": -180.0,
+            "rotation_max_angle": 180.0,
+            "translate": false,
+            "translate_min": -5,
+            "translate_max": 5,
+            "scale": false,
+            "scale_min": 0.9,
+            "scale_max": 1.1,
+            "uniform_noise": false,
+            "uniform_noise_min_val": 0.0,
+            "uniform_noise_max_val": 10.0,
+            "gaussian_noise": false,
+            "gaussian_noise_mean": 5.0,
+            "gaussian_noise_stddev": 1.0,
+            "contrast": false,
+            "contrast_min_gamma": 0.5,
+            "contrast_max_gamma": 2.0,
+            "brightness": false,
+            "brightness_min_val": 0.0,
+            "brightness_max_val": 10.0,
+            "random_crop": false,
+            "random_crop_height": 256,
+            "random_crop_width": 256,
+            "random_flip": false,
+            "flip_horizontal": false
+        },
+        "online_shuffling": true,
+        "shuffle_buffer_size": 128,
+        "prefetch": true,
+        "batch_size": 4,
+        "batches_per_epoch": 200,
+        "min_batches_per_epoch": 200,
+        "val_batches_per_epoch": 10,
+        "min_val_batches_per_epoch": 10,
+        "epochs": 50,
+        "optimizer": "adam",
+        "initial_learning_rate": 0.0001,
+        "learning_rate_schedule": {
+            "reduce_on_plateau": true,
+            "reduction_factor": 0.5,
+            "plateau_min_delta": 1e-06,
+            "plateau_patience": 5,
+            "plateau_cooldown": 3,
+            "min_learning_rate": 1e-08
+        },
+        "hard_keypoint_mining": {
+            "online_mining": false,
+            "hard_to_easy_ratio": 2.0,
+            "min_hard_keypoints": 2,
+            "max_hard_keypoints": null,
+            "loss_scale": 5.0
+        },
+        "early_stopping": {
+            "stop_training_on_plateau": true,
+            "plateau_min_delta": 1e-08,
+            "plateau_patience": 20
+        }
+    },
+    "outputs": {
+        "save_outputs": true,
+        "run_name": "231110_123927.centroid.n=40",
+        "run_name_prefix": "",
+        "run_name_suffix": "",
+        "runs_folder": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/models",
+        "tags": [
+            ""
+        ],
+        "save_visualizations": true,
+        "delete_viz_images": true,
+        "zip_outputs": false,
+        "log_to_csv": true,
+        "checkpointing": {
+            "initial_model": false,
+            "best_model": true,
+            "every_epoch": false,
+            "latest_model": false,
+            "final_model": false
+        },
+        "tensorboard": {
+            "write_logs": false,
+            "loss_frequency": "epoch",
+            "architecture_graph": false,
+            "profile_graph": false,
+            "visualizations": true
+        },
+        "zmq": {
+            "subscribe_to_controller": true,
+            "controller_address": "tcp://127.0.0.1:9000",
+            "controller_polling_timeout": 10,
+            "publish_updates": true,
+            "publish_address": "tcp://127.0.0.1:9001"
+        }
+    },
+    "name": "",
+    "description": "",
+    "sleap_version": "1.3.1",
+    "filename": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/models/231110_123927.centroid.n=40/training_config.json"
+}

+ 9 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_123927.centroid.n=40/training_log.csv

@@ -0,0 +1,9 @@
+epoch,loss,lr,val_loss
+0,0.0002184439799748361,1e-04,0.00021230740821920335
+1,0.0002080504928017035,1e-04,0.00020668406796175987
+2,0.00020308996317908168,1e-04,0.0001967951247934252
+3,0.00019262460409663618,1e-04,0.00018863013247027993
+4,0.000185396580491215,1e-04,0.00018487665511202067
+5,0.0001779084705049172,1e-04,0.0001854134170571342
+6,0.000169299470144324,1e-04,0.00019603493274189532
+7,0.00015556513972114772,1e-04,0.00018980806635227054

+ 1 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_130255.centered_instance.n=40/best_model.h5

@@ -0,0 +1 @@
+/annex/objects/MD5-s211231488--005bad023d5660f1f40b04ca7440abea

+ 197 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_130255.centered_instance.n=40/initial_config.json

@@ -0,0 +1,197 @@
+{
+    "data": {
+        "labels": {
+            "training_labels": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/labels.v001.slp",
+            "validation_labels": null,
+            "validation_fraction": 0.2,
+            "test_labels": null,
+            "split_by_inds": false,
+            "training_inds": [
+                11,
+                13,
+                8,
+                16,
+                2,
+                1,
+                12,
+                17,
+                6,
+                4,
+                14,
+                10,
+                3,
+                15,
+                7,
+                5
+            ],
+            "validation_inds": [
+                19,
+                9,
+                18,
+                0
+            ],
+            "test_inds": null,
+            "search_path_hints": [
+                ""
+            ],
+            "skeletons": []
+        },
+        "preprocessing": {
+            "ensure_rgb": false,
+            "ensure_grayscale": false,
+            "imagenet_mode": null,
+            "input_scaling": 1.0,
+            "pad_to_stride": 1,
+            "resize_and_pad_to_target": true,
+            "target_height": 1028,
+            "target_width": 1300
+        },
+        "instance_cropping": {
+            "center_on_part": null,
+            "crop_size": 256,
+            "crop_size_detection_padding": 16
+        }
+    },
+    "model": {
+        "backbone": {
+            "leap": null,
+            "unet": {
+                "stem_stride": null,
+                "max_stride": 32,
+                "output_stride": 4,
+                "filters": 24,
+                "filters_rate": 2.0,
+                "middle_block": true,
+                "up_interpolate": true,
+                "stacks": 1
+            },
+            "hourglass": null,
+            "resnet": null,
+            "pretrained_encoder": null
+        },
+        "heads": {
+            "single_instance": null,
+            "centroid": null,
+            "centered_instance": {
+                "anchor_part": null,
+                "part_names": [
+                    "snout",
+                    "left_ear",
+                    "right_ear",
+                    "centre",
+                    "tail_base",
+                    "tail_end"
+                ],
+                "sigma": 2.5,
+                "output_stride": 4,
+                "loss_weight": 1.0,
+                "offset_refinement": false
+            },
+            "multi_instance": null,
+            "multi_class_bottomup": null,
+            "multi_class_topdown": null
+        },
+        "base_checkpoint": null
+    },
+    "optimization": {
+        "preload_data": true,
+        "augmentation_config": {
+            "rotate": true,
+            "rotation_min_angle": -180.0,
+            "rotation_max_angle": 180.0,
+            "translate": false,
+            "translate_min": -5,
+            "translate_max": 5,
+            "scale": false,
+            "scale_min": 0.9,
+            "scale_max": 1.1,
+            "uniform_noise": false,
+            "uniform_noise_min_val": 0.0,
+            "uniform_noise_max_val": 10.0,
+            "gaussian_noise": false,
+            "gaussian_noise_mean": 5.0,
+            "gaussian_noise_stddev": 1.0,
+            "contrast": false,
+            "contrast_min_gamma": 0.5,
+            "contrast_max_gamma": 2.0,
+            "brightness": false,
+            "brightness_min_val": 0.0,
+            "brightness_max_val": 10.0,
+            "random_crop": false,
+            "random_crop_height": 256,
+            "random_crop_width": 256,
+            "random_flip": false,
+            "flip_horizontal": false
+        },
+        "online_shuffling": true,
+        "shuffle_buffer_size": 128,
+        "prefetch": true,
+        "batch_size": 4,
+        "batches_per_epoch": 200,
+        "min_batches_per_epoch": 200,
+        "val_batches_per_epoch": 10,
+        "min_val_batches_per_epoch": 10,
+        "epochs": 50,
+        "optimizer": "adam",
+        "initial_learning_rate": 0.0001,
+        "learning_rate_schedule": {
+            "reduce_on_plateau": true,
+            "reduction_factor": 0.5,
+            "plateau_min_delta": 1e-06,
+            "plateau_patience": 5,
+            "plateau_cooldown": 3,
+            "min_learning_rate": 1e-08
+        },
+        "hard_keypoint_mining": {
+            "online_mining": false,
+            "hard_to_easy_ratio": 2.0,
+            "min_hard_keypoints": 2,
+            "max_hard_keypoints": null,
+            "loss_scale": 5.0
+        },
+        "early_stopping": {
+            "stop_training_on_plateau": true,
+            "plateau_min_delta": 1e-08,
+            "plateau_patience": 10
+        }
+    },
+    "outputs": {
+        "save_outputs": true,
+        "run_name": "231110_130255.centered_instance.n=40",
+        "run_name_prefix": "",
+        "run_name_suffix": "",
+        "runs_folder": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/models",
+        "tags": [
+            ""
+        ],
+        "save_visualizations": true,
+        "delete_viz_images": true,
+        "zip_outputs": false,
+        "log_to_csv": true,
+        "checkpointing": {
+            "initial_model": false,
+            "best_model": true,
+            "every_epoch": false,
+            "latest_model": false,
+            "final_model": false
+        },
+        "tensorboard": {
+            "write_logs": false,
+            "loss_frequency": "epoch",
+            "architecture_graph": false,
+            "profile_graph": false,
+            "visualizations": true
+        },
+        "zmq": {
+            "subscribe_to_controller": true,
+            "controller_address": "tcp://127.0.0.1:9000",
+            "controller_polling_timeout": 10,
+            "publish_updates": true,
+            "publish_address": "tcp://127.0.0.1:9001"
+        }
+    },
+    "name": "",
+    "description": "",
+    "sleap_version": "1.3.1",
+    "filename": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/models/231110_130255.centered_instance.n=40/initial_config.json"
+}

BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_130255.centered_instance.n=40/labels_gt.train.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_130255.centered_instance.n=40/labels_gt.val.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_130255.centered_instance.n=40/labels_pr.train.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_130255.centered_instance.n=40/labels_pr.val.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_130255.centered_instance.n=40/metrics.train.npz


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_130255.centered_instance.n=40/metrics.val.npz


+ 385 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_130255.centered_instance.n=40/training_config.json

@@ -0,0 +1,385 @@
+{
+    "data": {
+        "labels": {
+            "training_labels": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/labels.v002.slp",
+            "validation_labels": null,
+            "validation_fraction": 0.2,
+            "test_labels": null,
+            "split_by_inds": false,
+            "training_inds": [
+                24,
+                39,
+                26,
+                14,
+                15,
+                18,
+                30,
+                12,
+                7,
+                31,
+                5,
+                4,
+                27,
+                17,
+                6,
+                0,
+                2,
+                3,
+                11,
+                8,
+                16,
+                32,
+                38,
+                22,
+                36,
+                28,
+                34,
+                21,
+                37,
+                1,
+                19,
+                33
+            ],
+            "validation_inds": [
+                9,
+                25,
+                35,
+                13,
+                10,
+                20,
+                29,
+                23
+            ],
+            "test_inds": null,
+            "search_path_hints": [
+                "",
+                ""
+            ],
+            "skeletons": [
+                {
+                    "directed": true,
+                    "graph": {
+                        "name": "Skeleton-3",
+                        "num_edges_inserted": 15
+                    },
+                    "links": [
+                        {
+                            "edge_insert_idx": 5,
+                            "key": 0,
+                            "source": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "snout",
+                                    "weight": 1.0
+                                }
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "left_ear",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/reduce": [
+                                    {
+                                        "py/type": "sleap.skeleton.EdgeType"
+                                    },
+                                    {
+                                        "py/tuple": [
+                                            1
+                                        ]
+                                    }
+                                ]
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 6,
+                            "key": 0,
+                            "source": {
+                                "py/id": 1
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "right_ear",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 7,
+                            "key": 0,
+                            "source": {
+                                "py/id": 1
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "centre",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 8,
+                            "key": 0,
+                            "source": {
+                                "py/id": 2
+                            },
+                            "target": {
+                                "py/id": 5
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 9,
+                            "key": 0,
+                            "source": {
+                                "py/id": 4
+                            },
+                            "target": {
+                                "py/id": 5
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 10,
+                            "key": 0,
+                            "source": {
+                                "py/id": 5
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "tail_base",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 14,
+                            "key": 0,
+                            "source": {
+                                "py/id": 6
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "tail_end",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        }
+                    ],
+                    "multigraph": true,
+                    "nodes": [
+                        {
+                            "id": {
+                                "py/id": 1
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 2
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 4
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 5
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 6
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 7
+                            }
+                        }
+                    ]
+                }
+            ]
+        },
+        "preprocessing": {
+            "ensure_rgb": false,
+            "ensure_grayscale": false,
+            "imagenet_mode": null,
+            "input_scaling": 1.0,
+            "pad_to_stride": 1,
+            "resize_and_pad_to_target": true,
+            "target_height": 1028,
+            "target_width": 1300
+        },
+        "instance_cropping": {
+            "center_on_part": null,
+            "crop_size": 256,
+            "crop_size_detection_padding": 16
+        }
+    },
+    "model": {
+        "backbone": {
+            "leap": null,
+            "unet": {
+                "stem_stride": null,
+                "max_stride": 32,
+                "output_stride": 4,
+                "filters": 24,
+                "filters_rate": 2.0,
+                "middle_block": true,
+                "up_interpolate": true,
+                "stacks": 1
+            },
+            "hourglass": null,
+            "resnet": null,
+            "pretrained_encoder": null
+        },
+        "heads": {
+            "single_instance": null,
+            "centroid": null,
+            "centered_instance": {
+                "anchor_part": null,
+                "part_names": [
+                    "snout",
+                    "left_ear",
+                    "right_ear",
+                    "centre",
+                    "tail_base",
+                    "tail_end"
+                ],
+                "sigma": 2.5,
+                "output_stride": 4,
+                "loss_weight": 1.0,
+                "offset_refinement": false
+            },
+            "multi_instance": null,
+            "multi_class_bottomup": null,
+            "multi_class_topdown": null
+        },
+        "base_checkpoint": null
+    },
+    "optimization": {
+        "preload_data": true,
+        "augmentation_config": {
+            "rotate": true,
+            "rotation_min_angle": -180.0,
+            "rotation_max_angle": 180.0,
+            "translate": false,
+            "translate_min": -5,
+            "translate_max": 5,
+            "scale": false,
+            "scale_min": 0.9,
+            "scale_max": 1.1,
+            "uniform_noise": false,
+            "uniform_noise_min_val": 0.0,
+            "uniform_noise_max_val": 10.0,
+            "gaussian_noise": false,
+            "gaussian_noise_mean": 5.0,
+            "gaussian_noise_stddev": 1.0,
+            "contrast": false,
+            "contrast_min_gamma": 0.5,
+            "contrast_max_gamma": 2.0,
+            "brightness": false,
+            "brightness_min_val": 0.0,
+            "brightness_max_val": 10.0,
+            "random_crop": false,
+            "random_crop_height": 256,
+            "random_crop_width": 256,
+            "random_flip": false,
+            "flip_horizontal": false
+        },
+        "online_shuffling": true,
+        "shuffle_buffer_size": 128,
+        "prefetch": true,
+        "batch_size": 4,
+        "batches_per_epoch": 200,
+        "min_batches_per_epoch": 200,
+        "val_batches_per_epoch": 10,
+        "min_val_batches_per_epoch": 10,
+        "epochs": 50,
+        "optimizer": "adam",
+        "initial_learning_rate": 0.0001,
+        "learning_rate_schedule": {
+            "reduce_on_plateau": true,
+            "reduction_factor": 0.5,
+            "plateau_min_delta": 1e-06,
+            "plateau_patience": 5,
+            "plateau_cooldown": 3,
+            "min_learning_rate": 1e-08
+        },
+        "hard_keypoint_mining": {
+            "online_mining": false,
+            "hard_to_easy_ratio": 2.0,
+            "min_hard_keypoints": 2,
+            "max_hard_keypoints": null,
+            "loss_scale": 5.0
+        },
+        "early_stopping": {
+            "stop_training_on_plateau": true,
+            "plateau_min_delta": 1e-08,
+            "plateau_patience": 10
+        }
+    },
+    "outputs": {
+        "save_outputs": true,
+        "run_name": "231110_130255.centered_instance.n=40",
+        "run_name_prefix": "",
+        "run_name_suffix": "",
+        "runs_folder": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/models",
+        "tags": [
+            ""
+        ],
+        "save_visualizations": true,
+        "delete_viz_images": true,
+        "zip_outputs": false,
+        "log_to_csv": true,
+        "checkpointing": {
+            "initial_model": false,
+            "best_model": true,
+            "every_epoch": false,
+            "latest_model": false,
+            "final_model": false
+        },
+        "tensorboard": {
+            "write_logs": false,
+            "loss_frequency": "epoch",
+            "architecture_graph": false,
+            "profile_graph": false,
+            "visualizations": true
+        },
+        "zmq": {
+            "subscribe_to_controller": true,
+            "controller_address": "tcp://127.0.0.1:9000",
+            "controller_polling_timeout": 10,
+            "publish_updates": true,
+            "publish_address": "tcp://127.0.0.1:9001"
+        }
+    },
+    "name": "",
+    "description": "",
+    "sleap_version": "1.3.1",
+    "filename": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/models/231110_130255.centered_instance.n=40/training_config.json"
+}

+ 19 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231110_130255.centered_instance.n=40/training_log.csv

@@ -0,0 +1,19 @@
+epoch,centre,left_ear,loss,lr,right_ear,snout,tail_base,tail_end,val_centre,val_left_ear,val_loss,val_right_ear,val_snout,val_tail_base,val_tail_end
+0,0.002979804063215852,0.0043107145465910435,0.003921978175640106,1e-04,0.004230549558997154,0.003399109933525324,0.0042969705536961555,0.004314721561968327,0.00226763472892344,0.004101933445781469,0.0037430976517498493,0.0041672661900520325,0.003766444278880954,0.004188250284641981,0.00396705511957407
+1,0.0021100095473229885,0.003187708556652069,0.0031836351845413446,1e-04,0.0030923369340598583,0.0028802596498280764,0.003544149687513709,0.004287349991500378,0.0020028012804687023,0.003090983722358942,0.0028753720689564943,0.0025742310099303722,0.003181572537869215,0.002519911853596568,0.0038827303797006607
+2,0.0016216295771300793,0.0016922126524150372,0.0021873267833143473,1e-04,0.001626256387680769,0.002015969716012478,0.0019271263154223561,0.004240764770656824,0.0013398670125752687,0.0018504677573218942,0.0021579102613031864,0.0015032951487228274,0.002626936649903655,0.001675136387348175,0.003951759077608585
+3,0.0011337501928210258,0.0009293855982832611,0.0016049948753789067,1e-04,0.0009108438389375806,0.0013572509633377194,0.0012186941457912326,0.004080043639987707,0.0013101018266752362,0.0018622649367898703,0.002128519583493471,0.0015296601923182607,0.002925099106505513,0.0012973258271813393,0.003846666542813182
+4,0.0009035109542310238,0.0006177047034725547,0.0013175669591873884,1e-04,0.0006022672750987113,0.0010037367464974523,0.0008585434406995773,0.003919636830687523,0.0012705614790320396,0.0015108520165085793,0.0018294885521754622,0.0011055322829633951,0.0023800316266715527,0.0009222301887348294,0.0037877231370657682
+5,0.0007158169173635542,0.0004453578148968518,0.0011125608580186963,1e-04,0.00046436264528892934,0.0007328977226279676,0.0006637117476202548,0.003653219435364008,0.001387168886139989,0.0011942218989133835,0.001712589873932302,0.000848861294798553,0.002300945343449712,0.000731523847207427,0.0038128183223307133
+6,0.0005727697280235589,0.0003432224621064961,0.0009521407773718238,1e-04,0.0003614735323935747,0.0005082495044916868,0.0005371812148950994,0.003389947349205613,0.0013351573143154383,0.0012730350717902184,0.0017913371557369828,0.0007690485799685121,0.0025596560444682837,0.0008016999927349389,0.004009425174444914
+7,0.00045179540757089853,0.00026287627406418324,0.000785349402576685,1e-04,0.00029065849957987666,0.00040246956632472575,0.0004208935715723783,0.002883403329178691,0.0012968669179826975,0.0010161229874938726,0.0015523491892963648,0.0008251190884038806,0.0022827819921076298,0.0006887670024298131,0.0032044369727373123
+8,0.00038349811802618206,0.0002373517636442557,0.0006505070487037301,1e-04,0.00024809749447740614,0.0003417211410123855,0.00034813303500413895,0.0023442411329597235,0.0012895372929051518,0.0012254143366590142,0.0015844047302380204,0.00085700664203614,0.002260286593809724,0.0006764883873984218,0.0031976946629583836
+9,0.00028737334650941193,0.0001920715149026364,0.0005095757078379393,1e-04,0.00019790633814409375,0.0002781792136374861,0.00027185401995666325,0.0018300686497241259,0.0013777766143903136,0.0014693308621644974,0.0016953323502093554,0.0008449973538517952,0.0025889654643833637,0.0006056862184777856,0.003285237355157733
+10,0.0002342347288504243,0.00016732202493585646,0.0003928267688024789,1e-04,0.00017119423137046397,0.00020870081789325923,0.00021105384803377092,0.0013644542777910829,0.0013121424708515406,0.0013084204401820898,0.0016675041988492012,0.0006997301243245602,0.0025518008042126894,0.0006621063803322613,0.003470823634415865
+11,0.00017372766160406172,0.00014227363863028586,0.00029171534697525203,1e-04,0.000141975557198748,0.00017098731768783182,0.00018164295761380345,0.0009396845125593245,0.001339973183348775,0.0012320365058258176,0.0015078686410561204,0.0007717006374150515,0.0022047427482903004,0.0005213133990764618,0.002977444790303707
+12,0.00013867374218534678,0.00012453993258532137,0.00024079832655843347,1e-04,0.00012084146874258295,0.00014987979375291616,0.00014284577628131956,0.0007680098060518503,0.001447547459974885,0.0010828112717717886,0.001493749557994306,0.0006726739229634404,0.0020912443287670612,0.0005436275387182832,0.0031245930586010218
+13,0.00013214163482189178,9.462520392844453e-05,0.0001785927452147007,1e-04,9.137794404523447e-05,0.00011262664338573813,0.00011682161130011082,0.0005239640013314784,0.0016087284311652184,0.0010769537184387445,0.0015766860451549292,0.0007882994832471013,0.002259306376799941,0.0006283824914135039,0.0030984454788267612
+14,0.00010709996422519907,8.431830792687833e-05,0.00014286748773884028,1e-04,7.387798541458324e-05,0.00011469870514702052,0.00010161715908907354,0.00037559273187071085,0.0011376618640497327,0.0009383756550960243,0.0014156718971207738,0.0006467389175668359,0.002375151962041855,0.0004760592128150165,0.002920042257755995
+15,9.247605339623988e-05,7.283804006874561e-05,0.00011474734492367133,1e-04,6.906933413119987e-05,8.699001773493364e-05,8.32467558211647e-05,0.00028386383200995624,0.0012302605900913477,0.0012754257768392563,0.0015043385792523623,0.0007549882866442204,0.0022702733986079693,0.0004481367359403521,0.003046946134418249
+16,8.865206473274156e-05,7.302586163859814e-05,0.00011562379950191826,1e-04,6.454079994000494e-05,9.913800022331998e-05,7.602135883644223e-05,0.00029236474074423313,0.001291008316911757,0.0012746864231303334,0.001473436364904046,0.0007212469354271889,0.002157513052225113,0.0003881196607835591,0.0030080433934926987
+17,6.89509033691138e-05,5.877269359189086e-05,9.204525122186169e-05,1e-04,5.8471527154324576e-05,7.282675505848601e-05,6.208116246853024e-05,0.0002311685384484008,0.0011893732007592916,0.0011226219357922673,0.0014291201950982213,0.000675660849083215,0.002237212611362338,0.00045234718709252775,0.0028975042514503

+ 1 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_101726.centroid.n=60/best_model.h5

@@ -0,0 +1 @@
+/annex/objects/MD5-s94476560--c48036402838fd143683733c09f15b51

+ 210 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_101726.centroid.n=60/initial_config.json

@@ -0,0 +1,210 @@
+{
+    "data": {
+        "labels": {
+            "training_labels": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/labels.v002.slp",
+            "validation_labels": null,
+            "validation_fraction": 0.1,
+            "test_labels": null,
+            "split_by_inds": false,
+            "training_inds": [
+                37,
+                3,
+                21,
+                33,
+                18,
+                7,
+                16,
+                34,
+                6,
+                13,
+                24,
+                25,
+                2,
+                5,
+                0,
+                38,
+                39,
+                30,
+                17,
+                4,
+                15,
+                9,
+                11,
+                1,
+                23,
+                29,
+                22,
+                19,
+                28,
+                32,
+                8,
+                10
+            ],
+            "validation_inds": [
+                31,
+                12,
+                20,
+                27,
+                14,
+                36,
+                35,
+                26
+            ],
+            "test_inds": null,
+            "search_path_hints": [
+                "",
+                ""
+            ],
+            "skeletons": []
+        },
+        "preprocessing": {
+            "ensure_rgb": false,
+            "ensure_grayscale": false,
+            "imagenet_mode": null,
+            "input_scaling": 0.5,
+            "pad_to_stride": 32,
+            "resize_and_pad_to_target": true,
+            "target_height": 1028,
+            "target_width": 1300
+        },
+        "instance_cropping": {
+            "center_on_part": null,
+            "crop_size": null,
+            "crop_size_detection_padding": 16
+        }
+    },
+    "model": {
+        "backbone": {
+            "leap": null,
+            "unet": {
+                "stem_stride": null,
+                "max_stride": 32,
+                "output_stride": 2,
+                "filters": 16,
+                "filters_rate": 2.0,
+                "middle_block": true,
+                "up_interpolate": true,
+                "stacks": 1
+            },
+            "hourglass": null,
+            "resnet": null,
+            "pretrained_encoder": null
+        },
+        "heads": {
+            "single_instance": null,
+            "centroid": {
+                "anchor_part": null,
+                "sigma": 2.5,
+                "output_stride": 2,
+                "loss_weight": 1.0,
+                "offset_refinement": false
+            },
+            "centered_instance": null,
+            "multi_instance": null,
+            "multi_class_bottomup": null,
+            "multi_class_topdown": null
+        },
+        "base_checkpoint": null
+    },
+    "optimization": {
+        "preload_data": true,
+        "augmentation_config": {
+            "rotate": true,
+            "rotation_min_angle": -180.0,
+            "rotation_max_angle": 180.0,
+            "translate": false,
+            "translate_min": -5,
+            "translate_max": 5,
+            "scale": false,
+            "scale_min": 0.9,
+            "scale_max": 1.1,
+            "uniform_noise": false,
+            "uniform_noise_min_val": 0.0,
+            "uniform_noise_max_val": 10.0,
+            "gaussian_noise": false,
+            "gaussian_noise_mean": 5.0,
+            "gaussian_noise_stddev": 1.0,
+            "contrast": false,
+            "contrast_min_gamma": 0.5,
+            "contrast_max_gamma": 2.0,
+            "brightness": false,
+            "brightness_min_val": 0.0,
+            "brightness_max_val": 10.0,
+            "random_crop": false,
+            "random_crop_height": 256,
+            "random_crop_width": 256,
+            "random_flip": false,
+            "flip_horizontal": false
+        },
+        "online_shuffling": true,
+        "shuffle_buffer_size": 128,
+        "prefetch": true,
+        "batch_size": 4,
+        "batches_per_epoch": 200,
+        "min_batches_per_epoch": 200,
+        "val_batches_per_epoch": 10,
+        "min_val_batches_per_epoch": 10,
+        "epochs": 50,
+        "optimizer": "adam",
+        "initial_learning_rate": 0.0001,
+        "learning_rate_schedule": {
+            "reduce_on_plateau": true,
+            "reduction_factor": 0.5,
+            "plateau_min_delta": 1e-06,
+            "plateau_patience": 5,
+            "plateau_cooldown": 3,
+            "min_learning_rate": 1e-08
+        },
+        "hard_keypoint_mining": {
+            "online_mining": false,
+            "hard_to_easy_ratio": 2.0,
+            "min_hard_keypoints": 2,
+            "max_hard_keypoints": null,
+            "loss_scale": 5.0
+        },
+        "early_stopping": {
+            "stop_training_on_plateau": true,
+            "plateau_min_delta": 1e-08,
+            "plateau_patience": 20
+        }
+    },
+    "outputs": {
+        "save_outputs": true,
+        "run_name": "231113_101726.centroid.n=60",
+        "run_name_prefix": "",
+        "run_name_suffix": "",
+        "runs_folder": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/models",
+        "tags": [
+            ""
+        ],
+        "save_visualizations": true,
+        "delete_viz_images": true,
+        "zip_outputs": false,
+        "log_to_csv": true,
+        "checkpointing": {
+            "initial_model": false,
+            "best_model": true,
+            "every_epoch": false,
+            "latest_model": false,
+            "final_model": false
+        },
+        "tensorboard": {
+            "write_logs": false,
+            "loss_frequency": "epoch",
+            "architecture_graph": false,
+            "profile_graph": false,
+            "visualizations": true
+        },
+        "zmq": {
+            "subscribe_to_controller": true,
+            "controller_address": "tcp://127.0.0.1:9000",
+            "controller_polling_timeout": 10,
+            "publish_updates": true,
+            "publish_address": "tcp://127.0.0.1:9001"
+        }
+    },
+    "name": "",
+    "description": "",
+    "sleap_version": "1.3.1",
+    "filename": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/models/231113_101726.centroid.n=60/initial_config.json"
+}

BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_101726.centroid.n=60/labels_gt.train.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_101726.centroid.n=60/labels_gt.val.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_101726.centroid.n=60/labels_pr.train.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_101726.centroid.n=60/labels_pr.val.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_101726.centroid.n=60/metrics.train.npz


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_101726.centroid.n=60/metrics.val.npz


+ 398 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_101726.centroid.n=60/training_config.json

@@ -0,0 +1,398 @@
+{
+    "data": {
+        "labels": {
+            "training_labels": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/labels.v002.slp",
+            "validation_labels": null,
+            "validation_fraction": 0.1,
+            "test_labels": null,
+            "split_by_inds": false,
+            "training_inds": [
+                51,
+                0,
+                2,
+                23,
+                46,
+                39,
+                43,
+                48,
+                24,
+                25,
+                47,
+                8,
+                21,
+                40,
+                56,
+                36,
+                7,
+                52,
+                19,
+                31,
+                5,
+                45,
+                22,
+                20,
+                50,
+                29,
+                6,
+                53,
+                10,
+                54,
+                12,
+                59,
+                9,
+                32,
+                58,
+                34,
+                35,
+                18,
+                44,
+                42,
+                57,
+                55,
+                1,
+                37,
+                14,
+                4,
+                33,
+                13,
+                15,
+                26,
+                38,
+                49,
+                16,
+                11
+            ],
+            "validation_inds": [
+                17,
+                3,
+                28,
+                27,
+                41,
+                30
+            ],
+            "test_inds": null,
+            "search_path_hints": [
+                "",
+                "",
+                ""
+            ],
+            "skeletons": [
+                {
+                    "directed": true,
+                    "graph": {
+                        "name": "Skeleton-3",
+                        "num_edges_inserted": 15
+                    },
+                    "links": [
+                        {
+                            "edge_insert_idx": 5,
+                            "key": 0,
+                            "source": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "snout",
+                                    "weight": 1.0
+                                }
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "left_ear",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/reduce": [
+                                    {
+                                        "py/type": "sleap.skeleton.EdgeType"
+                                    },
+                                    {
+                                        "py/tuple": [
+                                            1
+                                        ]
+                                    }
+                                ]
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 6,
+                            "key": 0,
+                            "source": {
+                                "py/id": 1
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "right_ear",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 7,
+                            "key": 0,
+                            "source": {
+                                "py/id": 1
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "centre",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 8,
+                            "key": 0,
+                            "source": {
+                                "py/id": 2
+                            },
+                            "target": {
+                                "py/id": 5
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 9,
+                            "key": 0,
+                            "source": {
+                                "py/id": 4
+                            },
+                            "target": {
+                                "py/id": 5
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 10,
+                            "key": 0,
+                            "source": {
+                                "py/id": 5
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "tail_base",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 14,
+                            "key": 0,
+                            "source": {
+                                "py/id": 6
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "tail_end",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        }
+                    ],
+                    "multigraph": true,
+                    "nodes": [
+                        {
+                            "id": {
+                                "py/id": 1
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 2
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 4
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 5
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 6
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 7
+                            }
+                        }
+                    ]
+                }
+            ]
+        },
+        "preprocessing": {
+            "ensure_rgb": false,
+            "ensure_grayscale": false,
+            "imagenet_mode": null,
+            "input_scaling": 0.5,
+            "pad_to_stride": 32,
+            "resize_and_pad_to_target": true,
+            "target_height": 1028,
+            "target_width": 1300
+        },
+        "instance_cropping": {
+            "center_on_part": null,
+            "crop_size": null,
+            "crop_size_detection_padding": 16
+        }
+    },
+    "model": {
+        "backbone": {
+            "leap": null,
+            "unet": {
+                "stem_stride": null,
+                "max_stride": 32,
+                "output_stride": 2,
+                "filters": 16,
+                "filters_rate": 2.0,
+                "middle_block": true,
+                "up_interpolate": true,
+                "stacks": 1
+            },
+            "hourglass": null,
+            "resnet": null,
+            "pretrained_encoder": null
+        },
+        "heads": {
+            "single_instance": null,
+            "centroid": {
+                "anchor_part": null,
+                "sigma": 2.5,
+                "output_stride": 2,
+                "loss_weight": 1.0,
+                "offset_refinement": false
+            },
+            "centered_instance": null,
+            "multi_instance": null,
+            "multi_class_bottomup": null,
+            "multi_class_topdown": null
+        },
+        "base_checkpoint": null
+    },
+    "optimization": {
+        "preload_data": true,
+        "augmentation_config": {
+            "rotate": true,
+            "rotation_min_angle": -180.0,
+            "rotation_max_angle": 180.0,
+            "translate": false,
+            "translate_min": -5,
+            "translate_max": 5,
+            "scale": false,
+            "scale_min": 0.9,
+            "scale_max": 1.1,
+            "uniform_noise": false,
+            "uniform_noise_min_val": 0.0,
+            "uniform_noise_max_val": 10.0,
+            "gaussian_noise": false,
+            "gaussian_noise_mean": 5.0,
+            "gaussian_noise_stddev": 1.0,
+            "contrast": false,
+            "contrast_min_gamma": 0.5,
+            "contrast_max_gamma": 2.0,
+            "brightness": false,
+            "brightness_min_val": 0.0,
+            "brightness_max_val": 10.0,
+            "random_crop": false,
+            "random_crop_height": 256,
+            "random_crop_width": 256,
+            "random_flip": false,
+            "flip_horizontal": false
+        },
+        "online_shuffling": true,
+        "shuffle_buffer_size": 128,
+        "prefetch": true,
+        "batch_size": 4,
+        "batches_per_epoch": 200,
+        "min_batches_per_epoch": 200,
+        "val_batches_per_epoch": 10,
+        "min_val_batches_per_epoch": 10,
+        "epochs": 50,
+        "optimizer": "adam",
+        "initial_learning_rate": 0.0001,
+        "learning_rate_schedule": {
+            "reduce_on_plateau": true,
+            "reduction_factor": 0.5,
+            "plateau_min_delta": 1e-06,
+            "plateau_patience": 5,
+            "plateau_cooldown": 3,
+            "min_learning_rate": 1e-08
+        },
+        "hard_keypoint_mining": {
+            "online_mining": false,
+            "hard_to_easy_ratio": 2.0,
+            "min_hard_keypoints": 2,
+            "max_hard_keypoints": null,
+            "loss_scale": 5.0
+        },
+        "early_stopping": {
+            "stop_training_on_plateau": true,
+            "plateau_min_delta": 1e-08,
+            "plateau_patience": 20
+        }
+    },
+    "outputs": {
+        "save_outputs": true,
+        "run_name": "231113_101726.centroid.n=60",
+        "run_name_prefix": "",
+        "run_name_suffix": "",
+        "runs_folder": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/models",
+        "tags": [
+            ""
+        ],
+        "save_visualizations": true,
+        "delete_viz_images": true,
+        "zip_outputs": false,
+        "log_to_csv": true,
+        "checkpointing": {
+            "initial_model": false,
+            "best_model": true,
+            "every_epoch": false,
+            "latest_model": false,
+            "final_model": false
+        },
+        "tensorboard": {
+            "write_logs": false,
+            "loss_frequency": "epoch",
+            "architecture_graph": false,
+            "profile_graph": false,
+            "visualizations": true
+        },
+        "zmq": {
+            "subscribe_to_controller": true,
+            "controller_address": "tcp://127.0.0.1:9000",
+            "controller_polling_timeout": 10,
+            "publish_updates": true,
+            "publish_address": "tcp://127.0.0.1:9001"
+        }
+    },
+    "name": "",
+    "description": "",
+    "sleap_version": "1.3.1",
+    "filename": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/models/231113_101726.centroid.n=60/training_config.json"
+}

+ 22 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_101726.centroid.n=60/training_log.csv

@@ -0,0 +1,22 @@
+epoch,loss,lr,val_loss
+0,0.00021577686129603535,1e-04,0.00021158398885745555
+1,0.00020746946393046528,1e-04,0.0002042750857071951
+2,0.00020128222240600735,1e-04,0.00019955888274125755
+3,0.0001928956771735102,1e-04,0.0001978289074031636
+4,0.0001889788982225582,1e-04,0.00019359862199053168
+5,0.00018360519607085735,1e-04,0.00019654780044220388
+6,0.0001767177163856104,1e-04,0.00020619042334146798
+7,0.00016950452118180692,1e-04,0.0001942328381119296
+8,0.00016382164903916419,1e-04,0.00020408390264492482
+9,0.00016078496992122382,1e-04,0.00019329867791384459
+10,0.000145401616464369,5e-05,0.00018423868459649384
+11,0.0001402578636771068,5e-05,0.00019877628074027598
+12,0.00013453420251607895,5e-05,0.00019188776786904782
+13,0.00013026119268033653,5e-05,0.0001897561742225662
+14,0.00012527384387794882,5e-05,0.00019530860299710184
+15,0.0001215528609463945,5e-05,0.00017430356820113957
+16,0.00011890720634255558,5e-05,0.00019112779409624636
+17,0.00011387977428967133,5e-05,0.00020796006720047444
+18,0.00010992641909979284,5e-05,0.00018773830379359424
+19,0.00010400023893453181,5e-05,0.00018939399160444736
+20,9.93711655610241e-05,5e-05,0.00018209396512247622

+ 1 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_105044.centered_instance.n=60/best_model.h5

@@ -0,0 +1 @@
+/annex/objects/MD5-s211231488--50104aa5c0a73af430561e009f87c495

+ 218 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_105044.centered_instance.n=60/initial_config.json

@@ -0,0 +1,218 @@
+{
+    "data": {
+        "labels": {
+            "training_labels": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/labels.v002.slp",
+            "validation_labels": null,
+            "validation_fraction": 0.1,
+            "test_labels": null,
+            "split_by_inds": false,
+            "training_inds": [
+                24,
+                39,
+                26,
+                14,
+                15,
+                18,
+                30,
+                12,
+                7,
+                31,
+                5,
+                4,
+                27,
+                17,
+                6,
+                0,
+                2,
+                3,
+                11,
+                8,
+                16,
+                32,
+                38,
+                22,
+                36,
+                28,
+                34,
+                21,
+                37,
+                1,
+                19,
+                33
+            ],
+            "validation_inds": [
+                9,
+                25,
+                35,
+                13,
+                10,
+                20,
+                29,
+                23
+            ],
+            "test_inds": null,
+            "search_path_hints": [
+                "",
+                ""
+            ],
+            "skeletons": []
+        },
+        "preprocessing": {
+            "ensure_rgb": false,
+            "ensure_grayscale": false,
+            "imagenet_mode": null,
+            "input_scaling": 1.0,
+            "pad_to_stride": 1,
+            "resize_and_pad_to_target": true,
+            "target_height": 1028,
+            "target_width": 1300
+        },
+        "instance_cropping": {
+            "center_on_part": null,
+            "crop_size": 256,
+            "crop_size_detection_padding": 16
+        }
+    },
+    "model": {
+        "backbone": {
+            "leap": null,
+            "unet": {
+                "stem_stride": null,
+                "max_stride": 32,
+                "output_stride": 4,
+                "filters": 24,
+                "filters_rate": 2.0,
+                "middle_block": true,
+                "up_interpolate": true,
+                "stacks": 1
+            },
+            "hourglass": null,
+            "resnet": null,
+            "pretrained_encoder": null
+        },
+        "heads": {
+            "single_instance": null,
+            "centroid": null,
+            "centered_instance": {
+                "anchor_part": null,
+                "part_names": [
+                    "snout",
+                    "left_ear",
+                    "right_ear",
+                    "centre",
+                    "tail_base",
+                    "tail_end"
+                ],
+                "sigma": 2.5,
+                "output_stride": 4,
+                "loss_weight": 1.0,
+                "offset_refinement": false
+            },
+            "multi_instance": null,
+            "multi_class_bottomup": null,
+            "multi_class_topdown": null
+        },
+        "base_checkpoint": null
+    },
+    "optimization": {
+        "preload_data": true,
+        "augmentation_config": {
+            "rotate": true,
+            "rotation_min_angle": -180.0,
+            "rotation_max_angle": 180.0,
+            "translate": false,
+            "translate_min": -5,
+            "translate_max": 5,
+            "scale": false,
+            "scale_min": 0.9,
+            "scale_max": 1.1,
+            "uniform_noise": false,
+            "uniform_noise_min_val": 0.0,
+            "uniform_noise_max_val": 10.0,
+            "gaussian_noise": false,
+            "gaussian_noise_mean": 5.0,
+            "gaussian_noise_stddev": 1.0,
+            "contrast": false,
+            "contrast_min_gamma": 0.5,
+            "contrast_max_gamma": 2.0,
+            "brightness": false,
+            "brightness_min_val": 0.0,
+            "brightness_max_val": 10.0,
+            "random_crop": false,
+            "random_crop_height": 256,
+            "random_crop_width": 256,
+            "random_flip": false,
+            "flip_horizontal": false
+        },
+        "online_shuffling": true,
+        "shuffle_buffer_size": 128,
+        "prefetch": true,
+        "batch_size": 4,
+        "batches_per_epoch": 200,
+        "min_batches_per_epoch": 200,
+        "val_batches_per_epoch": 10,
+        "min_val_batches_per_epoch": 10,
+        "epochs": 50,
+        "optimizer": "adam",
+        "initial_learning_rate": 0.0001,
+        "learning_rate_schedule": {
+            "reduce_on_plateau": true,
+            "reduction_factor": 0.5,
+            "plateau_min_delta": 1e-06,
+            "plateau_patience": 5,
+            "plateau_cooldown": 3,
+            "min_learning_rate": 1e-08
+        },
+        "hard_keypoint_mining": {
+            "online_mining": false,
+            "hard_to_easy_ratio": 2.0,
+            "min_hard_keypoints": 2,
+            "max_hard_keypoints": null,
+            "loss_scale": 5.0
+        },
+        "early_stopping": {
+            "stop_training_on_plateau": true,
+            "plateau_min_delta": 1e-08,
+            "plateau_patience": 10
+        }
+    },
+    "outputs": {
+        "save_outputs": true,
+        "run_name": "231113_105044.centered_instance.n=60",
+        "run_name_prefix": "",
+        "run_name_suffix": "",
+        "runs_folder": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/models",
+        "tags": [
+            ""
+        ],
+        "save_visualizations": true,
+        "delete_viz_images": true,
+        "zip_outputs": false,
+        "log_to_csv": true,
+        "checkpointing": {
+            "initial_model": false,
+            "best_model": true,
+            "every_epoch": false,
+            "latest_model": false,
+            "final_model": false
+        },
+        "tensorboard": {
+            "write_logs": false,
+            "loss_frequency": "epoch",
+            "architecture_graph": false,
+            "profile_graph": false,
+            "visualizations": true
+        },
+        "zmq": {
+            "subscribe_to_controller": true,
+            "controller_address": "tcp://127.0.0.1:9000",
+            "controller_polling_timeout": 10,
+            "publish_updates": true,
+            "publish_address": "tcp://127.0.0.1:9001"
+        }
+    },
+    "name": "",
+    "description": "",
+    "sleap_version": "1.3.1",
+    "filename": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/models/231113_105044.centered_instance.n=60/initial_config.json"
+}

BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_105044.centered_instance.n=60/labels_gt.train.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_105044.centered_instance.n=60/labels_gt.val.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_105044.centered_instance.n=60/labels_pr.train.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_105044.centered_instance.n=60/labels_pr.val.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_105044.centered_instance.n=60/metrics.train.npz


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_105044.centered_instance.n=60/metrics.val.npz


+ 406 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_105044.centered_instance.n=60/training_config.json

@@ -0,0 +1,406 @@
+{
+    "data": {
+        "labels": {
+            "training_labels": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/labels.v002.slp",
+            "validation_labels": null,
+            "validation_fraction": 0.1,
+            "test_labels": null,
+            "split_by_inds": false,
+            "training_inds": [
+                3,
+                30,
+                5,
+                23,
+                54,
+                8,
+                46,
+                18,
+                21,
+                51,
+                44,
+                41,
+                22,
+                1,
+                16,
+                34,
+                10,
+                26,
+                9,
+                52,
+                6,
+                37,
+                19,
+                43,
+                0,
+                20,
+                11,
+                25,
+                14,
+                57,
+                32,
+                35,
+                49,
+                56,
+                45,
+                33,
+                13,
+                53,
+                38,
+                28,
+                47,
+                12,
+                50,
+                55,
+                59,
+                7,
+                24,
+                31,
+                17,
+                58,
+                4,
+                29,
+                48,
+                15
+            ],
+            "validation_inds": [
+                36,
+                27,
+                2,
+                39,
+                42,
+                40
+            ],
+            "test_inds": null,
+            "search_path_hints": [
+                "",
+                "",
+                ""
+            ],
+            "skeletons": [
+                {
+                    "directed": true,
+                    "graph": {
+                        "name": "Skeleton-3",
+                        "num_edges_inserted": 15
+                    },
+                    "links": [
+                        {
+                            "edge_insert_idx": 5,
+                            "key": 0,
+                            "source": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "snout",
+                                    "weight": 1.0
+                                }
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "left_ear",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/reduce": [
+                                    {
+                                        "py/type": "sleap.skeleton.EdgeType"
+                                    },
+                                    {
+                                        "py/tuple": [
+                                            1
+                                        ]
+                                    }
+                                ]
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 6,
+                            "key": 0,
+                            "source": {
+                                "py/id": 1
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "right_ear",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 7,
+                            "key": 0,
+                            "source": {
+                                "py/id": 1
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "centre",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 8,
+                            "key": 0,
+                            "source": {
+                                "py/id": 2
+                            },
+                            "target": {
+                                "py/id": 5
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 9,
+                            "key": 0,
+                            "source": {
+                                "py/id": 4
+                            },
+                            "target": {
+                                "py/id": 5
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 10,
+                            "key": 0,
+                            "source": {
+                                "py/id": 5
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "tail_base",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        },
+                        {
+                            "edge_insert_idx": 14,
+                            "key": 0,
+                            "source": {
+                                "py/id": 6
+                            },
+                            "target": {
+                                "py/object": "sleap.skeleton.Node",
+                                "py/state": {
+                                    "name": "tail_end",
+                                    "weight": 1.0
+                                }
+                            },
+                            "type": {
+                                "py/id": 3
+                            }
+                        }
+                    ],
+                    "multigraph": true,
+                    "nodes": [
+                        {
+                            "id": {
+                                "py/id": 1
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 2
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 4
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 5
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 6
+                            }
+                        },
+                        {
+                            "id": {
+                                "py/id": 7
+                            }
+                        }
+                    ]
+                }
+            ]
+        },
+        "preprocessing": {
+            "ensure_rgb": false,
+            "ensure_grayscale": false,
+            "imagenet_mode": null,
+            "input_scaling": 1.0,
+            "pad_to_stride": 1,
+            "resize_and_pad_to_target": true,
+            "target_height": 1028,
+            "target_width": 1300
+        },
+        "instance_cropping": {
+            "center_on_part": null,
+            "crop_size": 256,
+            "crop_size_detection_padding": 16
+        }
+    },
+    "model": {
+        "backbone": {
+            "leap": null,
+            "unet": {
+                "stem_stride": null,
+                "max_stride": 32,
+                "output_stride": 4,
+                "filters": 24,
+                "filters_rate": 2.0,
+                "middle_block": true,
+                "up_interpolate": true,
+                "stacks": 1
+            },
+            "hourglass": null,
+            "resnet": null,
+            "pretrained_encoder": null
+        },
+        "heads": {
+            "single_instance": null,
+            "centroid": null,
+            "centered_instance": {
+                "anchor_part": null,
+                "part_names": [
+                    "snout",
+                    "left_ear",
+                    "right_ear",
+                    "centre",
+                    "tail_base",
+                    "tail_end"
+                ],
+                "sigma": 2.5,
+                "output_stride": 4,
+                "loss_weight": 1.0,
+                "offset_refinement": false
+            },
+            "multi_instance": null,
+            "multi_class_bottomup": null,
+            "multi_class_topdown": null
+        },
+        "base_checkpoint": null
+    },
+    "optimization": {
+        "preload_data": true,
+        "augmentation_config": {
+            "rotate": true,
+            "rotation_min_angle": -180.0,
+            "rotation_max_angle": 180.0,
+            "translate": false,
+            "translate_min": -5,
+            "translate_max": 5,
+            "scale": false,
+            "scale_min": 0.9,
+            "scale_max": 1.1,
+            "uniform_noise": false,
+            "uniform_noise_min_val": 0.0,
+            "uniform_noise_max_val": 10.0,
+            "gaussian_noise": false,
+            "gaussian_noise_mean": 5.0,
+            "gaussian_noise_stddev": 1.0,
+            "contrast": false,
+            "contrast_min_gamma": 0.5,
+            "contrast_max_gamma": 2.0,
+            "brightness": false,
+            "brightness_min_val": 0.0,
+            "brightness_max_val": 10.0,
+            "random_crop": false,
+            "random_crop_height": 256,
+            "random_crop_width": 256,
+            "random_flip": false,
+            "flip_horizontal": false
+        },
+        "online_shuffling": true,
+        "shuffle_buffer_size": 128,
+        "prefetch": true,
+        "batch_size": 4,
+        "batches_per_epoch": 200,
+        "min_batches_per_epoch": 200,
+        "val_batches_per_epoch": 10,
+        "min_val_batches_per_epoch": 10,
+        "epochs": 50,
+        "optimizer": "adam",
+        "initial_learning_rate": 0.0001,
+        "learning_rate_schedule": {
+            "reduce_on_plateau": true,
+            "reduction_factor": 0.5,
+            "plateau_min_delta": 1e-06,
+            "plateau_patience": 5,
+            "plateau_cooldown": 3,
+            "min_learning_rate": 1e-08
+        },
+        "hard_keypoint_mining": {
+            "online_mining": false,
+            "hard_to_easy_ratio": 2.0,
+            "min_hard_keypoints": 2,
+            "max_hard_keypoints": null,
+            "loss_scale": 5.0
+        },
+        "early_stopping": {
+            "stop_training_on_plateau": true,
+            "plateau_min_delta": 1e-08,
+            "plateau_patience": 10
+        }
+    },
+    "outputs": {
+        "save_outputs": true,
+        "run_name": "231113_105044.centered_instance.n=60",
+        "run_name_prefix": "",
+        "run_name_suffix": "",
+        "runs_folder": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/models",
+        "tags": [
+            ""
+        ],
+        "save_visualizations": true,
+        "delete_viz_images": true,
+        "zip_outputs": false,
+        "log_to_csv": true,
+        "checkpointing": {
+            "initial_model": false,
+            "best_model": true,
+            "every_epoch": false,
+            "latest_model": false,
+            "final_model": false
+        },
+        "tensorboard": {
+            "write_logs": false,
+            "loss_frequency": "epoch",
+            "architecture_graph": false,
+            "profile_graph": false,
+            "visualizations": true
+        },
+        "zmq": {
+            "subscribe_to_controller": true,
+            "controller_address": "tcp://127.0.0.1:9000",
+            "controller_polling_timeout": 10,
+            "publish_updates": true,
+            "publish_address": "tcp://127.0.0.1:9001"
+        }
+    },
+    "name": "",
+    "description": "",
+    "sleap_version": "1.3.1",
+    "filename": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/models/231113_105044.centered_instance.n=60/training_config.json"
+}

+ 38 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231113_105044.centered_instance.n=60/training_log.csv

@@ -0,0 +1,38 @@
+epoch,centre,left_ear,loss,lr,right_ear,snout,tail_base,tail_end,val_centre,val_left_ear,val_loss,val_right_ear,val_snout,val_tail_base,val_tail_end
+0,0.0028689149767160416,0.004197894129902124,0.003839622251689434,1e-04,0.004166348837316036,0.003335922723636031,0.004259725566953421,0.004208920057862997,0.0027962499298155308,0.0034569636918604374,0.0036219521425664425,0.0037070054095238447,0.0026543312706053257,0.004464910365641117,0.004652251489460468
+1,0.0019250918412581086,0.002986219013109803,0.002994115697219968,1e-04,0.002827445277944207,0.0028242082335054874,0.0032138065434992313,0.004187923856079578,0.0024764027912169695,0.0018740564119070768,0.0025316844694316387,0.0017511077458038926,0.001964026130735874,0.0025179721415042877,0.004606541246175766
+2,0.0014423440443351865,0.001739145489409566,0.0021376260556280613,1e-04,0.0015614278381690383,0.002120974473655224,0.0018064623000100255,0.004155404400080442,0.001919986098073423,0.001144015695899725,0.0018905911128968,0.0008520289557054639,0.0012761354446411133,0.0016648415476083755,0.004486539401113987
+3,0.0011301289778202772,0.0011571381473913789,0.0016586297424510121,1e-04,0.000974000315181911,0.0015232539735734463,0.0010669347830116749,0.0041003236547112465,0.001548189902678132,0.0005947721074335277,0.0015682438388466835,0.000703969388268888,0.0010301051661372185,0.0011171752121299505,0.004415251314640045
+4,0.0009486943017691374,0.0008327771211043,0.0014132520882412791,1e-04,0.0006792660569772124,0.0012128035305067897,0.0008179042488336563,0.003988068550825119,0.0015479024732485414,0.0004317353595979512,0.0014801733195781708,0.0006986908847466111,0.0007840694161131978,0.0010366529459133744,0.004381988197565079
+5,0.0008215762791223824,0.0005864081322215497,0.0012326545547693968,1e-04,0.000501604750752449,0.0009647325496189296,0.0006359974504448473,0.003885607933625579,0.0017548635369166732,0.0003468597133178264,0.0015510815428569913,0.0005608996143564582,0.0010297575499862432,0.0012799982214346528,0.004334110766649246
+6,0.0007580749806948006,0.00047581203398294747,0.0011291271075606346,1e-04,0.00043172744335606694,0.0008278531604446471,0.000524235307238996,0.003757060505449772,0.0015268985880538821,0.00027391096227802336,0.0014511501649394631,0.0006709778681397438,0.0008415172924287617,0.001115203252993524,0.004278392530977726
+7,0.0006752963527105749,0.0003709855955094099,0.001007465529255569,1e-04,0.0003158368926960975,0.0006587630487047136,0.0004302759771235287,0.0035936362110078335,0.001439330168068409,0.00031151625444181263,0.0012437017867341638,0.000561307359021157,0.000639294390566647,0.000875846657436341,0.0036349159199744463
+8,0.0006351085030473769,0.0003250803565606475,0.0009088241495192051,1e-04,0.0002950166817754507,0.0005632932297885418,0.00039676798041909933,0.003237680532038212,0.0014726377557963133,0.00018417871615383774,0.001229977235198021,0.0005934107466600835,0.0006501552416011691,0.0009264112450182438,0.0035530696623027325
+9,0.0005662188050337136,0.0003005312173627317,0.0008204446057789028,1e-04,0.0002435400092508644,0.0004947366542182863,0.00034250575117766857,0.0029751358088105917,0.0015586204826831818,0.00027084769681096077,0.0013865816872566938,0.0006925961351953447,0.0008700041798874736,0.001181823667138815,0.00374559685587883
+10,0.0004786456993315369,0.00023181730648502707,0.0007054405868984759,1e-04,0.00021584992646239698,0.00043277652002871037,0.0002829345175996423,0.0025906208902597427,0.0014377974439412355,0.000149206563946791,0.0009805922163650393,0.00042169453809037805,0.0006473091198131442,0.0008299220353364944,0.002397623611614108
+11,0.00042101595317944884,0.0001979828957701102,0.0005951547645963728,1e-04,0.00017021989333443344,0.00035231225774623454,0.0002467887243255973,0.0021826080046594143,0.0016079384367913008,0.0002105162711814046,0.0010502273216843605,0.000552872777916491,0.0004654466174542904,0.0006421093712560833,0.002822479698807001
+12,0.00038222703733481467,0.00018251611618325114,0.0005298266769386828,1e-04,0.0001448004477424547,0.000299403298413381,0.0002130395150743425,0.0019569722935557365,0.0012128996895626187,0.00013644693535752594,0.0008354570018127561,0.0004101293161511421,0.0004510964790824801,0.0006208732374943793,0.002181296469643712
+13,0.00034291474730707705,0.00015445382450707257,0.0004502242663875222,1e-04,0.00013237404346000403,0.0002867855946533382,0.00018734595505520701,0.0015974714187905192,0.001603778451681137,0.00018149831157643348,0.0010053704027086496,0.00043501341133378446,0.0005426687421277165,0.0007295316318050027,0.002539732027798891
+14,0.00031564681557938457,0.00014436505443882197,0.00040847252239473164,1e-04,0.00012717976642306894,0.0002711223205551505,0.00016930518904700875,0.0014232147950679064,0.001630292972549796,0.00014540014672093093,0.0008599013090133667,0.0004248409532010555,0.00042259838664904237,0.0005606735358014703,0.00197560153901577
+15,0.0002604779147077352,0.00012204459926579148,0.00033959385473281145,1e-04,0.00011156809341628104,0.00023293889535125345,0.000158196926349774,0.001152336597442627,0.0015362536069005728,0.00010756550909718499,0.0008305016672238708,0.0004356847202870995,0.0005353853339329362,0.0005460333195514977,0.0018220876809209585
+16,0.000225773430429399,0.00011122204159619287,0.00028221087995916605,1e-04,9.608618711354211e-05,0.00021406885934993625,0.00013759736611973494,0.0009085176861844957,0.0017353653674945235,0.00015315753989852965,0.0010457560420036316,0.000461916730273515,0.0006283815018832684,0.0005909249885007739,0.002704789862036705
+17,0.00021044346794951707,0.0001108042779378593,0.00025634700432419777,1e-04,9.651704749558121e-05,0.00019368161156307906,0.00014052445476409048,0.0007861109334044158,0.0018258787458762527,0.00016005896031856537,0.0010385626228526235,0.000466145109385252,0.0004735535476356745,0.0004209626349620521,0.002884776797145605
+18,0.00020937576482538134,9.39007950364612e-05,0.00021476982510648668,1e-04,8.222473115893081e-05,0.00017504209245089442,0.00011988908227067441,0.0006081865285523236,0.0015274896286427975,0.00012320568202994764,0.0008055021171458066,0.0003737419028766453,0.00035675097024068236,0.0005667656078003347,0.001885059056803584
+19,0.0001748838258208707,8.318744949065149e-05,0.00017883794498629868,1e-04,7.252849900396541e-05,0.00014335766900330782,0.00011058671225327998,0.000488483696244657,0.0015793876955285668,0.0001274607056984678,0.0008409559959545732,0.00033505610190331936,0.00032934785122051835,0.000561614811886102,0.002112868009135127
+20,0.00014948581520002335,8.234621054725721e-05,0.00016049123951233923,1e-04,7.28108425391838e-05,0.0001410550466971472,0.00010329498036298901,0.00041395489824935794,0.001645207405090332,0.00011453307524789125,0.0007840031757950783,0.0004798948357347399,0.00039669149555265903,0.00039092794759199023,0.0016767646884545684
+21,0.00015425909077748656,8.311145211337134e-05,0.00015959591837599874,1e-04,7.942053343867883e-05,0.00016734161181375384,0.00010571561870165169,0.00036772762541659176,0.001633110106922686,9.777427476365119e-05,0.0007816758006811142,0.0003443102177698165,0.00038705585757270455,0.0004977080971002579,0.0017300962936133146
+22,0.00011236234422540292,6.081675746827386e-05,0.000105150873423554,1e-04,5.8921665186062455e-05,0.00010243070573778823,7.745570474071428e-05,0.00021891809592489153,0.0018799047684296966,0.00010065027890959755,0.0008964149164967239,0.00038441066863015294,0.0004563217516988516,0.0005315301823429763,0.002025671536102891
+23,0.00013513013254851103,5.906254955334589e-05,0.00010671989002730697,1e-04,5.769802737631835e-05,0.00010490383283467963,7.907750841695815e-05,0.00020444743859115988,0.0017048906302079558,8.505586447427049e-05,0.0008673593401908875,0.0003176565223839134,0.0003556643205229193,0.0005294220754876733,0.002211466198787093
+24,0.0001176949153887108,6.0573918744921684e-05,0.0001013589499052614,1e-04,5.535833406611346e-05,0.00012519594747573137,7.045982056297362e-05,0.00017887106514535844,0.0015509899239987135,9.895839320961386e-05,0.00080195686314255,0.0003902521566487849,0.0003960869216825813,0.00046101026237010956,0.0019144434481859207
+25,0.00011070807522628456,5.2430266805458814e-05,9.135839354712516e-05,1e-04,4.9313282943330705e-05,0.0001007961473078467,6.758332892786711e-05,0.0001673191727604717,0.0015077657299116254,0.00012178694305475801,0.0007256615208461881,0.000459696922916919,0.0002734905865509063,0.0004366409848444164,0.0015545879723504186
+26,9.408509504282847e-05,4.293651363695972e-05,7.271274807862937e-05,1e-04,4.248058030498214e-05,6.94752307026647e-05,5.53372738067992e-05,0.00013196177314966917,0.0014393290039151907,9.822486754273996e-05,0.0006888470379635692,0.00036986341001465917,0.00034008134389296174,0.0004325321060605347,0.001453051227144897
+27,9.407047036802396e-05,4.3030522647313774e-05,6.885706534376368e-05,1e-04,3.94923408748582e-05,6.41497244942002e-05,5.1844566769432276e-05,0.00012055469414917752,0.0019079961348325014,9.963798947865143e-05,0.0007637740927748382,0.000341656559612602,0.00042267824755981565,0.0003575164300855249,0.0014531593769788742
+28,8.20405621198006e-05,4.2616553400876e-05,6.661820953013375e-05,1e-04,4.180150062893517e-05,6.066016430850141e-05,5.0395312428008765e-05,0.00012219505151733756,0.001581248245202005,9.740209497977048e-05,0.0007988301804289222,0.00037575175520032644,0.0002800291986204684,0.0004813258128706366,0.00197722390294075
+29,6.780799594707787e-05,3.720818131114356e-05,5.594883987214416e-05,1e-04,3.625695171649568e-05,4.955976692144759e-05,4.495079701882787e-05,9.990928811021149e-05,0.0016676064115017653,8.638822328066453e-05,0.0007580980891361833,0.00036302057560533285,0.0003283394908066839,0.00043058319715783,0.0016726503381505609
+30,6.313294579740614e-05,3.605767778935842e-05,5.5090844398364425e-05,1e-04,3.518231460475363e-05,5.368265192373656e-05,4.213972351863049e-05,0.00010034966544480994,0.001819042139686644,8.765469829086214e-05,0.0008405626867897809,0.00037685135612264276,0.000363118335371837,0.00042648165253922343,0.0019702273420989513
+31,8.223901386372745e-05,4.0348200855078176e-05,6.854441744508222e-05,1e-04,4.0967086533783004e-05,6.02321742917411e-05,4.917864134768024e-05,0.00013830122770741582,0.0018557595321908593,9.470763325225562e-05,0.0007834426942281425,0.00042802919051609933,0.0002644708438310772,0.0003995716688223183,0.0016581176314502954
+32,4.675757008953951e-05,2.7685404347721487e-05,4.643780994229019e-05,5e-05,2.746156678767875e-05,3.802817082032561e-05,3.407201438676566e-05,0.00010462219506734982,0.0017773956060409546,9.652522567193955e-05,0.0008540991693735123,0.0003877611015923321,0.00033279473427683115,0.00042933589429594576,0.0021007824689149857
+33,3.0331164452945814e-05,2.04585594474338e-05,2.859581945813261e-05,5e-05,2.0273559130146168e-05,2.445006975904107e-05,2.273641075589694e-05,5.3325136832427233e-05,0.0015912961680442095,9.024188329931349e-05,0.0008130032802000642,0.0003320401592645794,0.00029509811429306865,0.0004980385419912636,0.002071304013952613
+34,2.318464612471871e-05,1.902483927551657e-05,2.4892680812627077e-05,5e-05,1.8690281649469398e-05,2.112350739480462e-05,2.12844224733999e-05,4.604837522492744e-05,0.001928893500007689,7.025203376542777e-05,0.0007887100800871849,0.0003154841251671314,0.00036447931779548526,0.000342865678248927,0.0017102856654673815
+35,2.0454495825106278e-05,1.7483742340118624e-05,2.311018033651635e-05,5e-05,1.7459526134189218e-05,2.129571294062771e-05,1.927960147440899e-05,4.268803240847774e-05,0.001758776605129242,9.212891745846719e-05,0.0008295042207464576,0.0003558055032044649,0.0002866789582185447,0.0004309517680667341,0.002052683848887682
+36,1.9851155229844153e-05,1.7376514733769e-05,2.1894440578762442e-05,5e-05,1.7052414477802813e-05,1.97574045159854e-05,1.8943506802315824e-05,3.838564589386806e-05,0.0015766877913847566,8.393589814659208e-05,0.0007824223721399903,0.00036337817437015474,0.00035886827390640974,0.00048471830086782575,0.001826944760978222

+ 1 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231121_151642.centroid.n=60/best_model.h5

@@ -0,0 +1 @@
+/annex/objects/MD5-s94476560--bf10cce25ed88f9bc9ddcdd99a2dfd62

+ 231 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231121_151642.centroid.n=60/initial_config.json

@@ -0,0 +1,231 @@
+{
+    "data": {
+        "labels": {
+            "training_labels": "/Users/nsirmpilatze/Data/PoseEstimation/Mouse-EPM/derivatives/behav/SLEAP/labels.v002.slp",
+            "validation_labels": null,
+            "validation_fraction": 0.1,
+            "test_labels": null,
+            "split_by_inds": false,
+            "training_inds": [
+                51,
+                0,
+                2,
+                23,
+                46,
+                39,
+                43,
+                48,
+                24,
+                25,
+                47,
+                8,
+                21,
+                40,
+                56,
+                36,
+                7,
+                52,
+                19,
+                31,
+                5,
+                45,
+                22,
+                20,
+                50,
+                29,
+                6,
+                53,
+                10,
+                54,
+                12,
+                59,
+                9,
+                32,
+                58,
+                34,
+                35,
+                18,
+                44,
+                42,
+                57,
+                55,
+                1,
+                37,
+                14,
+                4,
+                33,
+                13,
+                15,
+                26,
+                38,
+                49,
+                16,
+                11
+            ],
+            "validation_inds": [
+                17,
+                3,
+                28,
+                27,
+                41,
+                30
+            ],
+            "test_inds": null,
+            "search_path_hints": [
+                "",
+                "",
+                ""
+            ],
+            "skeletons": []
+        },
+        "preprocessing": {
+            "ensure_rgb": false,
+            "ensure_grayscale": false,
+            "imagenet_mode": null,
+            "input_scaling": 0.5,
+            "pad_to_stride": 32,
+            "resize_and_pad_to_target": true,
+            "target_height": 1028,
+            "target_width": 1300
+        },
+        "instance_cropping": {
+            "center_on_part": null,
+            "crop_size": null,
+            "crop_size_detection_padding": 16
+        }
+    },
+    "model": {
+        "backbone": {
+            "leap": null,
+            "unet": {
+                "stem_stride": null,
+                "max_stride": 32,
+                "output_stride": 2,
+                "filters": 16,
+                "filters_rate": 2.0,
+                "middle_block": true,
+                "up_interpolate": true,
+                "stacks": 1
+            },
+            "hourglass": null,
+            "resnet": null,
+            "pretrained_encoder": null
+        },
+        "heads": {
+            "single_instance": null,
+            "centroid": {
+                "anchor_part": null,
+                "sigma": 2.5,
+                "output_stride": 2,
+                "loss_weight": 1.0,
+                "offset_refinement": false
+            },
+            "centered_instance": null,
+            "multi_instance": null,
+            "multi_class_bottomup": null,
+            "multi_class_topdown": null
+        },
+        "base_checkpoint": null
+    },
+    "optimization": {
+        "preload_data": true,
+        "augmentation_config": {
+            "rotate": true,
+            "rotation_min_angle": -180.0,
+            "rotation_max_angle": 180.0,
+            "translate": false,
+            "translate_min": -5,
+            "translate_max": 5,
+            "scale": false,
+            "scale_min": 0.9,
+            "scale_max": 1.1,
+            "uniform_noise": false,
+            "uniform_noise_min_val": 0.0,
+            "uniform_noise_max_val": 10.0,
+            "gaussian_noise": false,
+            "gaussian_noise_mean": 5.0,
+            "gaussian_noise_stddev": 1.0,
+            "contrast": false,
+            "contrast_min_gamma": 0.5,
+            "contrast_max_gamma": 2.0,
+            "brightness": false,
+            "brightness_min_val": 0.0,
+            "brightness_max_val": 10.0,
+            "random_crop": false,
+            "random_crop_height": 256,
+            "random_crop_width": 256,
+            "random_flip": false,
+            "flip_horizontal": false
+        },
+        "online_shuffling": true,
+        "shuffle_buffer_size": 128,
+        "prefetch": true,
+        "batch_size": 4,
+        "batches_per_epoch": 200,
+        "min_batches_per_epoch": 200,
+        "val_batches_per_epoch": 10,
+        "min_val_batches_per_epoch": 10,
+        "epochs": 50,
+        "optimizer": "adam",
+        "initial_learning_rate": 0.0001,
+        "learning_rate_schedule": {
+            "reduce_on_plateau": true,
+            "reduction_factor": 0.5,
+            "plateau_min_delta": 1e-06,
+            "plateau_patience": 5,
+            "plateau_cooldown": 3,
+            "min_learning_rate": 1e-08
+        },
+        "hard_keypoint_mining": {
+            "online_mining": false,
+            "hard_to_easy_ratio": 2.0,
+            "min_hard_keypoints": 2,
+            "max_hard_keypoints": null,
+            "loss_scale": 5.0
+        },
+        "early_stopping": {
+            "stop_training_on_plateau": true,
+            "plateau_min_delta": 1e-08,
+            "plateau_patience": 20
+        }
+    },
+    "outputs": {
+        "save_outputs": true,
+        "run_name": "231121_151642.centroid.n=60",
+        "run_name_prefix": "",
+        "run_name_suffix": "",
+        "runs_folder": "/Users/nsirmpilatze/Dropbox (UCL)/NIU/data/behav-analysis-course/mouse-EPM/derivatives/behav/software-SLEAP_project/models",
+        "tags": [
+            ""
+        ],
+        "save_visualizations": true,
+        "delete_viz_images": true,
+        "zip_outputs": false,
+        "log_to_csv": true,
+        "checkpointing": {
+            "initial_model": false,
+            "best_model": true,
+            "every_epoch": false,
+            "latest_model": false,
+            "final_model": false
+        },
+        "tensorboard": {
+            "write_logs": false,
+            "loss_frequency": "epoch",
+            "architecture_graph": false,
+            "profile_graph": false,
+            "visualizations": true
+        },
+        "zmq": {
+            "subscribe_to_controller": true,
+            "controller_address": "tcp://127.0.0.1:9000",
+            "controller_polling_timeout": 10,
+            "publish_updates": true,
+            "publish_address": "tcp://127.0.0.1:9001"
+        }
+    },
+    "name": "",
+    "description": "",
+    "sleap_version": "1.3.1",
+    "filename": "/Users/nsirmpilatze/Dropbox (UCL)/NIU/data/behav-analysis-course/mouse-EPM/derivatives/behav/software-SLEAP_project/models/231121_151642.centroid.n=60/initial_config.json"
+}

BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231121_151642.centroid.n=60/labels_gt.train.slp


BIN
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231121_151642.centroid.n=60/labels_gt.val.slp


+ 0 - 0
mouse-EPM/derivatives/behav/software-SLEAP_project/models/231121_151642.centroid.n=60/labels_pr.train.slp


Some files were not shown because too many files changed in this diff