From 9916891a0c0c9a9c31e8f8b56357d70f446d8436 Mon Sep 17 00:00:00 2001 From: tsubramanian Date: Mon, 5 Aug 2024 16:03:28 -0700 Subject: [PATCH] whaleshark detector and effnet labeler is added --- wbia/algo/detect/efficientnet.py | 3 ++- wbia/algo/detect/lightnet.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/wbia/algo/detect/efficientnet.py b/wbia/algo/detect/efficientnet.py index 76181b8ad..f819a40e7 100644 --- a/wbia/algo/detect/efficientnet.py +++ b/wbia/algo/detect/efficientnet.py @@ -32,7 +32,8 @@ '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' + '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 f0d25f4d0..24fb9bb1d 100644 --- a/wbia/algo/detect/lightnet.py +++ b/wbia/algo/detect/lightnet.py @@ -113,7 +113,8 @@ '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', - 'grouper_v1': 'https://wildbookiarepository.azureedge.net/models/detect.lightnet.grouper.v1.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' }