bagnet 测试

1060上，1个640*640都跑不了 

权重60多m

  

   

    

     

    
    

     

      import time
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import torch.nn as nn
     
    
   

    

     

    
    

     

      import math
     
    
   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     

      from collections import OrderedDict
     
    
   

    

     

    
    

     

      from torch.utils import model_zoo
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import os
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      dir_path = os.path.dirname(os.path.realpath(__file__))
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      __all__ = ['bagnet9', 'bagnet17', 'bagnet33']
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      model_urls = {
     
    
   

    

     

    
    

     
 'bagnet9': 'https://bitbucket.org/wielandbrendel/bag-of-feature-pretrained-models/raw/249e8fa82c0913623a807d9d35eeab9da7dcc2a8/bagnet8-34f4ccd2.pth.tar',
     
    
   

    

     

    
    

     
 'bagnet17': 'https://bitbucket.org/wielandbrendel/bag-of-feature-pretrained-models/raw/249e8fa82c0913623a807d9d35eeab9da7dcc2a8/bagnet16-105524de.pth.tar',
     
    
   

    

     

    
    

     
 'bagnet33': 'https://bitbucket.org/wielandbrendel/bag-of-feature-pretrained-models/raw/249e8fa82c0913623a807d9d35eeab9da7dcc2a8/bagnet32-2ddd53ed.pth.tar',
     
    
   

    

     

    
    

     

      }
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      class Bottleneck(nn.Module):
     
    
   

    

     

    
    

     

       expansion = 4
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 def __init__(self, inplanes, pla
     
    
  
 

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

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