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
  
 

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/91896110