DetHub: Object Detection Model Hub
Project description
DetHub: Object Detection Model Hub
Installation
git clone https://github.com/kadirnar/dethub
cd dethub
pip install -r requirements.txt
Yolov5 Object Prediction and Visualization
run('yolov5', 'dethub/models/yolov5/yolov5n.pt', 'data/highway1.jpg')
Torchvision Object Prediction and Visualization
run("torchvision", "dethub/models/torchvision/fasterrcnn_resnet50_fpn.pth", "data/highway1.jpg")
TfHub Object Prediction and Visualization
run('tensorflow', 'https://tfhub.dev/tensorflow/efficientdet/d3/1', 'data/highway1.jpg')
Contributing
Before opening a PR:
- Reformat with black and isort:
black . --config pyproject.toml
isort .
References:
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
dethub-0.0.2.tar.gz
(5.1 kB
view hashes)
