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

4
modules/ANSODEngine/ANSONNXSEG.h
View File

@@ -51,8 +51,10 @@ namespace ANSCENTER {
void warmupModel();
bool Init(const std::string& modelPath, bool useGPU=true, int deviceId = 0);
cv::Mat preprocess(const cv::Mat& image,float*& blobPtr,std::vector<int64_t>& inputTensorShape);
// outputs 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& origSize,const cv::Size& letterboxSize,
const std::vector<Ort::Value>& outputs, const std::string& camera_id);
std::vector<Ort::Value>& outputs, const std::string& camera_id);
std::vector<Object> segment(const cv::Mat& image, const std::string& camera_id);
std::vector<cv::Point2f> maskToPolygon(const cv::Mat& binaryMask,
const cv::Rect& boundingBox,