|
|
---
|
|
|
license: cc-by-nc-4.0
|
|
|
base_model:
|
|
|
- Ultralytics/YOLOv8
|
|
|
pipeline_tag: image-segmentation
|
|
|
datasets:
|
|
|
- industoai/PackDet
|
|
|
tags:
|
|
|
- object-segmentation
|
|
|
- YOLO
|
|
|
- Package-Detection
|
|
|
library_name: pytorch
|
|
|
---
|
|
|
|
|
|
# Description
|
|
|
This model is a YOLO-based model which is trained on [Package Detection (PackDet)](https://huggingface.co/datasets/industoai/PackDet).
|
|
|
|
|
|
## Code
|
|
|
The complete code can be found [here](https://github.com/industoai/Deep-Package-Detection).
|
|
|
|
|
|
## How to use
|
|
|
One can use the model within his/her code with the following commands:
|
|
|
```shell
|
|
|
from huggingface_hub import hf_hub_download
|
|
|
from ultralytics import YOLO
|
|
|
|
|
|
model_path = hf_hub_download(repo_id="industoai/Package-Detection", filename="model/package_detection.pt")
|
|
|
model = YOLO(model_path)
|
|
|
result = model("path/to/image")
|
|
|
```
|
|
|
|
|
|
## License
|
|
|
This model is licensed under the Attribution–NonCommercial 4.0 International License (CC BY-NC 4.0).
|
|
|
|
