网络结构
三寸光阴___
这个作者很懒,什么都没留下…
展开
-
pth转onnx转mnn bug总结
pytorch版yolov3转onnx样例import torchimport torchvisionimport numpy as npfrom onnxruntime.datasets import get_exampleimport onnxruntimefrom onnx import shape_inferenceimport onnximport osfrom models import *img_size = 416cfg = 'cfg/yolov3.cfg'wei原创 2020-09-23 17:33:42 · 1387 阅读 · 0 评论 -
tensorflow模型裁剪资料收集
model-optimization官方:https://github.com/tensorflow/model-optimizationhttps://github.com/tensorflow/model-optimization/blob/master/tensorflow_model_optimization/g3doc/guide/pruning/pruning_with_keras.ipynbtensorflow.contrib.model_pruning.python剪枝模块:from原创 2020-08-21 17:00:10 · 446 阅读 · 0 评论 -
经典网络通道剪枝总结
resnet18resnet50resnet18v2senet(resnet18)resnext50mobilenetv2shufflenetv2xceptiondarknet53densenet121原创 2020-08-10 12:31:35 · 4359 阅读 · 12 评论 -
pytorch Resnet50分类模型剪枝
Resnet50网络结构:https://www.jianshu.com/p/993c03c22d52剪枝方式1.基于network-slimming论文的方法:pytorch版代码:https://github.com/Eric-mingjie/network-slimming思路:去掉downsample里面的BN层,为了方便采用Resnetv2的结构:BN-Conv-ReLU,在每一个bottleneck的第一个BN后自定义一个通道选择层(全1层),训练过程中不影响,剪枝时先生成BN的通道ma原创 2020-07-29 18:25:00 · 6381 阅读 · 30 评论 -
CenterNet(objects as points)算法总结:Mimic、DCNv1、DCNv2、DLA
DCNv1参考文章:https://zhuanlan.zhihu.com/p/37578271主要思路:卷积操作不是在规规矩矩的3x3的格子里做了,而是有了种种偏移。deformable convolution在特征图的每个位置都生成一组偏移量。以kernel=3x3为例,图中的N就是9,经过卷积后输出HxWx18的feature map,每个location位置的18分别对应3x3卷积...原创 2020-03-05 14:15:39 · 1795 阅读 · 0 评论 -
经典网络结构总结
VGGResnetinceptionV3SENetSqueezeNetMobilenetv1Mobilenetv2Mobilenetv3ShuffleNetHRNet原创 2020-02-20 22:30:16 · 278 阅读 · 0 评论