zishuo/nanodet-plus-opencv
1
0
Fork 0
mirror of https://github.com/hpc203/nanodet-plus-opencv.git synced 2025-12-24 11:51:09 +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:04 +08:00
committed by GitHub
parent 4a2ca340de
commit eb46dc46e8
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
opencv/main.cpp
View File

@@ -255,7 +255,8 @@ void NanoDet_Plus::detect(Mat& srcimg)
int main()
{
NanoDet_Plus mynet("onnxmodel/nanodet-plus-m_320.onnx", "onnxmodel/coco.names", 320, 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", 320, 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);
@@ -265,4 +266,4 @@ int main()
imshow(kWinName, srcimg);
waitKey(0);
destroyAllWindows();
}
}