HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

onnxruntime cpp yolo-cls fp16 fix (#9412)

Browse source
This commit is contained in:
Myyura 2024-03-31 05:58:57 +08:00 • committed by GitHub
parent 03d0ffd9f5
commit 1ca14a9ad0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 14 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
examples/YOLOv8-ONNXRuntime-CPP/inference.h
View file

@ -29,6 +29,7 @@ enum MODEL_TYPE
//FLOAT16 MODEL
YOLO_DETECT_V8_HALF = 4,
YOLO_POSE_V8_HALF = 5,
YOLO_CLS_HALF = 6
};