【Python/Pytorch - 网络模型】-- 手把手搭建3D VGG感知损失模型

文章目录

00 写在前面

感知损失：对于提升图片的肉眼可见细节，效果十分明显；对于一些指标如（SSIM、PSNR）这些，效果不明显。

在01中，可以根据3D VGG的网络结构，进行模块化编程，主要包括VGG3D模块。

在模型调试过程中，可以先通过简单测试代码，进行代码调试。

01 基于Pytorch版本的3D VGG代码

# 库函数调用
import torch
import torch.nn as nn

# VGG3D模块
class CustomVGG3D(nn.Module):
    def __init__(self, in_channels=3, out_channels=2):
        super(CustomVGG3D, self).__init__()
        self.features = nn.Sequential(
            nn.Conv3d(in_channels, 64, kernel_size=(3, 3, 3), padding=(1, 1, 1)),
            nn.ReLU(inplace=True),
            nn.Conv3d(64, 64, kernel_size=(3, 3, 3), padding=(1, 1, 1)),
            nn.ReLU(inplace=True),
            nn.MaxPool3d(kernel_size=(2, 2, 2), stride=(2, 2, 2)),
            nn.Conv3d(64, 128, kernel_size=(3, 3, 3), padding=(1, 1, 1)),
            nn.ReLU(inplace=True),
            nn.Conv3d(128, 128, kernel_size=(3, 3, 3), padding=(1, 1, 1)),
            nn.ReLU(inplace=True),
            nn.MaxPool3d(kernel_size=(2, 2, 2), stride=(2, 2, 2)),
            nn.Conv3d(128, 256, kernel_size=(3, 3, 3), padding=(1, 1, 1)),
            nn.ReLU(inplace=True),
            nn.Conv3d(256, 256, kernel_size=(3, 3, 3), padding=(1, 1, 1)),
            nn.ReLU(inplace=True),
            nn.Conv3d(256, 256, kernel_size=(3, 3, 3), padding=(1, 1, 1)),
            nn.ReLU(inplace=True),
            nn.Conv3d(256, 256, kernel_size=(3, 3, 3), padding=(1, 1, 1)),
            nn.ReLU(inplace=True),
            # nn.MaxPool3d(kernel_size=(2, 2, 2), stride=(2, 2, 2)),
            # nn.Conv3d(256, 512, kernel_size=(3, 3, 3), padding=(1, 1, 1)),
            # nn.ReLU(inplace=True),
            # nn.Conv3d(512, 512, kernel_size=(3, 3, 3), padding=(1, 1, 1)),
            # nn.ReLU(inplace=True),
            # nn.Conv3d(512, 512, kernel_size=(3, 3, 3), padding=(1, 1, 1)),
            # nn.ReLU(inplace=True),
            # nn.MaxPool3d(kernel_size=(2, 2, 2), stride=(2, 2, 2)),
            # nn.Conv3d(512, 512, kernel_size=(3, 3, 3), padding=(1, 1, 1)),
            # nn.ReLU(inplace=True),
            # nn.Conv3d(512, 512, kernel_size=(3, 3, 3), padding=(1, 1, 1)),
            # nn.ReLU(inplace=True),
            # nn.Conv3d(512, 512, kernel_size=(3, 3, 3), padding=(1, 1, 1)),
            # nn.ReLU(inplace=True),
            # nn.MaxPool3d(kernel_size=(2, 2, 2), stride=(2, 2, 2)),
        )
        self.classifier = nn.Sequential(
            nn.Linear(512 * 8 * 8 * 1, 4096),
            nn.ReLU(True),
            nn.Linear(4096, 4096),
            nn.ReLU(True),
            nn.Linear(4096, out_channels),
            nn.Sigmoid()
        )

    def forward(self, x):
        x = self.features(x)
        # x = x.view(x.size(0), -1)
        # x = self.classifier(x)
        return x

# 测试代码
# if __name__ == '__main__':
#     x = torch.ones([2, 4, 256, 256, 32])
#     model = CustomVGG3D(in_channels=4, out_channels=1)
#     f = model(x)
#     print(f)

02 论文下载

Very deep convolutional neural network based image classification using small training sample size
arXiv: VERY DEEP CONVOLUTIONAL NETWORKS FOR LARGE-SCALE IMAGE RECOGNITION