https://github.com/aslafy-z/yolo-object-detection-cli
https://github.com/aslafy-z/yolo-object-detection-cli
Last synced: 3 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/aslafy-z/yolo-object-detection-cli
- Owner: aslafy-z
- License: mit
- Created: 2024-11-06T16:25:19.000Z (7 months ago)
- Default Branch: main
- Last Pushed: 2025-02-13T18:57:22.000Z (4 months ago)
- Last Synced: 2025-02-13T19:39:27.371Z (4 months ago)
- Language: Python
- Size: 370 KB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 14
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# yolo-object-detection-cli
## Usage
### Docker
Pull dependencies
```shell
git lfs clone https://github.com/aslafy-z/yolo-object-detection-samples ./samples
docker pull ghcr.io/aslafy-z/yolo-object-detection-cli:main
```
Run with CPU support:
```shell
docker run -it --rm \
-v $PWD/samples/data:/samples \
-p 8000:8000 \
ghcr.io/aslafy-z/yolo-object-detection-cli:main \
--source=/samples/shop.mp4
```
Run with GPU support:
```shell
docker run -it --rm \
-v $PWD/samples/data:/samples \
-p 8000:8000 \
--ipc=host --gpus=all --runtime=nvidia \
ghcr.io/aslafy-z/yolo-object-detection-cli:main \
--source=/samples/shop.mp4 \
--device=cuda:0
```
### System
```shell
sudo apt update && sudo apt install -y ffmpeg libsm6 libxext6
git clone https://github.com/aslafy-z/yolo-object-detection-cli && cd ./yolo-object-detection-cli
git lfs clone https://github.com/aslafy-z/yolo-object-detection-samples ./samples
python -m virtualenv venv
source ./venv/bin/activate
pip install -r requirements.txt
python main.py --source=samples/data/shop.mp4
python main.py --source=rtsp://user:[email protected]/Src/MediaInput/stream_1
```
## Development
### Lock dependencies
```shell
uv lock
uv pip compile pyproject.toml -o requirements.txt
```
> Note: Docker image uses a subset of the dependencies, see in `requirements.docker.txt`.
### Run
```shell
git lfs clone https://github.com/aslafy-z/yolo-object-detection-samples ./samples
uv run main.py --source=samples/data/shop.mp4
```