espnetv2

效果比dicenet稍微好一点，不知道为什么

https://blog.csdn.net/jacke121/article/details/114148435

import os
import time

import torch
from torch import nn

from torch.nn import init
import torch.nn.functional as F


rep_layers = [0, 5, 7, 3]

# limits for the receptive field at each spatial level
recept_limit = [13, 11, 9, 7, 5]
branches = 4

# input reinforcement related parameters
config_inp_reinf = 3
input_reinforcement = True

class CDilated(nn.Module):
    '''
    This class defines the dilated convolution.
    '''

    def __init__(self, nIn, nOut, kSize, stride=1, d=1, groups=1):
        '''
        :param nIn: number of input channels
        :param nOut: number of output channels
        :param kSize: kernel size
        :param stride: optional stride rate for down-sampling
        :param d: optional