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: 实体名,即
相关推荐
西西弗Sisyphus4 小时前
YOLO26 图像分割 - 训练的数据加载:取图、在线增强与标签缓存
yolo·yolo26
YOLO_DATA9 小时前
无人机高速公路道路缺陷数据集 道路损伤数据集 公路裂缝识别 AI大疆数据集 10798期
人工智能·深度学习·yolo·机器学习·cnn
YOLO数据集集合1 天前
小目标无人机、飞机、直升机检测数据集 | 小目标检测 无人机检测 空中目标识别 低空安防 反无人机9090期
深度学习·yolo·目标检测·无人机·飞机·小目标·直升机
YOLO数据集集合1 天前
全球无人机航拍树冠覆盖数据集 | 树冠分割 语义分割 无人机航拍 全球森林监测 GeoTIFF 生态恢复9083期
深度学习·yolo·目标检测·无人机·无人机视角·全球森林
Python图像识别1 天前
10-【2027毕设】YOLO11PCB缺陷检测识别系统 - Python完整源码+PyQt5界面+训练模型+数据集
python·深度学习·yolo·毕业设计·毕设
YOLO数据集集合1 天前
无人机树木识别检测数据集 | 树木检测 树种分类 无人机航拍 林业调查9082期
深度学习·yolo·目标检测·无人机·无人机视角·树冠识别·树冠
qq7422349841 天前
YOLO 之外的另一半:RF-DETR 检测实战
yolo
YOLO数据集集合2 天前
无人机低空影像语义分割数据集 | 语义分割 遥感影像 无人机低空 地物分类 Potsdam Vaihingen LoveDA 9080期
人工智能·深度学习·yolo·目标检测·计算机视觉·语义分割·无人机数据集
智购科技自动贩卖机2 天前
自动售货机嵌入式系统时钟同步与时间管理实战:从RTC校准到断网时间保持的工程实践
大数据·linux·数据库·人工智能·yolo
计算机编程-吉哥3 天前
深度学习:我用YOLO11-L做了一个水下垃圾检测系统 对比YOLOv8-L/Faster R-CNN【计算机毕业设计选题推荐】
人工智能·深度学习·yolo·课程设计·计算机毕业设计选题