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:
1 maintainer
Metadata
1 maintainer
Release files for dethub 0.1.2
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| dethub-0.1.2.tar.gz | 9.4 kB | Details |
Release files / dethub-0.1.2.tar.gz
| Download URL | dethub-0.1.2.tar.gz |
|---|---|
| Size | 9.4 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
ff3378b86e9ed33ee23e8d0275aca86f4db812097783f3c942f7dfbc4fa440bf
|
|
BLAKE2b-256 checksum How to use checksums |
13b44a3e14b9c3dec05328eed9f1cf42f86d20fa48035199649e8597fd3d1a58
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/4.0.1 CPython/3.9.12
|
