shuffleNetv2 测试

代码地址：

https://github.com/yangyuke001/FIIQA_pytorch/blob/master/shufflenetv2.py

在tx2上， 这个速度90ms，尺度2是115ms，精度比原来高

 

 input = torch.randn(1, 3, 224, 224).cuda()

这个返回最后一层的特征：

import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.autograd import Variable
from collections import OrderedDict
from torch.nn import init
import math


def conv_bn(inp, oup, stride):
    return nn.Sequential(
        nn.Conv2d(inp, oup, 3, stride, 1, bias=False),
        nn.BatchNorm2d(oup),
        nn.ReLU(inplace=True)
    )


def conv_1x1_bn(inp, oup):
    return nn.Sequential(
        nn.Conv2d(inp, oup, 1, 1, 0, bias=False),
        nn.BatchNorm2d(oup),
        nn.ReLU(inplace=True)
    )


def channel