File size: 1,120 Bytes
cd98553 f581713 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
---
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
```
|