Middle
Որ շերտը մեծացնում է չափը U-Net-ում հետադարձ փոխանցման ժամանակ և ինչպես է աշխատում:
sobes.tech AI
Պատասխան AI-ից
U-Net- ում, հետադարձ ճանապարհին հատկությունների չափը մեծացնելու համար, օգտագործվում է upsampling շերտ, որը սովորաբար իրականացվում է որպես փոխված կոնվոլյուցիա (transposed convolution) կամ upsampling գործողություն (օրինակ, bilinear interpolation) հաջորդած կոնվոլյուցիայի հետ:
Այս շերտը մեծացնում է մուտքային տենսորի տարածքային չափերը (լայնությունը և բարձրությունը), վերականգնելով նվազեցված լուծաչափը, որը կորցվել է downsampling փուլում: Դա օգնում է ցանցին վերականգնել պատկերային մանրամասները, համակցելով տեղեկատվությունը համապատասխան շերտերի հետ ուղիղ անցումից՝ skip կապերի միջոցով:
PyTorch-ում օրինակ՝ փոխված կոնվոլյուցիայի հետ:
import torch.nn as nn
upconv = nn.ConvTranspose2d(in_channels=64, out_channels=32, kernel_size=2, stride=2)
# մեծացնում է մուտքի չափը 2 անգամ լայնությամբ և բարձրությամբ