ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

yolov5训练警告反光标检测模型日志

2022-05-21 14:34:04  阅读:250  来源: 互联网

标签:yolov5 False models 768 -- 384 common 日志 光标


yolov5训练警告反光标检测模型日志

 

1、标注数据

 

 

2、整理数据

 

3、训练:修改:myvoc.yaml

 myvoc.yaml  
train: VOC_2022052001/train.txt
val: VOC_2022052001/val.txt

# number of classes
nc: 1

# class names
names: ["warning"]

 

4、开始训练

python train_20220520.py --img-size 640 --batch-size 1 --epochs 300 --data ./data/myvoc.yaml --cfg ./models/yolov5m.yaml --workers 0

 

训练日志
(wind_2021) F:\PytorchProject\yolov5_train_warning>
(wind_2021) F:\PytorchProject\yolov5_train_warning>
(wind_2021) F:\PytorchProject\yolov5_train_warning>python train_20220520.py --img-size 640 --batch-size 1 --epochs 300 --data ./data/myvoc.yaml --cfg ./models/yolov5m.yaml --workers 0
Using torch 1.8.1+cu111 CUDA:0 (NVIDIA GeForce RTX 3080 Laptop GPU, 16383.5MB)


Namespace(adam=False, batch_size=1, bucket='', cache_images=False, cfg='./models/yolov5m.yaml', data='./data/myvoc.yaml', device='', epochs=300, evolve=False, exist_ok=False, global_rank=-1, hyp='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', noautoanchor=False, nosave=False, notest=False, project='runs/train', quad=False, rect=False, resume=False, save_dir='runs\\train\\exp2', single_cls=False, sync_bn=False, total_batch_size=1, weights='yolov5s.pt', workers=0, world_size=1)
Start Tensorboard with "tensorboard --logdir runs/train", view at http://localhost:6006/
Hyperparameters {'lr0': 0.01, 'lrf': 0.2, 'momentum': 0.937, 'weight_decay': 0.0005, '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, 'anchor_t': 4.0, 'fl_gamma': 0.0, 'hsv_h': 0.015, 'hsv_s': 0.7, 'hsv_v': 0.4, 'degrees': 0.0, 'translate': 0.1, 'scale': 0.5, 'shear': 0.0, 'perspective': 0.0, 'flipud': 0.0, 'fliplr': 0.5, 'mosaic': 1.0, 'mixup': 0.0}
Overriding model.yaml nc=80 with nc=1

                 from  n    params  module                                  arguments
  0                -1  1      5280  models.common.Focus                     [3, 48, 3]
  1                -1  1     41664  models.common.Conv                      [48, 96, 3, 2]
  2                -1  1     65280  models.common.C3                        [96, 96, 2]
  3                -1  1    166272  models.common.Conv                      [96, 192, 3, 2]
  4                -1  1    629760  models.common.C3                        [192, 192, 6]
  5                -1  1    664320  models.common.Conv                      [192, 384, 3, 2]
  6                -1  1   2512896  models.common.C3                        [384, 384, 6]
  7                -1  1   2655744  models.common.Conv                      [384, 768, 3, 2]
  8                -1  1   1476864  models.common.SPP                       [768, 768, [5, 9, 13]]
  9                -1  1   4134912  models.common.C3                        [768, 768, 2, False]
 10                -1  1    295680  models.common.Conv                      [768, 384, 1, 1]
 11                -1  1         0  torch.nn.modules.upsampling.Upsample    [None, 2, 'nearest']
 12           [-1, 6]  1         0  models.common.Concat                    [1]
 13                -1  1   1182720  models.common.C3                        [768, 384, 2, False]
 14                -1  1     74112  models.common.Conv                      [384, 192, 1, 1]
 15                -1  1         0  torch.nn.modules.upsampling.Upsample    [None, 2, 'nearest']
 16           [-1, 4]  1         0  models.common.Concat                    [1]
 17                -1  1    296448  models.common.C3                        [384, 192, 2, False]
 18                -1  1    332160  models.common.Conv                      [192, 192, 3, 2]
 19          [-1, 14]  1         0  models.common.Concat                    [1]
 20                -1  1   1035264  models.common.C3                        [384, 384, 2, False]
 21                -1  1   1327872  models.common.Conv                      [384, 384, 3, 2]
 22          [-1, 10]  1         0  models.common.Concat                    [1]
 23                -1  1   4134912  models.common.C3                        [768, 768, 2, False]
 24      [17, 20, 23]  1     24246  models.yolo.Detect                      [1, [[10, 13, 16, 30, 33, 23], [30, 61, 62, 45, 59, 119], [116, 90, 156, 198, 373, 326]], [192, 384, 768]]
Model Summary: 391 layers, 21056406 parameters, 21056406 gradients, 50.4 GFLOPS

Transferred 59/506 items from yolov5s.pt
Scaled weight_decay = 0.0005
Optimizer groups: 86 .bias, 86 conv.weight, 83 other
Scanning 'VOC_2022052001\labels' for images and labels... 341 found, 0 missing, 0 empty, 0 corrupted: 100%|█████████████████████████████████████████████| 341/341 [00:00<00:00, 1891.93it/s]
New cache created: VOC_2022052001\labels.cache
Scanning 'VOC_2022052001\labels' for images and labels... 61 found, 0 missing, 0 empty, 0 corrupted: 100%|████████████████████████████████████████████████| 61/61 [00:00<00:00, 2084.54it/s]
New cache created: VOC_2022052001\labels.cache                                                                                                                       | 0/61 [00:00<?, ?it/s]
                                                                                                                                                                                            Plotting labels... 052001\labels.cache' for images and labels... 61 found, 0 missing, 0 empty, 0 corrupted: 100%|████████████████████████████████████████████████████| 61/61 [00:00<?, ?it/s]
Scanning 'VOC_2022052001\labels.cache' for images and labels... 341 found, 0 missing, 0 empty, 0 corrupted: 100%|█████████████████████████████████████████████████| 341/341 [00:00<?, ?it/s]
Scanning 'VOC_2022052001\labels.cache' for images and labels... 61 found, 0 missing, 0 empty, 0 corrupted: 100%|████████████████████████████████████████████████████| 61/61 [00:00<?, ?it/s]

Analyzing anchors... anchors/target = 5.90, Best Possible Recall (BPR) = 1.0000
Image sizes 640 train, 640 test
Using 0 dataloader workers
Logging results to runs\train\exp2
Starting training for 300 epochs...
Starting training for 300 epochs...

     Epoch   gpu_mem       box       obj       cls     total   targets  img_size
     0/299     1.22G    0.1006   0.03639         0     0.137         2       640: 100%|███████████████████████████████████████████████████████████████████| 341/341 [01:16<00:00,  4.44it/s]
               Class      Images     Targets           P           R      mAP@.5  mAP@.5:.95: 100%|█████████████████████████████████████████████████████████| 61/61 [00:05<00:00, 11.15it/s]
                 all          61         124    5.46e-05     0.00806    1.79e-06    1.79e-07

     Epoch   gpu_mem       box       obj       cls     total   targets  img_size
     1/299      1.2G    0.0903   0.04164         0    0.1319         2       640: 100%|███████████████████████████████████████████████████████████████████| 341/341 [01:10<00:00,  4.81it/s]
               Class      Images     Targets           P           R      mAP@.5  mAP@.5:.95: 100%|█████████████████████████████████████████████████████████| 61/61 [00:04<00:00, 14.48it/s]
                 all          61         124           0           0    3.48e-05    3.48e-06

     Epoch   gpu_mem       box       obj       cls     total   targets  img_size
     2/299      1.2G   0.07705   0.03868         0    0.1157         1       640: 100%|███████████████████████████████████████████████████████████████████| 341/341 [01:11<00:00,  4.75it/s]
               Class      Images     Targets           P           R      mAP@.5  mAP@.5:.95: 100%|█████████████████████████████████████████████████████████| 61/61 [00:04<00:00, 14.37it/s]
                 all          61         124      0.0459      0.0726      0.0135     0.00226

     Epoch   gpu_mem       box       obj       cls     total   targets  img_size
     3/299      1.2G    0.0841   0.03379         0    0.1179         4       640: 100%|███████████████████████████████████████████████████████████████████| 341/341 [01:10<00:00,  4.82it/s]
               Class      Images     Targets           P           R      mAP@.5  mAP@.5:.95: 100%|█████████████████████████████████████████████████████████| 61/61 [00:04<00:00, 14.10it/s]
                 all          61         124           0           0     0.00207     0.00028

  

 

 

 

 

 

 

 

 

 

 

##############################

标签:yolov5,False,models,768,--,384,common,日志,光标
来源: https://www.cnblogs.com/herd/p/16295008.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有