|
@@ -238,9 +238,9 @@ char* YOLO_V8::TensorProcess(clock_t& starttime_1, cv::Mat& iImg, N& blob, std::
|
|
rawData = cv::Mat(signalResultNum, strideNum, CV_16F, output);
|
|
rawData = cv::Mat(signalResultNum, strideNum, CV_16F, output);
|
|
rawData.convertTo(rawData, CV_32F);
|
|
rawData.convertTo(rawData, CV_32F);
|
|
}
|
|
}
|
|
- //Note:
|
|
|
|
- //ultralytics add transpose operator to the output of yolov8 model.which make yolov8/v5/v7 has same shape
|
|
|
|
- //https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8n.pt
|
|
|
|
|
|
+ // Note:
|
|
|
|
+ // ultralytics add transpose operator to the output of yolov8 model.which make yolov8/v5/v7 has same shape
|
|
|
|
+ // https://github.com/ultralytics/assets/releases/download/v8.3.0/yolov8n.pt
|
|
rawData = rawData.t();
|
|
rawData = rawData.t();
|
|
|
|
|
|
float* data = (float*)rawData.data;
|
|
float* data = (float*)rawData.data;
|