Milline kiht suurendab suurust U-Net-i tagasipöördumisel ja kuidas see töötab?

U-Net-is kasutatakse tagasiteel omaduste suurendamiseks upsampling-kihte, mis tavaliselt realiseeritakse transpoositud konvolutsiooni (transposed convolution) või upsampling-operatsioonina (näiteks bilineaarinterpolatsioon) koos järgnevate konvolutsioonidega.

See kiht suurendab sisendi tensorite ruumilisi mõõtmeid (laiust ja kõrgust), taastades madalamale tasemele langenud resolutsiooni. See aitab võrgul taastada pildi detaile, ühendades teavet vastavate otseülekande kihtidega skip-ühenduste kaudu.

Näide PyTorchis transpoositud konvolutsiooniga:

import torch.nn as nn

upconv = nn.ConvTranspose2d(in_channels=64, out_channels=32, kernel_size=2, stride=2)
# Suurendab sisendi suurust kaks korda laiuses ja kõrguses