当前位置:   article > 正文

yolov5模型压缩-eagleeye

yolov5模型压缩

参考论文:EagleEye: Fast Sub-net Evaluation for Efficient Neural Network Pruning(https://arxiv.org/abs/2007.02491)
基本原理:揭示剪枝不同的DNN结构对最终精度的影响,从而找到现在的剪枝后精度最高的模型
模型压缩效果
在这里插入图片描述在这里插入图片描述在这里插入图片描述1、新建eagleeye.py,添加如下代码:

import argparse
import os
import random
import sys
from copy import deepcopy
import numpy as np
os
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/花生_TL007/article/detail/604803
推荐阅读
相关标签
  

闽ICP备14008679号