本项目以提高yolo在cup资源上的推理速度为目标,对yolov5模型进行了剪枝实践,包括通道剪枝和块剪枝。 通道剪枝的实现有很多案例资源,但块剪枝的很少,本项目除了通道剪枝还实践了块剪枝,块剪枝具体实现是先用权重置0的方式进行软剪枝,再导出为ONNX模型,操作ONNX计算图删除需要修剪的结构块。 本项目的yolo检测任务是...
参考:midasklr/yolov5prune at v6.0 (github.com) 0、D:\code\codePy\yolov5-6.1\runs\train\exp25文件夹下有 events.out.tfevents.1675823043.DESKTOP-ACC9FL4.52196 类似文件; 1、在当前文件夹下执行tensorboard --logdir . 点击 蓝色部分 进去浏览器(上面选项几个都可看) 这里bn主要是稀疏训练需要...