vgg16测试

这个网络会用到：

https://github.com/tgisaturday/S3FD_ATSS_SAPD

1070 640 b:4 3ms

多了就内存溢出，vgg16没有比较特殊的网络结构，gpu占用很硬核

  

   

    

     

    
    

     

      import time
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     

      from torch import nn
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      class S3FDNet(nn.Module):
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 def __init__(self, device='cuda'):
     
    
   

    

     

    
    

     

       super(S3FDNet, self).__init__()
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       self.vgg = nn.ModuleList([
     
    
   

    

     

    
    

     

       nn.Conv2d(3, 64, 3, 1, padding=1),
     
    
   

    

     

    
    

     

       nn.ReLU(inplace=True),
     
    
   

    

     

    
    

     

       nn.Conv2d(64, 64, 3, 1, padding=1),
     
    
   

    

     

    
    

     

       nn.ReLU(inplace=True),
     
    
   

    

     

    
    

     

       nn.MaxPool2d(2, 2),
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       nn.Conv2d(64, 128, 3, 1, padding=1),
     
    
   

    

     

    
    

     

       nn.ReLU(inplace=True),
     
    
   

    

     

    
    

     

       nn.Conv2d(128, 128, 3, 1, padding=1),
     
    
   

    

     

    
    

     

       nn.ReLU(inplace=True),
     
    
   

    

     

    
    

     

       nn.MaxPool2d(2, 2),
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       nn.Conv2d(128, 256, 3, 1, padding=1),
     
    
   

    

     

    
    

     

       nn.ReLU(inplace=True),
     
    
   

    

     

    
    

     

       n
     
    
  
 

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

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