From f2a2ad5b8e4c2228ed6b5843ac8c68b32b57a9c7 Mon Sep 17 00:00:00 2001 From: tsubramanian Date: Sun, 7 Jul 2024 22:32:11 -0400 Subject: [PATCH 1/2] Update efficientnet.py Adding Grouper models --- wbia/algo/detect/efficientnet.py | 1 + 1 file changed, 1 insertion(+) 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' } From 90f583126754b1188ac098233186bee6bbfbf93c Mon Sep 17 00:00:00 2001 From: tsubramanian Date: Sun, 7 Jul 2024 22:36:55 -0400 Subject: [PATCH 2/2] Update lightnet.py Added Grouper --- wbia/algo/detect/lightnet.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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' }