detect.py中参数解释
default默认值
def parse_opt():
parser = argparse.ArgumentParser()
parser.add_argument(‘–weights’, nargs=’+’, type=str, default=ROOT / ‘yolov5s.pt’, help=’model path(s)’)
parser.add_argument(‘–source’, type=str, default=ROOT / ‘data/images’, help=’file/dir/URL/glob, 0 for webcam’)
parser.add_argument(‘–imgsz’, ‘–img’, ‘–img-size’, nargs=’+’, type=int, default=[640], help=’inference size h,w’)
parser.add_argument(‘–conf-thres’, type=float, default=0.25, help=’confidence threshold’)
#conf-thres表示置信度，default=0.25当有超过25%的可能时，显示预测框
parser.add_argument(‘–iou-thres’, type=float, default=0.45, help=’NMS IoU threshold’)
#iou-thres表示交并比，控制单个目标框的多少，default=0.45表示iou小于该阈值时当作两个框处理，default=1框与框完全重合才能合并，结果中会有多个框出现。default=0表示只要框与框有交集部分就可以合并，故结果中没有重合的框。
parser.add_argument(‘–max-det’, type=int, default=1000, help=’maximum detections per image’)
parser.add_argument(‘–device’, default=”, help=’cuda device, i.e. 0 or 0,1,2,3 or cpu’)#设备
parser.add_argument(‘–view-img’, action=’store_true’, help=’show results’)
#弹出显示结果,具体操作如下
parser.add_argument(‘–save-txt’, action=’store_true’, help=’save results to .txt’)
#保存txt格式的结果
parser.add_argument(‘–save-conf’, action=’store_true’, help=’save confidences in –save-txt labels’) #保存置信度txt
parser.add_argument(‘–save-crop’, action=’store_true’, help=’save cropped prediction boxes’)
parser.add_argument(‘–nosave’, action=’store_true’, help=’do not save images/videos’)#不保存
parser.add_argument(‘–classes’, nargs=’+’, type=int, help=’filter by class: –classes 0, or –classes 0 2 3′)
#指定预测结果只显示person标签或balabala
parser.add_argument(‘–agnostic-nms’, action=’store_true’, help=’class-agnostic NMS’)
parser.add_argument(‘–augment’, action=’store_true’, help=’augmented inference’)#增强（效果不明显）
parser.add_argument(‘–visualize’, action=’store_true’, help=’visualize features’)
parser.add_argument(‘–update’, action=’store_true’, help=’update all models’)
parser.add_argument(‘–project’, default=ROOT / ‘runs/detect’, help=’save results to project/name’) #结果保存位置
parser.add_argument(‘–name’, default=’exp’, help=’save results to project/name’) #保存结果文件名
parser.add_argument(‘–exist-ok’, action=’store_true’, help=’existing project/name ok, do not increment’)
#如果设置这个参数，exp文件夹不会增加结果保存在指定（exp）文件夹下
parser.add_argument(‘–line-thickness’, default=3, type=int, help=’bounding box thickness (pixels)’)
parser.add_argument(‘–hide-labels’, default=False, action=’store_true’, help=’hide labels’)
parser.add_argument(‘–hide-conf’, default=False, action=’store_true’, help=’hide confidences’)
parser.add_argument(‘–half’, action=’store_true’, help=’use FP16 half-precision inference’)
parser.add_argument(‘–dnn’, action=’store_true’, help=’use OpenCV DNN for ONNX inference’)
opt = parser.parse_args()
opt.imgsz = 2 if len(opt.imgsz) == 1 else 1 # expand
print_args(FILE.stem, opt)
return opt
补充：opt打断点可以看到参数状态
参考视频：如何利用yolov5进行预测
Original: https://blog.csdn.net/weixin_47381639/article/details/123401881
Author: 我有个想法~不，你不想
Title: yolov5 detect文件参数解释（部分）
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/709073/
转载文章受原作者版权保护。转载请注明原作者出处！