squeezenet代码_SqueezeNet(Pytroch实现)

论文在此: SQUEEZENET: ALEXNET-LEVEL ACCURACY WIT 50X FEWER PARAMETERS AND <0.5MB MODEL SIZE

网络结构图:

Pytorch代码实现:

import torch

import torch.nn as nn

import torch.nn.init as init

class Fire(nn.Module):

def __init__(self, inplanes, squeeze_planes,

expand1x1_planes, expand3x3_planes):

super(Fire, self).__init__()

self.inplanes = inplanes

self.squeeze = nn.Conv2d(inplanes, squeeze_planes,