diff --git a/wbia_cnn/_plugin.py b/wbia_cnn/_plugin.py index b9f08f7..bddd162 100644 --- a/wbia_cnn/_plugin.py +++ b/wbia_cnn/_plugin.py @@ -1221,7 +1221,12 @@ def generate_species_background(ibs, chip_list, species=None, nInput=None): model = models.BackgroundModel(batch_size=batch_size, data_shape=data_shape) weights_path = grabmodels.ensure_model('background_scout_v0', redownload=False) canvas_key = 1 - elif species in ['deer']: + elif species in [ + 'deer', + 'deer_cervus_nippon', + 'deer_axis_axis', + 'deer_rucervis_eldii' + ]: LEGACY = False species = 'deer' confidence_thresh = 0.2