图像分类论文阅读
该论文通过结合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)