title: Primate Detection V2 | |
emoji: π | |
colorFrom: gray | |
colorTo: purple | |
sdk: gradio | |
sdk_version: 4.23.0 | |
app_file: app.py | |
pinned: false | |
license: apache-2.0 | |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference | |
# 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_V2 | |
``` | |
2. Install the requirements using conda and pip. | |
``` shell | |
conda create -n primate python=3.12.7 | |
conda activate primate | |
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 load the interface. | |
``` shell | |
python app.py | |
``` |