Merge pull request #285 from WildMeOrg/whaleshark-v3

whaleshark detector and effnet labeler is added
WildMeOrg · Aug 6, 2024 · e250749 · e250749
2 parents 7e8122b + 9916891
commit e250749
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/wbia/algo/detect/efficientnet.py b/wbia/algo/detect/efficientnet.py
@@ -32,7 +32,8 @@
     'trout_effnet_v0': 'https://wildbookiarepository.azureedge.net/models/labeler_trout_effnet.v0.zip',
     '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',
-    'grouper_effnet_v0': 'https://wildbookiarepository.azureedge.net/models/labeler_grouper_effnet.v0.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
@@ -113,7 +113,8 @@
     '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',
-    'grouper_v1': 'https://wildbookiarepository.azureedge.net/models/detect.lightnet.grouper.v1.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'
 }