Формула расчета сверточной сети:
N=(W-F+2P)/S+1
где N: выходной размер
W: размер ввода
F: размер ядра свертки
P: размер значения заполнения
S: размер шага
Пример:
nn.Conv2d(in_channels=3,out_channels=96,kernel_size=12,stride=4,padding=2)
in_channels=3: указывает количество входных каналов.Поскольку это тип RGB, количество каналов равно 3.
out_channels=96: указывает количество выходных каналов, установите количество выходных каналов на 96 (это можно установить в соответствии с вашими потребностями)
kernel_size = 12: указывает, что размер ядра свертки составляет 12x12, что является «F» выше, F = 12
шаг=4: указывает, что размер шага равен 4, то есть выше S, S=4
padding=2: указывает, что размер значения заполнения равен 2, то есть указанному выше P, P=2
Если входной размер вашего изображения 256x256, формула расчета знает N=(256-12+2x2)/4+1=63, то есть выходной размер 63x63x96