切片推理
此代码 Fork obss/sahi, 添加如下功能
- 添加rknn支持
detection_model = AutoDetectionModel.from_pretrained(
model_type='yolov8rknn',
model_path=yolov8_onnx_model_path,
confidence_threshold=0.3,
category_mapping=category_mapping,
device="cpu", # or 'cuda:0'
)
- 添加yolov10 导出 onnx 支持
detection_model = AutoDetectionModel.from_pretrained(
model_type='yolov10onnx',
model_path=yolov8_onnx_model_path,
confidence_threshold=0.3,
category_mapping=category_mapping,
)
- 添加参数 EP_LIST, 使用onnx推理时直接传递
detection_model = AutoDetectionModel.from_pretrained(
model_type='yolov10onnx',
model_path=yolov8_onnx_model_path,
confidence_threshold=0.3,
category_mapping=category_mapping,
ep_list=['CUDAExecutionProvider', 'CPUExecutionProvider']
)
1 maintainer
Metadata
1 maintainer
Release files for sahi-fei 0.11.19
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| sahi_fei-0.11.19.tar.gz | 108.6 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| sahi_fei-0.11.19-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 229.4 kB
Release files / sahi_fei-0.11.19.tar.gz
| Download URL | sahi_fei-0.11.19.tar.gz |
|---|---|
| Size | 108.6 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
bc843efcac03423381a9d219958f1293fe323b6611392f46e78bc02d72ae7599
|
|
BLAKE2b-256 checksum How to use checksums |
e81c7bd522184e3f64b8d312fdca0a0be55e9793031a7a2b4e594a1eb6ed30db
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/5.1.1 CPython/3.8.18
|
Release files / sahi_fei-0.11.19-py3-none-any.whl
| Download URL | sahi_fei-0.11.19-py3-none-any.whl |
|---|---|
| Size | 120.8 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
908682d3e1a3370805b1cec00c0e5c376dd33d8d33ea432c6a4fd261fccd579c
|
|
BLAKE2b-256 checksum How to use checksums |
752a203f32fec90f7e7291e0e32b15f0ff8786d997e96253b224e0e83393892d
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/5.1.1 CPython/3.8.18
|
