Merge pull request #14 from WildMeOrg/deer-background-v0

Adding Background model for Deer

.github/workflows/testing.yml

@@ -52,7 +52,16 @@ jobs:
         uses: actions/setup-python@v2
         with:
           python-version: ${{ matrix.python-version }}
-
+
+      - name: Free Disk space
+        run: |
+             sudo swapoff -a
+             sudo rm -f /swapfile
+             sudo rm -rf /opt/hostedtoolcache
+             sudo apt clean
+             docker rmi $(docker image ls -aq)
+             df -h
+             
       - name: Build project
         run: |
           bash run_developer_setup.sh

requirements/build.txt

@@ -1,6 +1,6 @@
 cmake
 ninja
 scikit-build
-setuptools
+setuptools>=61
 setuptools_scm[toml]
 wheel

wbia_cnn/_plugin.py

@@ -1221,6 +1221,13 @@ 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']:
+        LEGACY = False
+        species = 'deer'
+        confidence_thresh = 0.2
+        model = models.BackgroundModel(batch_size=batch_size, data_shape=data_shape)
+        weights_path = grabmodels.ensure_model('background_deer_v0', redownload=False)
+        canvas_key = 1
     elif species in [
         'whale_fin',
         'whale_fin+fin_dorsal',

wbia_cnn/_plugin_grabmodels.py

@@ -112,6 +112,8 @@
     'vggnet_conv': 'pretrained.caffe.vgg.slice_0_32_None.pkl',
     'vggnet_full': 'pretrained.caffe.vgg.pkl',
     'background_sea_turtle_new_v0': 'background.sea_turtle_new.v0.pkl',
+    'background_deer_v0': 'background.deer.v0.pkl'
+
 }