[目标检测]–yolov5-train.py参数详解

  • A+
所属分类:专业基础

weigths

1
parser.add_argument('--weights', type=str, default=ROOT / 'yolov5s.pt', help='initial weights path')

指的是训练好的网络模型,用来初始化网络权重

cfg

1
parser.add_argument('--cfg', type=str, default='', help='model.yaml path')

为configuration的缩写,指的是网络结构,一般对应models文件夹下的xxx.yaml文件

data

1
parser.add_argument('--data', type=str, default=ROOT / 'data/coco128.yaml', help='dataset.yaml path')

训练数据路径,一般为data文件夹下的xxx.yaml文件

hyp

1
parser.add_argument('--hyp', type=str, default=ROOT / 'data/hyps/hyp.scratch-low.yaml', help='hyperparameters path')

训练网络的一些超参数设置,(一般用不到)

epochs

1
parser.add_argument('--epochs', type=int, default=300, help='total training epochs')

设置训练的轮数

batch-size

1
parser.add_argument('--batch-size', type=int, default=16, help='total batch size for all GPUs, -1 for autobatch')

每次输出给神经网络的图片数

img-size

1
parser.add_argument('--imgsz', '--img', '--img-size', type=int, default=640, help='train, val image size (pixels)')

rect: 是否采用矩形训练

1
parser.add_argument('--rect', action='store_true', help='rectangular training')

resume: 指定之前训练的网络模型,并继续训练这个模型

1
parser.add_argument('--resume', nargs='?', const=True, default=False, help='resume most recent training')

nosave: 只保留最后一次的网络模型

1
parser.add_argument('--nosave', action='store_true', help='only save final checkpoint')

noval:只在最后一次进行验证

1
parser.add_argument('--noval', action='store_true', help='only validate final epoch')

noautoanchor:是否采用锚点

1
parser.add_argument('--noautoanchor', action='store_true', help='disable AutoAnchor')
weinxin
XCJinggai的微信
这是我的微信二维码,有任何问题请加微信联系我!或发送邮件至lqzhangfengxi@163.com
zfy

发表评论

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen: