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', )