detect.py和train.py的参数解释

内容来自博主:春马与夏

detect.py(检测)

复制代码
weights: 模型权重路径,即指定使用的模型权重文件的路径。
source: 输入图像或视频的路径,即指定输入文件/目录/URL/屏幕/0(webcam)的路径。
data: 数据集配置文件的路径,即指定数据集配置文件的路径。
imgsz: 推理大小(高度,宽度),即用于推理的输入图像大小。
conf_thres: 置信度阈值,即过滤掉小于此值的检测结果。
iou_thres: NMS(非极大值抑制)的 IOU 阈值,即用于去除重叠的检测结果。
max_det: 每张图像的最大检测数,即指定每张图像最多检测多少个目标。
device: 设备类型,即指定使用的设备类型,如 CPU 或 GPU。
view_img: 显示结果,即指定是否在推理过程中显示结果。
save_txt: 保存检测结果到文本文件中,即指定是否将检测结果保存到 .txt 文件中。
save_conf: 将置信度保存在 --save-txt 标签中,即在保存 .txt 文件时同时保存置信度值。
save_crop: 保存检测结果的裁剪图像,即指定是否保存检测结果的裁剪图像。
nosave: 不保存图像或视频,即指定是否在推理期间保存图像或视频。
classes: 按类过滤,即指定要保留的类的列表。
agnostic_nms: 类不可知的 NMS,即指定是否使用类不可知的 NMS。
augment: 扩充推理,即指定是否在推理期间应用数据扩充。
visualize: 可视化特征,即指定是否在推理过程中可视化特征。
update: 更新所有模型,即指定是否在推理期间更新所有模型。
project: 结果保存的项目路径,即指定要将结果保存到的项目路径。
name: 结果保存的名称,即指定要将结果保存为的名称。
exist_ok: 存在的项目/名称是否已存在,即指定是否应将结果保存为现有名称/项目,而不是自动递增。
line_thickness: 边框线条粗细(像素),即指定绘制边框时使用的线条粗细。
hide_labels: 隐藏标签,即指定是否隐藏绘制的标签。
hide_conf: 隐藏置信度,即指定是否隐藏绘制的置信度值。
half: 使用 FP16 半精度推理,即指定是否使用 FP16 半精度推理。
dnn: 使用 OpenCV DNN 进行 ONNX 推理,即指定是否使用 OpenCV DNN 进行 ONNX 推理。
vid_stride: 视频帧率步长,即指定在推理视频时跳过的帧数。

train.py(训练)

复制代码
weights: 模型权重路径,即指定要加载的初始权重文件的路径。
cfg: 模型配置文件的路径,即指定要加载的模型配置文件的路径。
data: 数据集配置文件的路径,即指定要加载的数据集配置文件的路径。
hyp: 超参数配置文件的路径,即指定要加载的超参数配置文件的路径。
epochs: 总训练周期数,即指定要训练的总周期数。
batch-size: 总批次大小,即指定所有 GPU 的总批次大小,如果设置为 -1,则自动调整批次大小。
imgsz: 训练和验证图像大小(像素),即指定用于训练和验证的输入图像大小。
rect: 矩形训练,即指定是否使用矩形训练。
resume: 恢复训练,即指定是否恢复最近的训练。
nosave: 仅保存最终检查点,即指定是否只保存最终检查点。
noval: 仅验证最后一个周期,即指定是否只在最后一个周期进行验证。
noautoanchor: 禁用自动锚点,即指定是否禁用自动锚点。
noplots: 不保存绘图文件,即指定是否不保存绘图文件。
evolve: 进化超参数,即指定要进化的超参数的代数。
bucket: gsutil 存储桶,即指定 gsutil 存储桶的名称。
cache: 图像缓存模式(RAM / 磁盘),即指定要使用的图像缓存模式。
image-weights: 使用加权图像选择进行训练,即指定是否使用加权图像选择进行训练。
device: 设备类型,即指定使用的设备类型,如 CPU 或 GPU。
multi-scale: 变化图像大小,即指定是否在训练期间随机变化图像大小。
single-cls: 将多类数据训练为单类,即指定是否将多类数据训练为单类。
optimizer: 优化器类型,即指定使用的优化器类型,如 SGD、Adam 或 AdamW。
sync-bn: 使用 SyncBatchNorm,即指定是否在 DDP 模式下使用 SyncBatchNorm。
workers: 最大数据加载器工作线程数,即指定每个 DDP 模式下的工作线程数。
project: 项目路径,即指定将训练结果保存到的项目路径。
name: 结果保存的名称,即指定要将结果保存为的名称。
exist-ok: 存在的项目/名称是否已存在,即指定是否应将结果保存为现有名称/项目,而不是自动递增。
quad: 四路数据加载器,即指定是否使用四路数据加载器。
cos-lr: 余弦学习率调度器,即指定是否使用余弦学习率调度器。
label-smoothing: 标签平滑度,即指定标签平滑度的 epsilon 值。
patience: EarlyStopping 模型训练的耐心周期数,即指定在多少个周期内没有提高性能后停止训练。
freeze: 冻结层,即指定要冻结的层。
save-period: 每 x 个周期保存一次检查点,即指定是否每 x 个周期保存一次检查点(如果设置为 -1,则禁用此功能)。
seed: 全局训练种子,即指定用于训练的全局种子。
local_rank: 自动 DDP 多 GPU 参数,不要修改。
entity: 实体名,即
相关推荐
Dev7z28 分钟前
基于YOLOv11的独居老人摔倒识别设计与实现(数据集+UI界面+训练代码+数据分析)
yolo
智驱力人工智能1 小时前
无人机河道漂浮物检测 从项目构建到价值闭环的系统工程 无人机河道垃圾识别 农村河道漂浮物智能清理方案 无人机辅助河道清洁预警
opencv·算法·安全·yolo·目标检测·无人机·边缘计算
Norach2 小时前
Ubuntu升级opencv版本至4.9.0
linux·经验分享·opencv·yolo·ubuntu·dnn
wfeqhfxz25887823 小时前
基于YOLOv10n的热带海洋蝴蝶鱼物种识别与分类系统_P3456数据集训练_1
yolo·分类·数据挖掘
码上掘金5 小时前
基于YOLO与MobileSAM的路面缺陷智能检测系统
yolo
Wulida00999112 小时前
军用装备视觉识别与分类_yolov10n-PST模型详解
yolo·目标跟踪·分类
FL162386312912 小时前
[C#][winform]基于yolov11的淡水鱼种类检测识别系统C#源码+onnx模型+评估指标曲线+精美GUI界面
人工智能·yolo·目标跟踪
音沐mu.15 小时前
【36】行人红绿灯数据集(有v5/v8模型)/YOLO行人红绿灯检测
yolo·目标检测·数据集·行人红绿灯数据集·行人红绿灯检测·闯红灯检测
Wulida00999118 小时前
【目标检测】基于改进YOLOv13-C3k2-DWR的铲斗定位系统研究
人工智能·yolo·目标检测
Coding茶水间19 小时前
基于深度学习的水下海洋生物检测系统演示与介绍(YOLOv12/v11/v8/v5模型+Pyqt5界面+训练代码+数据集)
图像处理·人工智能·深度学习·yolo·目标检测·机器学习·计算机视觉