DetHub: Object Detection Model Hub
Project description
DetHub: Object Detection Model Hub
Installation
pip install dethub
Yolov5 Object Prediction and Visualization
from dethub.model import Yolov5
from dethub.visualize import show
detection_model = Yolov5(model_path= "yolov5s.pt", device="cpu", confidence_threshold=0.5, image_size=640)
show("highway1.jpg", detection_model)
Torchvision Object Prediction and Visualization
from dethub.model import TorchVision
from dethub.visualize import show
detection_model = TorchVision(model_path= "dethub/models/torchvision/fasterrcnn_resnet50_fpn.pth", device="cpu", confidence_threshold=0.5, image_size=640)
show("highway1.jpg", detection_model)
TensorflowHub Object Prediction and Visualization
from dethub.model import TensorflowHub
from dethub.visualize import show
detection_model = TensorflowHub(model_path= "https://tfhub.dev/tensorflow/efficientdet/d3/1", device="cpu", confidence_threshold=0.5, image_size=640)
show("highway1.jpg", detection_model)
Contributing
Before opening a PR:
- Reformat with black and isort:
black . --config pyproject.toml
isort .
References:
Project details
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.1.2.tar.gz
(9.4 kB
view details)
File details
Details for the file dethub-0.1.2.tar.gz
.
File metadata
- Download URL: dethub-0.1.2.tar.gz
- Upload date:
- Size: 9.4 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.9.12
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | ff3378b86e9ed33ee23e8d0275aca86f4db812097783f3c942f7dfbc4fa440bf |
|
MD5 | 347a7e8dfa308609216500a114101734 |
|
BLAKE2b-256 | 13b44a3e14b9c3dec05328eed9f1cf42f86d20fa48035199649e8597fd3d1a58 |