zishuo/nanodet-plus-opencv
1
0
Fork 0
mirror of https://github.com/hpc203/nanodet-plus-opencv.git synced 2026-04-22 23:17:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update main.cpp

Browse Source
This commit is contained in:
hpc203
2021-12-28 17:47:54 +08:00
committed by GitHub
parent eb46dc46e8
commit d3cba74e53
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
onnxruntime/main.cpp
+5 -4
View File
@@ -270,8 +270,8 @@ void NanoDet_Plus::detect(Mat& srcimg)
auto allocator_info = MemoryInfo::CreateCpu(OrtDeviceAllocator, OrtMemTypeCPU);
Value input_tensor_ = Value::CreateTensor<float>(allocator_info, input_image_.data(), input_image_.size(), input_shape_.data(), input_shape_.size());
// ¿ªÊ¼ÍÆÀí
vector<Value> ort_outputs = ort_session->Run(RunOptions{ nullptr }, &input_names[0], &input_tensor_, 1, output_names.data(), output_names.size()); // ¿ªÊ¼ÍÆÀí
// ¿ªÊ¼ÍÆÀí
vector<Value> ort_outputs = ort_session->Run(RunOptions{ nullptr }, &input_names[0], &input_tensor_, 1, output_names.data(), output_names.size()); // ¿ªÊ¼ÍÆÀí
/////generate proposals
vector<BoxInfo> generate_boxes;
const float* preds = ort_outputs[0].GetTensorMutableData<float>();
@@ -297,7 +297,8 @@ void NanoDet_Plus::detect(Mat& srcimg)
int main()
{
NanoDet_Plus mynet("onnxmodel/nanodet-plus-m_320.onnx", "onnxmodel/coco.names", 0.5, 0.5); /// choice = ["picodet_m_320_coco.onnx", "picodet_m_416_coco.onnx", "picodet_s_320_coco.onnx", "picodet_s_416_coco.onnx"]
NanoDet_Plus mynet("onnxmodel/nanodet-plus-m_320.onnx", "onnxmodel/coco.names", 0.5, 0.5); /// choice = ["onnxmodel/nanodet-plus-m_320.onnx", "onnxmodel/nanodet-plus-m_416.onnx",
"onnxmodel/nanodet-plus-m-1.5x_320.onnx", "onnxmodel/nanodet-plus-m-1.5x_416.onnx"]
string imgpath = "imgs/person.jpg";
Mat srcimg = imread(imgpath);
mynet.detect(srcimg);
@@ -307,4 +308,4 @@ int main()
imshow(kWinName, srcimg);
waitKey(0);
destroyAllWindows();
}
}