VGG19代码实现以及提取特定网络层

VGG:通过使用重复的基础块,来构建深层网络。
block:使用数个相同的填充为1,窗口形状大小为3的卷积层,接上一个窗口大小为2,步幅为2的最大池化。

VGG19 网络结构

import torch
import torch.nn as nn
import torch.nn.functional as F


class FlattenLayer(nn.Module):
    def __init__(self)

你可能感兴趣的:(python,卷积神经网络,pytorch)