fz-mirrors/MaaAssistantArknights
0
0
Fork 0
mirror of https://github.com/MaaAssistantArknights/MaaAssistantArknights.git synced 2026-07-18 02:10:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore: remove some deprecated api of onnxruntime

Browse Source 
ref: 1e4f600e5e
This commit is contained in:
Horror Proton
2023-04-06 20:07:44 +08:00
parent fe73546128
commit 7f7f64eee5
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/MaaCore/Vision/Battle/BattleOperatorsImageAnalyzer.cpp
View File

@@ -35,13 +35,14 @@ bool asst::BattleOperatorsImageAnalyzer::analyze()
auto& session = OnnxSessions::get_instance().get("operators_det");
Ort::AllocatorWithDefaultOptions allocator;
std::vector<const char*> input_names;
std::vector<const char*> output_names;
input_names.emplace_back(session.GetInputName(0, allocator));
output_names.emplace_back(session.GetOutputName(0, allocator));
std::string input_name = session.GetInputNameAllocated(0, allocator).get();
std::string output_name = session.GetOutputNameAllocated(0, allocator).get();
std::vector input_names = { input_name.c_str() };
std::vector output_names = { output_name.c_str() };
Ort::RunOptions run_options;
auto outout_tensors = session.Run(run_options, input_names.data(), &input_tensor, 1, output_names.data(), 1);
auto outout_tensors = session.Run(run_options, input_names.data(), &input_tensor, input_names.size(),
output_names.data(), output_names.size());
const float* raw_output = outout_tensors[0].GetTensorData<float>();
// output_shape is { 1, 5, 8400 }