dla60

1070:

512  1: 18ms

512 4  66ms

dla34
 

512  1: 12ms  8: 66ms 

416 1:11ms  8:47ms   16 90ms

  

   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      #!/usr/bin/env python
     
    
   

    

     

    
    

     

      # -*- coding: utf-8 -*-
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import math
     
    
   

    

     

    
    

     

      import time
     
    
   

    

     

    
    

     

      from os.path import join
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     

      from torch import nn
     
    
   

    

     

    
    

     

      import torch.utils.model_zoo as model_zoo
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      # import dataset
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      BatchNorm = nn.BatchNorm2d
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      WEB_ROOT = 'http://dl.yf.io/dla/models'
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def get_model_url(data, name):
     
    
   

    

     

    
    

     
 return join(WEB_ROOT, data.name,
     
    
   

    

     

    
    

     
 '{}-{}.pth'.format(name, data.model_hash[name]))
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def conv3x3(in_planes, out_planes, stride=1):
     
    
   

    

     

    
    

     
 "3x3 convolution with padding"
     
    
   

    

     

    
    

     
 return nn.Conv2d(in_planes, out_planes, kernel_size=3, stride=stride,
     
    
   

    

     

    
    

     

       padding=1, bias=False)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      class BasicBlock(nn.Module):
     
    
   

    

     

    
    

     
 def __init__(self, inplanes, planes, stride=1, dilation=1):
     
    
  
 

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

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