carafe 上采样

代码：

https://github.com/sephirothhua/Lite_Upsample_Network_Carafe-unoffical-/blob/master/carafe.py

80 128 26 26 p100 500 ms

torch自带的上采样为0ms

  

   

    

     

    
    

     

      import time
     
    
   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     

      from torch import nn
     
    
   

    

     

    
    

     

      from torch.nn import functional as F
     
    
   

    

     

    
    

     

      from collections import OrderedDict
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      class KernelPredeictionModule(nn.Module):
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 def __init__(self, input_channel, channel_cm=64, kernel_up=5, kernel_encoder=3, enlarge_rate=2):
     
    
   

    

     

    
    

     

       super(KernelPredeictionModule,self).__init__()
     
    
   

    

     

    
    

     

       self.input_channel = input_channel
     
    
   

    

     

    
    

     

       self.channel_cm = channel_cm
     
    
   

    

     

    
    

     

       self.kernel_up = kernel_up
     
    
   

    

     

    
    

     

       self.kernel_encoder = kernel_encoder
     
    
   

    

     

    
    

     

       self.enlarge_rate = enlarge_rate
     
    
   

    

     

    
    

     

       self.channel_compressor = nn.Sequential(
     
    
   

    

     

    
    

     

       OrderedDict([
     
    
  
 

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

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