diff --git a/YOLOPv2.cpp b/YOLOPv2.cpp index 35bc929..2ddbc2f 100644 --- a/YOLOPv2.cpp +++ b/YOLOPv2.cpp @@ -269,9 +269,9 @@ Mat YOLOPv2::detect(Mat& frame) if (left > -1 && right > -1) { int mid = (left + right) / 2; for (int k = -5; k <= 5; k++) { - outimg.at(i, mid+k)[0] = 255; - outimg.at(i, mid+k)[1] = 255; - outimg.at(i, mid+k)[2] = 0; + outimg.at(i, mid+k)[0] = 0; + outimg.at(i, mid+k)[1] = 0; + outimg.at(i, mid+k)[2] = 255; } } //(需要考虑的问题 1.双车道3条线 2.拐角处曲线 3.近处显示不全 4.两条线粘连) diff --git a/build/YOLOPv2.o b/build/YOLOPv2.o index 08eafa8..0605643 100644 Binary files a/build/YOLOPv2.o and b/build/YOLOPv2.o differ diff --git a/build/fast-yolopv2 b/build/fast-yolopv2 index 1cf9cf2..37134b4 100755 Binary files a/build/fast-yolopv2 and b/build/fast-yolopv2 differ diff --git a/build/mainwindow.o b/build/mainwindow.o index e40a69d..94dc093 100644 Binary files a/build/mainwindow.o and b/build/mainwindow.o differ diff --git a/build/onnx/yolopv2_736x1280.onnx b/build/onnx/yolopv2_192x320.onnx similarity index 99% rename from build/onnx/yolopv2_736x1280.onnx rename to build/onnx/yolopv2_192x320.onnx index ea31bf7..07d7cc7 100755 Binary files a/build/onnx/yolopv2_736x1280.onnx and b/build/onnx/yolopv2_192x320.onnx differ diff --git a/mainwindow.cpp b/mainwindow.cpp index c8edfe0..8143aba 100644 --- a/mainwindow.cpp +++ b/mainwindow.cpp @@ -59,7 +59,7 @@ void MainWindow::ShowVideo() static const string kWinName = "Deep learning object detection in ONNXRuntime"; namedWindow(kWinName, WINDOW_NORMAL); - cv::VideoCapture cap("/home/wuxianfu/Projects/Fast-YolopV2/build/videos/566a351c29c00924a337e91e85fa7dec.mp4"); + cv::VideoCapture cap("/home/wuxianfu/Projects/Fast-YolopV2/build/videos/test1.mp4"); if (!cap.isOpened()) { std::cerr << "Error opening video stream" << std::endl; return;