递归的数量为3,6,9,12时候的模型的效果
递归数量为3的时候
from torch import nn
import torch
import numpy as np
import os
os.environ['CUDA_VISIBLE_DEVICES']='1'
class Conv_ReLU_Block(nn.Module):
def __init__(self,in_channels=64,out_channels=64,kernel_size=3):
super(Conv_ReLU_Block, self).__init__()
self.conv = nn.Conv2d(in_channels=in_channels, out_channels=out_channels, kernel_size=kernel_size, stride=1, padding=kernel_size//2, bias=False)
self.relu = nn.ReLU(inplace=True)
def forward(self, x):
return self.relu(self.conv(x))
#递归多尺度残差块
class msrb(nn.Module):
def __init__(self,conv=Conv_ReLU_Block,in_channels=64,out_channels=128):
super(msrb, self).__init__()
self.num=3
self.conv_1_1=conv(in_channels=in_channels,out_channels=out_channels,kernel_size=1)
self.conv_