Merge pull request #281 from WildMeOrg/add-ms-labeler-flip

adds ms labeler trained with flip augmentation
WildMeOrg · Aug 14, 2024 · dbb17ba · dbb17ba
2 parents e250749 + e2bc07f
commit dbb17ba
Showing 1 changed file with 1 addition and 0 deletions.
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',
+    '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'
 }