[pytorch] nn.Upsample
2021. 6. 16. 13:17ㆍDeep Learning/framework
728x90
주어진 멀티채널 1D / 2D / 3D 데이터를 업샘플링한다.
인풋데이터는 minibatch x channels x [optional depth] x [optional height] x width
형태로 가정되나, spatial input(2D)은 4D 텐서, volumetric input(3D)은 5D텐서가능한 알고리즘 : nearest neighbor, linear, bilinear, bicubic, 3D이상의 데이터에 대해서는 trilinear
- parameters
size (int/tuple) : output 사이즈
scale_factor(float/tuple) : Depth, Height, Width에 입력받은 정수 만큼 곱해준다. (1,1,2,2) 가 인풋으로 들어오면 (1,1,4,4)가 된다.
mode (str) – 업샘플링 알고리즘 : 'nearest', 'linear', 'bilinear', 'bicubic', 'trilinear'. Default: 'nearest'
align_corners(bool) : True이면 인풋과 아웃풋의 코너 픽셀들이 정렬됨. 그 픽셀들의 값이 보존된다. 단, mode가 linear, bilinear, trilinear일 때만 적용 가능. Default: False
size와 scala_factor는 함께 사용하지 않음.
downsampling이나 일반적인 resizing을 원하면 interpolate()을 사용하라고 한다.
https://pytorch.org/docs/stable/generated/torch.nn.Upsample.html
728x90
'Deep Learning > framework' 카테고리의 다른 글
JSON파일 Python 으로 다루기 (0) | 2021.07.04 |
---|---|
[Pytorch] Variable (0) | 2021.06.19 |
[Pytorch] nn.ReflectionPad2d (0) | 2021.06.16 |
[Pytorch] Initializer (torch.nn.init) (0) | 2021.06.15 |
[Pytorch] Tensor 의 .view()메소드 / .reshape()메소드 (0) | 2021.05.24 |