Skip to content

Commit 2f93ac7

Browse files
authored
Update main_yolov5.py
1 parent 6d5007f commit 2f93ac7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

main_yolov5.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ def detect(self, srcimg):
102102

103103
if __name__ == "__main__":
104104
parser = argparse.ArgumentParser()
105-
parser.add_argument("--imgpath", type=str, default='/home/wangbo/Desktop/data/yolo/yolov34-cpp-opencv-dnn-master/bus.jpg', help="image path")
105+
parser.add_argument("--imgpath", type=str, default='bus.jpg', help="image path")
106106
parser.add_argument('--net_type', default='yolov5s', choices=['yolov5s', 'yolov5l', 'yolov5m', 'yolov5x'])
107107
parser.add_argument('--confThreshold', default=0.5, type=float, help='class confidence')
108108
parser.add_argument('--nmsThreshold', default=0.5, type=float, help='nms iou thresh')
@@ -118,4 +118,4 @@ def detect(self, srcimg):
118118
cv2.namedWindow(winName, cv2.WINDOW_NORMAL)
119119
cv2.imshow(winName, srcimg)
120120
cv2.waitKey(0)
121-
cv2.destroyAllWindows()
121+
cv2.destroyAllWindows()

0 commit comments

Comments
 (0)