Image Classification Paper Reading

This paper realizes the classification of breast ultrasound images by combining VGG-19 and VIT models Breast Ultrasound Images Dataset | Kaggle

PyTorch VGG19 reproduction code

# 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)