YOLO 推理模块

参数说明

  • weights: 指定测试模型,默认为
  • source: 测试数据来源,支持图片、视频、 YouTubeRTSP
  • imgsz: 测试图片大小,一般默认
  • conf-thres: 置信度,大于指定数值就认为检测成功
  • iou-thres: 交并比
  • view-img: 边检测边输出效果
  • save-txt: 是否保留预测坐标,默认 False
  • view_img: 是否实时显示结果
  • project: 保存结果目录
  • name: 输出文件夹名

使用

  • 单模型单数据源
  • 多模型多数据源
  • 多模型单数据源

单模型多数据源

使用同一个模型检测指定目录下的多个文件夹(可指定图片后缀)--source images/**

缺点是只能是输出到同一个文件夹。