Add files via upload

hpc203 · web-flow · commit 58cbf9c8af7f · 2021-01-21T16:48:49.000+08:00
diff --git a/main_yolov5.py b/main_yolov5.py
@@ -102,17 +102,20 @@ def detect(self, srcimg):
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
-    parser.add_argument("--imgpath", type=str, default='bus.jpg', help="image path")
+    parser.add_argument("--imgpath", type=str, default='/home/wangbo/Desktop/data/yolo/yolov34-cpp-opencv-dnn-master/bus.jpg', help="image path")
     parser.add_argument('--net_type', default='yolov5s', choices=['yolov5s', 'yolov5l', 'yolov5m', 'yolov5x'])
+    parser.add_argument('--confThreshold', default=0.5, type=float, help='class confidence')
+    parser.add_argument('--nmsThreshold', default=0.5, type=float, help='nms iou thresh')
+    parser.add_argument('--objThreshold', default=0.5, type=float, help='object confidence')
     args = parser.parse_args()
 
-    yolonet = yolov5(args.net_type)
+    yolonet = yolov5(args.net_type, confThreshold=args.confThreshold, nmsThreshold=args.nmsThreshold, objThreshold=args.objThreshold)
     srcimg = cv2.imread(args.imgpath)
     dets = yolonet.detect(srcimg)
     srcimg = yolonet.postprocess(srcimg, dets)
 
     winName = 'Deep learning object detection in OpenCV'
-    cv2.namedWindow(winName, 0)
+    cv2.namedWindow(winName, cv2.WINDOW_NORMAL)
     cv2.imshow(winName, srcimg)
     cv2.waitKey(0)
     cv2.destroyAllWindows()
