Merge pull request #282 from WildMeOrg/grouper_v0_detect

Grouper Detector
WildMeOrg · Jul 8, 2024 · a115f1d · a115f1d
2 parents a5d511c + 90f5831
commit a115f1d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/wbia/algo/detect/efficientnet.py b/wbia/algo/detect/efficientnet.py
@@ -32,6 +32,7 @@
     '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'
 }
 
 

diff --git a/wbia/algo/detect/lightnet.py b/wbia/algo/detect/lightnet.py
@@ -112,7 +112,8 @@
     '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_v0': 'https://wildbookiarepository.azureedge.net/models/detect.lightnet.grouper.v0.py'
 }