Merge pull request #139 from hci-unihd/fix_zenodo_models

Fix incorrect model weights on Zenodo
kreshuklab · Apr 6, 2023 · 83723a1 · 83723a1
2 parents 160f864 + 10d4f83
commit 83723a1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/plantseg/resources/models_zoo.yaml b/plantseg/resources/models_zoo.yaml
@@ -3,7 +3,7 @@
 
 ## Aliases for best performing networks for confocal and light-sheet
 generic_confocal_3D_unet:
-  model_url: https://zenodo.org/record/7768223/files/unet3d-arabidopsis-ovules-confocal-ds2x.pytorch
+  model_url: https://zenodo.org/record/7805434/files/unet3d-arabidopsis-ovules-confocal-ds2x.pytorch
   resolution: [0.235, 0.150, 0.150]
   description: "Unet trained on confocal images of Arabidopsis Ovules on 1/2-resolution in XY with BCEDiceLoss."
 generic_light_sheet_3D_unet:
@@ -17,7 +17,7 @@ confocal_3D_unet_ovules_ds1x:
   resolution: [0.235, 0.075, 0.075]
   description: "Unet trained on confocal images of Arabidopsis Ovules on original resolution with BCEDiceLoss."
 confocal_3D_unet_ovules_ds2x:
-  model_url: https://zenodo.org/record/7768223/files/unet3d-arabidopsis-ovules-confocal-ds2x.pytorch
+  model_url: https://zenodo.org/record/7805434/files/unet3d-arabidopsis-ovules-confocal-ds2x.pytorch
   resolution: [0.235, 0.150, 0.150]
   description: "Unet trained on confocal images of Arabidopsis Ovules on 1/2-resolution in XY with BCEDiceLoss."
 confocal_3D_unet_ovules_ds3x: