yolov3 测试

darknet53 在我1060上gpu 352*352 需要21ms，cpu需要900ms，但是参数比较多，占用内存比较大？
 

总时间需要30多ms

  

   

    

     

    
    

     

      import datetime
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     

      import torch.nn as nn
     
    
   

    

     

    
    

     

      import math
     
    
   

    

     

    
    

     

      from collections import OrderedDict
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      __all__ = ['darknet21', 'darknet53']
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      class BasicBlock(nn.Module):
     
    
   

    

     

    
    

     
 def __init__(self, inplanes, planes):
     
    
   

    

     

    
    

     

       super(BasicBlock, self).__init__()
     
    
   

    

     

    
    

     

       self.conv1 = nn.Conv2d(inplanes, planes[0], kernel_size=1,
     
    
   

    

     

    
    

     

       stride=1, padding=0, bias=False)
     
    
   

    

     

    
    

     

       self.bn1 = nn.BatchNorm2d(planes[0])
     
    
   

    

     

    
    

     

       self.relu1 = nn.LeakyReLU(0.1)
     
    
   

    

     

    
    

     

       self.conv2 = nn.Conv2d(planes[0], planes[1], kernel_size=3,
     
    
   

    

     

    
    

     

       stride=1, padding=1, bias=False)
     
    
   

    

     

    
    

     

       self.bn2 = nn.BatchNorm2d(planes[1])
     
    
   

    

     

    
    

     

       self.relu2 = nn.LeakyReLU(0.1)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 def forward(se
     
    
  
 

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

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