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

8
modules/ANSODEngine/ANSONNXOBB.cpp
View File

@@ -1089,7 +1089,7 @@ namespace ANSCENTER {
std::vector<Object> ANSONNXOBB::postprocess(
const cv::Size& originalImageSize,
const cv::Size& resizedImageShape,
const std::vector<Ort::Value>& outputTensors,
std::vector<Ort::Value>& outputTensors,
int topk,
const std::string& camera_id)
{
@@ -1103,8 +1103,10 @@ namespace ANSCENTER {
return {};
}
// Extract output tensor data and shape [1, num_features, num_detections]
const float* rawOutput = outputTensors[0].GetTensorData<float>();
// Extract output tensor data and shape [1, num_features, num_detections].
// GetTensorMutableData (not GetTensorData) on DML — const variant
// hangs on AMD after ~8 calls. Read-only despite the name.
const float* rawOutput = outputTensors[0].GetTensorMutableData<float>();
const std::vector<int64_t> outputShape = outputTensors[0].GetTensorTypeAndShapeInfo().GetShape();
if (outputShape.size() < 3) {