A Python-based project for detecting objects in images and videos using popular deep learning frameworks.
- Detects multiple object classes in images and videos
- Easy-to-use command-line interface
- Supports model customization and retraining
- Outputs annotated images with bounding boxes
- Python 3.7+
- PyTorch or TensorFlow
- OpenCV
- NumPy
Install dependencies:
pip install -r requirements.txt
Detect objects in an image:
python detect.py --image path/to/image.jpg
Detect objects in a video:
python detect.py --video path/to/video.mp4
- Edit
config.yaml
to set model parameters and detection thresholds.
To train a custom model, follow the instructions in TRAINING.md
.
Contributions are welcome! Please open issues or submit pull requests.
This project is licensed under the MIT License.