| title: Primate Detection Gradio | |
| emoji: 🐢 | |
| colorFrom: red | |
| colorTo: indigo | |
| sdk: gradio | |
| sdk_version: 4.23.0 | |
| app_file: app.py | |
| pinned: false | |
| license: apache-2.0 | |
| # Primate Detection | |
| ## Setup | |
| 1. Clone this repository through your local terminal by entering: | |
| ``` shell | |
| git lfs install | |
| git clone https://huggingface.co/spaces/caltech-animal-tracking/Primate-Detection-GPU | |
| ``` | |
| 2. Install the requirements using conda and pip. | |
| ``` shell | |
| conda create -n primates | |
| conda activate primates | |
| pip install -r requirements.txt | |
| ``` | |
| 3. Change the environment variables to fit your system. | |
| In ```app.py``` change ```BATCH_SIZE``` and ```CUDA_PATH``` to reflect current CUDA version and the size of your GPU. You can find your CUDA version by navigating to ```/usr/local``` on your computer. You may need to test a few different batch sizes to find one that works for your computer. | |
| 4. (Optional) Install FFmpeg if it's missing on your system by following the directions [here](https://ffmpeg.org/download.html). | |
| 5. Run ```app.py``` to install GroundingDINO locally and load the interface. | |
| ``` shell | |
| python3 app.py | |
| ``` | |