diff --git a/wbia/algo/detect/efficientnet.py b/wbia/algo/detect/efficientnet.py index 1482ff830..b54b541c3 100644 --- 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 index 276019e44..24fb9bb1d 100644 --- 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' }