From 8a2682b51b42032deffba29e86ab16c89ec3cc82 Mon Sep 17 00:00:00 2001 From: Jason Parham Date: Mon, 20 Sep 2021 12:45:21 -0700 Subject: [PATCH] Add Sperm Whale V0 --- wbia/algo/detect/densenet.py | 1 + wbia/algo/detect/lightnet.py | 1 + wbia/constants.py | 4 ++++ 3 files changed, 6 insertions(+) diff --git a/wbia/algo/detect/densenet.py b/wbia/algo/detect/densenet.py index 7b5b6fa88b..658c70a8b5 100644 --- a/wbia/algo/detect/densenet.py +++ b/wbia/algo/detect/densenet.py @@ -45,6 +45,7 @@ 'spotted_skunk_v0': 'https://wildbookiarepository.azureedge.net/models/labeler.skunk_spotted.v0.zip', 'humpback_dorsal': 'https://wildbookiarepository.azureedge.net/models/labeler.whale_humpback.dorsal.v0.zip', 'orca_v0': 'https://wildbookiarepository.azureedge.net/models/labeler.whale_orca.v0.zip', + 'whale_sperm_v0': 'https://wildbookiarepository.azureedge.net/models/labeler.whale_sperm.v0.zip', 'fins_v0': 'https://wildbookiarepository.azureedge.net/models/labeler.fins.v0.zip', 'fins_v1': 'https://wildbookiarepository.azureedge.net/models/labeler.fins.v1.1.zip', 'wilddog_v0': 'https://wildbookiarepository.azureedge.net/models/labeler.wild_dog.v0.zip', diff --git a/wbia/algo/detect/lightnet.py b/wbia/algo/detect/lightnet.py index 0cd248eae6..25c87a8542 100644 --- a/wbia/algo/detect/lightnet.py +++ b/wbia/algo/detect/lightnet.py @@ -50,6 +50,7 @@ 'hendrik_dorsal': 'https://wildbookiarepository.azureedge.net/models/detect.lightnet.hendrik.dorsal.py', 'humpback_dorsal': 'https://wildbookiarepository.azureedge.net/models/detect.lightnet.whale_humpback.dorsal.v0.py', 'orca_v0': 'https://wildbookiarepository.azureedge.net/models/detect.lightnet.whale_orca.v0.py', + 'whale_sperm_v0': 'https://wildbookiarepository.azureedge.net/models/detect.lightnet.whale_sperm.v0.py', 'fins_v0': 'https://wildbookiarepository.azureedge.net/models/detect.lightnet.fins.v0.py', 'fins_v1_fluke': 'https://wildbookiarepository.azureedge.net/models/detect.lightnet.fins.v1.py', 'fins_v1_dorsal': 'https://wildbookiarepository.azureedge.net/models/detect.lightnet.fins.v1.1.py', diff --git a/wbia/constants.py b/wbia/constants.py index 95ef5e62a8..000325b0c3 100644 --- a/wbia/constants.py +++ b/wbia/constants.py @@ -449,6 +449,8 @@ class TEST_SPECIES(object): # NOQA 'grey_seal_pup', 'harbour_seal', 'hawaiian_monk_seal', + 'whale_sperm', + 'whale_sperm+fluke', # Latin (Flukebook) 'chelonioidea', 'chelonioidea+head', @@ -496,6 +498,8 @@ class TEST_SPECIES(object): # NOQA 'phoca_vitulina', 'halichoerus_grypus', 'monachus_schauinslandi', + 'physeter_macrocephalus', + 'physeter_macrocephalus+fluke', )