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([