图像分类论文阅读

该论文通过结合VGG-19和VIT模型，实现乳腺超声图像的分类Breast Ultrasound Images Dataset | Kaggle

PyTorch VGG19复现代码

# VGG19.py
import torch
import torch.nn as nn

class Conv(nn.Module):
    def __init__(self, in_channels, out_channels, kernel_size=1, stride=1,
                 padding=None, groups=1, activation=True):
        super(Conv, self).__init__()
        padding = kernel_size // 2 if padding is None else padding
        self.conv = nn.Conv2d(in_channels, out_channels, kernel_size, stride,
                              padding, groups=groups, bias=True)