python train.py --img 640 --batch 2 --epochs 5 --data pcb.yaml --weights yolov5s.pt

(base) C:\Users\Administrator>cd I:\01ldzx\YOLO\yolov5\yolov5

(base) C:\Users\Administrator>I:

(base) I:\01ldzx\YOLO\yolov5\yolov5>activate yolo

(yolo) I:\01ldzx\YOLO\yolov5\yolov5>python train.py --img 640 --batch 2 --epochs
5 --data pcb.yaml --weights yolov5s.pt
Using torch 1.7.1 CUDA:0 (GeForce GTX 950, 2048MB)

Namespace(adam=False, batch_size=2, bucket=’’, cache_images=False, cfg=’’, data=
‘pcb.yaml’, device=’’, epochs=5, evolve=False, exist_ok=False, global_rank=-1, h
yp=‘data/hyp.scratch.yaml’, image_weights=False, img_size=[640, 640], local_rank
=-1, log_artifacts=False, log_imgs=16, multi_scale=False, name=‘exp’, noautoanch
or=False, nosave=False, notest=False, project=‘runs/train’, rect=False, resume=F
alse, save_dir=‘runs\train\exp14’, single_cls=False, sync_bn=False, total_batc
h_size=2, weights=‘yolov5s.pt’, workers=1, world_size=1)
Start Tensorboard with “tensorboard --logdir runs/train”, view at http://localho
st:6006/
Hyperparameters {‘lr0’: 0.01, ‘lrf’: 0.2, ‘momentum’: 0.937, ‘weight_decay’: 0.0
005, ‘warmup_epochs’: 3.0, ‘warmup_momentum’: 0.8, ‘warmup_bias_lr’: 0.1, ‘box’:
0.05, ‘cls’: 0.5, ‘cls_pw’: 1.0, ‘obj’: 1.0, ‘obj_pw’: 1.0, ‘iou_t’: 0.2, ‘anch
or_t’: 4.0, ‘fl_gamma’: 0.0, ‘hsv_h’: 0.015, ‘hsv_s’: 0.7, ‘hsv_v’: 0.4, ‘degree
s’: 0.0, ‘translate’: 0.1, ‘scale’: 0.5, ‘shear’: 0.0, ‘perspective’: 0.0, ‘flip
ud’: 0.0, ‘fliplr’: 0.5, ‘mosaic’: 1.0, ‘mixup’: 0.0}
Overriding model.yaml nc=80 with nc=6

             from  n    params  module                                  argu

ments
0 -1 1 3520 models.common.Focus [3,
32, 3]
1 -1 1 18560 models.common.Conv [32,
64, 3, 2]
2 -1 1 19904 models.common.BottleneckCSP [64,
64, 1]
3 -1 1 73984 models.common.Conv [64,
128, 3, 2]
4 -1 1 161152 models.common.BottleneckCSP [128
, 128, 3]
5 -1 1 295424 models.common.Conv [128
, 256, 3, 2]
6 -1 1 641792 models.common.BottleneckCSP [256
, 256, 3]
7 -1 1 1180672 models.common.Conv [256
, 512, 3, 2]
8 -1 1 656896 models.common.SPP [512
, 512, [5, 9, 13]]
9 -1 1 1248768 models.common.BottleneckCSP [512
, 512, 1, False]
10 -1 1 131584 models.common.Conv [512
, 256, 1, 1]
11 -1 1 0 torch.nn.modules.upsampling.Upsample [Non
e, 2, ‘nearest’]
12 [-1, 6] 1 0 models.common.Concat [1]

13 -1 1 378624 models.common.BottleneckCSP [512
, 256, 1, False]
14 -1 1 33024 models.common.Conv [256
, 128, 1, 1]
15 -1 1 0 torch.nn.modules.upsampling.Upsample [Non
e, 2, ‘nearest’]
16 [-1, 4] 1 0 models.common.Concat [1]

17 -1 1 95104 models.common.BottleneckCSP [256
, 128, 1, False]
18 -1 1 147712 models.common.Conv [128
, 128, 3, 2]
19 [-1, 14] 1 0 models.common.Concat [1]

20 -1 1 313088 models.common.BottleneckCSP [256
, 256, 1, False]
21 -1 1 590336 models.common.Conv [256
, 256, 3, 2]
22 [-1, 10] 1 0 models.common.Concat [1]

23 -1 1 1248768 models.common.BottleneckCSP [512
, 512, 1, False]
24 [17, 20, 23] 1 29667 models.yolo.Detect [6,
[[10, 13, 16, 30, 33, 23], [30, 61, 62, 45, 59, 119], [116, 90, 156, 198, 373, 3
26]], [128, 256, 512]]
Model Summary: 283 layers, 7268579 parameters, 7268579 gradients, 16.9 GFLOPS

Transferred 364/370 items from yolov5s.pt
Optimizer groups: 62 .bias, 70 conv.weight, 59 other
Scanning ‘…\pcb\labels\train’ for images and labels… 1035 found, 0 missing,
New cache created: …\pcb\labels\train.cache
Scanning ‘…\pcb\labels\train.cache’ for images and labels… 1035 found, 0 mis
Scanning ‘…\pcb\labels\train.cache’ for images and labels… 1035 found, 0 mis

Analyzing anchors… anchors/target = 6.47, Best Possible Recall (BPR) = 1.0000
Image sizes 640 train, 640 test
Using 1 dataloader workers
Logging results to runs\train\exp14
Starting training for 5 epochs…

 Epoch   gpu_mem       box       obj       cls     total   targets  img_size

   0/4    0.847G   0.06872   0.01596   0.02651    0.1112         1       64

你可能感兴趣的:(python train.py --img 640 --batch 2 --epochs 5 --data pcb.yaml --weights yolov5s.pt)