Merge branch 'main' into add-ms-labeler-flip

WildMeOrg · Aug 14, 2024 · e2bc07f · e2bc07f
2 parents bb521c1 + e250749
commit e2bc07f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/wbia/algo/detect/efficientnet.py b/wbia/algo/detect/efficientnet.py
@@ -33,6 +33,8 @@
     'shark_effnet_v0': 'https://wildbookiarepository.azureedge.net/models/labeler_shark_effnet.v0.zip',
     'msv2_multilabel_effnet_v0': 'https://cthulhu.dyn.wildme.io/public/models/labeler_msv2_multilabel_effnet.v2.zip',
     'msv2_multilabel_flip_effnet_v0': 'https://cthulhu.dyn.wildme.io/public/models/labeler_msv2_multilabel_flip_effnet.v2.zip',
+    'grouper_effnet_v0': 'https://wildbookiarepository.azureedge.net/models/labeler_grouper_effnet.v0.zip',
+    'whaleshark_effnet_v0': 'https://wildbookiarepository.azureedge.net/models/labeler_whaleshark_effnet.v0.zip'
 }
 
 

diff --git a/wbia/algo/detect/lightnet.py b/wbia/algo/detect/lightnet.py
@@ -112,7 +112,9 @@
     'deer_v0': 'https://wildbookiarepository.azureedge.net/models/detect.lightnet.deer.v0.py',
     'trout_v0': 'https://wildbookiarepository.azureedge.net/models/detect.lightnet.trout.v0.py',
     'trout_v1': 'https://wildbookiarepository.azureedge.net/models/detect.lightnet.trout.v1.py',
-    'shark_v0': 'https://wildbookiarepository.azureedge.net/models/detect.lightnet.shark.v0.py'
+    'shark_v0': 'https://wildbookiarepository.azureedge.net/models/detect.lightnet.shark.v0.py',
+    'grouper_v1': 'https://wildbookiarepository.azureedge.net/models/detect.lightnet.grouper.v1.py',
+    'whaleshark_v0': 'https://wildbookiarepository.azureedge.net/models/detect.lightnet.whaleshark.v0.py'
 }