- A+
所属分类：专业基础
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文件
1 | parser.add_argument('--data', type=str, default=ROOT / 'data/coco128.yaml', help='dataset.yaml path') |
训练数据路径，一般为data文件夹下的xxx.yaml文件
1 | parser.add_argument('--hyp', type=str, default=ROOT / 'data/hyps/hyp.scratch-low.yaml', help='hyperparameters path') |
训练网络的一些超参数设置，(一般用不到)
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') |
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') |
XCJinggai的微信
这是我的微信二维码，有任何问题请加微信联系我！或发送邮件至lqzhangfengxi@163.com
![[半导体电子学]反偏漏电流](https://zhangfengxi.top/wp-content/themes/begin/timthumb.php?src=https://zhangfengxi.top/wp-content/uploads/2022/10/Reverse-Current.png&w=280&h=210&a=&zc=1)
