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: 实体名,即
相关推荐
今天炼丹了吗7 小时前
YOLOv11融合[ECCV2024]FADformer中的FFCM模块
yolo
红色的山茶花14 小时前
YOLOv9-0.1部分代码阅读笔记-loss_tal.py
笔记·深度学习·yolo
机器懒得学习17 小时前
基于YOLOv5的智能水域监测系统:从目标检测到自动报告生成
人工智能·yolo·目标检测
AI莫大猫1 天前
(6)YOLOv4算法基本原理以及和YOLOv3 的差异
算法·yolo
KeepThinking!1 天前
YOLO-World:Real-Time Open-Vocabulary Object Detection
人工智能·yolo·目标检测·多模态
前网易架构师-高司机1 天前
游泳溺水识别数据集,对9984张原始图片进行YOLO,COCO JSON, VOC XML 格式的标注,平均识别率在91.7%以上
yolo·溺水·游泳溺水·游泳安全
发呆小天才O.oᯅ1 天前
YOLOv8目标检测——详细记录使用OpenCV的DNN模块进行推理部署C++实现
c++·图像处理·人工智能·opencv·yolo·目标检测·dnn
深度学习lover2 天前
<项目代码>YOLO Visdrone航拍目标识别<目标检测>
python·yolo·目标检测·计算机视觉·visdrone航拍目标识别
深度学习lover2 天前
[项目代码] YOLOv8 遥感航拍飞机和船舶识别 [目标检测]
python·yolo·目标检测·计算机视觉·遥感航拍飞机和船舶识别
学习BigData2 天前
【使用PyQt5和YOLOv11开发电脑屏幕区域的实时分类GUI】——选择检测区域
qt·yolo·分类