UNeedCryDear / yolov8-opencv-onnxruntime-cpp

yolov8 hub,cpp with onnxruntime and opencv
Apache License 2.0
319 stars 56 forks source link

batch推理时候代码需要改那些地方 #17

Closed panbo-bridge closed 1 year ago

panbo-bridge commented 1 year ago

我想使用batch来推理,我需要修改数据输入的格式,输出的处理逻辑吗?

UNeedCryDear commented 1 year ago

opencv的话要改动一些,onnxruntime的话直接调用batch的接口即可 检测调用这个:

https://github.com/UNeedCryDear/yolov8-opencv-onnxruntime-cpp/blob/4b90bf0c3d86d67faf1a3683ccace79163b7c16d/yolov8_onnx.cpp#L179

分割调用这个 https://github.com/UNeedCryDear/yolov8-opencv-onnxruntime-cpp/blob/4b90bf0c3d86d67faf1a3683ccace79163b7c16d/yolov8_seg_onnx.cpp#L207

panbo-bridge commented 1 year ago

我使用onnxruntime batch推理时候,这行代码报异常了mask = mask(temp_rect - Point(left, top)) > mask_threshold;

UNeedCryDear commented 1 year ago

image

panbo-bridge commented 1 year ago

output类型要改吗?我看直接注释取消了编译会报错

panbo-bridge commented 1 year ago

1694166711930_4957E13C-3B92-479f-A5FE-D154A35FB26E

UNeedCryDear commented 1 year ago

换成temp_output啊,你cpp是一点都不会吗,类型不对

panbo-bridge commented 1 year ago

我刚接触cpp,现在可以跑了,之前报错是模型没有设置dynamic=True。感谢感谢!