0
点赞
收藏
分享

微信扫一扫

(笔记自用)yolov3-tiny剪枝 (VOC数据集)

杰克逊爱学习 2022-04-25 阅读 76

使用代码:GitHub - tanluren/yolov3-channel-and-layer-pruning: yolov3 yolov4 channel and layer pruning, Knowledge Distillation 层剪枝,通道剪枝,知识蒸馏 

下载权重

wget https://pjreddie.com/media/files/yolov3_tiny.weights 

数据预处理

参考:详细步骤!使用yolov3-tiny训练,测试、验证VOC数据集 - 灰信网(软件开发博客聚合)

下载VOC数据集

官方: http://host.robots.ox.ac.uk/pascal/VOC/

制作yolov3-tiny-voc.cfg  yolov3-tiny-voc.data  yolov3-tiny-voc.names

训练

python train.py --cfg cfg/yolov3-tiny-voc.cfg --data data/yolov3-tiny-voc.data --weights weights/yolov3-tiny.weights --batch-size 64

出现报错:AssertionError: No labels found. Recommend correcting image and label paths

解决方法:将utils/datasets.py中的'images'改成’JPEGImages' 

参考:YOLO3模型剪枝系列(二)稀疏训练_ltshan139的博客-CSDN博客_稀疏化训练icon-default.png?t=M3K6https://blog.csdn.net/avideointerfaces/article/details/103409856?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522165089208016781667874531%2522%252C%2522scm%2522%253A%252220140713.130102334..%2522%257D&request_id=165089208016781667874531&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~all~baidu_landing_v2~default-1-103409856.142^v9^control,157^v4^control&utm_term=AssertionError%3A+No+labels+found.+Recommend+correcting+image+and+label+paths.&spm=1018.2226.3001.4187

279line self.label_files = [x.replace('images', 'labels').replace(os.path.splitext(x)[-1], '.txt')
改为     self.label_files = [x.replace('JPEGImages', 'labels').replace(os.path.splitext(x)[-1], '.txt')

出现报错:UnboundLocalError: local variable 'labels' referenced before assignment

解决办法: 将'labels4.append(labels)向右一个Tab

 

举报

相关推荐

0 条评论