yolov5模型剪枝报错
收藏回复举报
yolov5模型剪枝报错
t('forum.solved') 已解决
发表于2024-04-12 15:52:20
0 查看

使用的应用案例yolov5剪枝训练报错找不到images and labes

train: Scanning 'VOC/images/train.cache' images and labels... 0 found, 4952 missing, 0 empty, 0 corrupted: 100%|██████████████████████| 4952/4952 [00:00<?, ?it/s] 

Traceback (most recent call last): 

我是按照案例链接下载的数据集,然后按步骤直接稀疏训练,还需要自己标图吗?或者修改文件夹名?具体在哪改呢

│  .dockerignore
│  .gitignore
│  allbn.txt
│  detect.py
│  detect_prune.py
│  Dockerfile
│  finetune_pruned.py
│  finetune_pruned2.py
│  finetune_prune_conv.py
│  getweight.py
│  hubconf.py
│  map.txt
│  modelparse.py
│  model_change.txt
│  prune.py
│  prune2.py
│  prune_conv.py
│  prune_convbn.py
│  prune_utils.py
│  README.md
│  READMEYolov.md
│  reprune.py
│  requirements.txt
│  showbn.py
│  test.py
│  testprune.py
│  train.py
│  train_prune_sparsity.py
│  train_sparsity.py
│  train_sparsity2.py
│  train_sparsity3.py
│  train_sparsity4.py
│  tutorial.ipynb
│  VOCtestnoimgs_06-Nov-2007.tar
│  VOCtest_06-Nov-2007.tar
│  VOCtrainval_06-Nov-2007.tar
│  yolov5prune.md
│  yolov5s.pt
│  批处理获取文件名.bat
│  文件树.txt
│  
├─data
│  │  argoverse_hd.yaml
│  │  coco.yaml
│  │  coco128.yaml
│  │  GlobalWheat2020.yaml
│  │  hyp.finetune.yaml
│  │  hyp.finetune_objects365.yaml
│  │  hyp.scratch.yaml
│  │  objects365.yaml
│  │  VisDrone.yaml
│  │  voc.yaml
│  │  
│  ├─images
│  │      bus.jpg
│  │      zidane.jpg
│  │      
│  ├─scripts
│  │      get_argoverse_hd.sh
│  │      get_coco.sh
│  │      get_coco128.sh
│  │      get_voc.sh
│  │      
│  └─VOC
│      ├─images
│      │  ├─train
│      │  └─val
│      └─labels
│          ├─train
│          └─val
├─img
│      Screenshot from 2021-05-23 20-19-08.png
│      Screenshot from 2021-05-23 20-19-30.png
│      Screenshot from 2021-05-24 22-17-16.png
│      Screenshot from 2021-05-25 00-26-23.png
│      Screenshot from 2021-05-25 00-26-45.png
│      Screenshot from 2021-05-25 00-28-15.png
│      Screenshot from 2021-05-25 00-28-52.png
│      Screenshot from 2021-05-27 22-20-33.png
│      Screenshot from 2021-05-28 08-33-25.png
│      Screenshot from 2021-05-31 22-29-12.png
│      Screenshot from 2021-05-31 22-30-21.png
│      Screenshot from 2021-06-05 00-06-27.png
│      Selection_007.png
│      
├─models
│  │  common.py
│  │  experimental.py
│  │  export.py
│  │  modul.txt
│  │  pruned_common.py
│  │  yolo.py
│  │  yolov5l.yaml
│  │  yolov5m.yaml
│  │  yolov5s-opt.param
│  │  yolov5s.yaml
│  │  yolov5slite.yaml
│  │  yolov5sprune.yaml
│  │  yolov5x.yaml
│  │  __init__.py
│  │  
│  ├─hub
│  │      anchors.yaml
│  │      yolov3-spp.yaml
│  │      yolov3-tiny.yaml
│  │      yolov3.yaml
│  │      yolov5-fpn.yaml
│  │      yolov5-p2.yaml
│  │      yolov5-p6.yaml
│  │      yolov5-p7.yaml
│  │      yolov5-panet.yaml
│  │      yolov5l6.yaml
│  │      yolov5m6.yaml
│  │      yolov5s-transformer.yaml
│  │      yolov5s6.yaml
│  │      yolov5x6.yaml
│  │      
│  └─__pycache__
│          common.cpython-39.pyc
│          experimental.cpython-39.pyc
│          pruned_common.cpython-39.pyc
│          yolo.cpython-39.pyc
│          __init__.cpython-39.pyc
│          
├─runs
│  └─train
│      └─exp
│          │  events.out.tfevents.1712906547.ubuntu.3025.0
│          │  hyp.yaml
│          │  opt.yaml
│          │  
│          └─weights
├─utils
│  │  activations.py
│  │  autoanchor.py
│  │  datasets.py
│  │  general.py
│  │  google_utils.py
│  │  loss.py
│  │  metrics.py
│  │  plots.py
│  │  torch_utils.py
│  │  __init__.py
│  │  
│  ├─aws
│  │      mime.sh
│  │      resume.py
│  │      userdata.sh
│  │      __init__.py
│  │      
│  ├─flask_rest_api
│  │      example_request.py
│  │      README.md
│  │      restapi.py
│  │      
│  ├─google_app_engine
│  │      additional_requirements.txt
│  │      app.yaml
│  │      Dockerfile
│  │      
│  ├─wandb_logging
│  │  │  log_dataset.py
│  │  │  wandb_utils.py
│  │  │  __init__.py
│  │  │  
│  │  └─__pycache__
│  │          wandb_utils.cpython-39.pyc
│  │          __init__.cpython-39.pyc
│  │          
│  └─__pycache__
│          autoanchor.cpython-39.pyc
│          datasets.cpython-39.pyc
│          general.cpython-39.pyc
│          google_utils.cpython-39.pyc
│          loss.cpython-39.pyc
│          metrics.cpython-39.pyc
│          plots.cpython-39.pyc
│          torch_utils.cpython-39.pyc
│          __init__.cpython-39.pyc
│          
├─VOC
│  ├─Annotations
│  │      000001.xml
│  │   .......
│  │      009962.xml
│  │      009963.xml
│  │      
│  ├─images
│  │  │  train.cache
│  │  │  
│  │  ├─test
│  │  │      000001.jpg
│  │  │      000002.jpg
│  │  │   .....
│  │  │      009963.jpg
│  │  │      
│  │  └─train
│  │          000001.jpg
│  │          ....
│  │          009963.jpg
│  │          
│  ├─ImageSets
│  │  ├─Layout
│  │  │      test.txt
│  │  │      
│  │  ├─Main
│  │  │      aeroplane_test.txt
│  │  │      bicycle_test.txt
│  │  │      bird_test.txt
│  │  │      boat_test.txt
│  │  │      bottle_test.txt
│  │  │      bus_test.txt
│  │  │      car_test.txt
│  │  │      cat_test.txt
│  │  │      chair_test.txt
│  │  │      cow_test.txt
│  │  │      diningtable_test.txt
│  │  │      dog_test.txt
│  │  │      horse_test.txt
│  │  │      motorbike_test.txt
│  │  │      person_test.txt
│  │  │      pottedplant_test.txt
│  │  │      sheep_test.txt
│  │  │      sofa_test.txt
│  │  │      test.txt
│  │  │      train_test.txt
│  │  │      tvmonitor_test.txt
│  │  │      
│  │  └─Segmentation
│  │          test.txt
│  │          
│  ├─JPEGImages
│  │      000001.jpg
│  │      。。。
│  │      009963.jpg
│  │      
│  ├─SegmentationClass
│  │      000068.png
│  │    。。。
│  │      009901.png
│  │      
│  └─SegmentationObject
│          000068.png
│         。。。
│          009901.png
│

我要发帖子