ANSCENTER/ANSCORE
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix AMD by changing from GetTensorData<T>() to GetTensorMutableData<T>()

Browse Source
This commit is contained in:
Tuan Nghia Nguyen
2026-04-28 13:25:02 +10:00
parent f4b74c837e
commit dcf974c35c
18 changed files with 359 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
modules/ANSODEngine/ANSYOLO12OD.h
View File

@@ -49,8 +49,11 @@ namespace ANSCENTER {
std::vector<Object> detect(const cv::Mat& image, float confThreshold = 0.4f, float iouThreshold = 0.45f);
//cv::Mat preprocess(const cv::Mat& image, float*& blob, std::vector<int64_t>& inputTensorShape);
cv::Mat preprocess(const cv::Mat& image, std::vector<float>& blob, std::vector<int64_t>& inputTensorShape);
// outputTensors is non-const because GetTensorMutableData()
// requires a non-const Ort::Value receiver — see ANSONNXCL.h
// for full note.
std::vector<Object> postprocess(const cv::Size& originalImageSize, const cv::Size& resizedImageShape,
const std::vector<Ort::Value>& outputTensors,
std::vector<Ort::Value>& outputTensors,
float confThreshold, float iouThreshold);
private: