반응형

pytorch 3

torch. squeeze의 역할

squeeze의 역할은 텐서에서 차원의 크기가 1인 차원을 제거한다는 것이다. 그러면 차원의 크기가 1인 애들을 왜 제거하는 것일까? 1. 차원축소: 차원이 1인 경우, 해당 차원은 단 하나의 요소를 가지기 때문에 정보적으로 불필요하다. 따라서 차원을 제거함으로써 텐서의 크기를 줄이고 메모리 사용을 줄일 수 있다. 그렿다면 왜 단 하나의 요소만을 가지면 정보적으로 불필요한 것일까? 이 이미지를 보면 이해가 된다. [ [ [1,2] , [3,4] ] ] 은 3차원 텐서로 텐서의 shape은 (1,2,2)이다. 이때 squeeze가 크기가 1인 차원을 없앤다는 것이다. 그래서 squeeze를 하고 나면 [ [1,2], [3,4] ]가 되면서 2차원의 텐서가 되며 (2,2) shape을 가지게 된다. 이때 크..

[PyTorch, MONAI] UNETR 모델 생성 및 Forward

UNETR이 무엇인지는 2023.01.11 - [Programming] - [MONAI] UNETR 이란? (feat. Vision Transformers) [MONAI] UNETR 이란? (feat. Vision Transformers) 이 설명은 당장 UNETR를 써야하는데 빨리 뭔지 대충 알고 싶은 경우만 살짝 도움이 될뿐 자세한 내용은 아래의 논문에서 확인가능 https://arxiv.org/abs/2103.10504 Background 한 줄 정리 FCNN(Fully Convolutional N jedemanfangwohnteinzauberinne.tistory.com UNETR 모델에 쓰기위한 데이터의 전처리는 2023.01.11 - [Programming] - [MONAI, PyTorch] ..

[MONAI, PyTorch] MONAI를 이용해 데이터 전처리 하기

이번에 인턴십 중에 MONAI를 이용한 프로젝트를 하게 되었다. 내용을 공부할겸 여기에 정리를 해본다. MONAI MONAI(Medical Open Network for AI) 는 엔비디아가 만든 헬스케어용 오픈소스 프레임워크(파이토치 기반)다. UNETR 이란? 이전 글에서 간략하게 다룬적이 있다. 2023.01.11 - [Programming] - [MONAI] UNETR 이란? (feat. Vision Transformers) [MONAI] UNETR 이란? (feat. Vision Transformers) 이 설명은 당장 UNETR를 써야하는데 빨리 뭔지 대충 알고 싶은 경우만 살짝 도움이 될뿐 자세한 내용은 아래의 논문에서 확인가능 https://arxiv.org/abs/2103.10504 Ba..

반응형