diff --git a/wbia/algo/detect/efficientnet.py b/wbia/algo/detect/efficientnet.py index 19ce53e3f..76181b8ad 100644 --- 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 index 276019e44..83f173468 100644 --- 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' }