Fix model optimisation
This commit is contained in:
@@ -213,6 +213,12 @@ namespace ANSCENTER {
|
||||
// ONNX Runtime inference engine
|
||||
std::unique_ptr<ONNXYOLO> m_ortEngine;
|
||||
|
||||
// DML device-lost recovery: when DirectML's GPU device is removed
|
||||
// (HRESULT 887A0005), the session is permanently broken. We detect
|
||||
// this once, attempt a CPU-fallback recreation, and suppress further
|
||||
// error-log flooding.
|
||||
bool _dmlDeviceLost{ false };
|
||||
|
||||
// Internal detection pipeline
|
||||
std::vector<Object> DetectObjects(const cv::Mat& inputImage,
|
||||
const std::string& camera_id);
|
||||
|
||||
Reference in New Issue
Block a user