该论文通过结合VGG-19和VIT模型，实现乳腺超声图像的分类Breast Ultrasound Images Dataset | Kaggle
PyTorch VGG19复现代码
# VGG19.py
import torch
import torch.nn as nnclass 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 paddingself.conv = nn.Conv2d(in_channels, out_channels, kernel_size, stride,padding, groups=groups, bias=True)
