🔅 PyTorch 에서 Conv2d, MaxPool2d 적용시 자동으로 이미지 output shape 계산해 주는 계산기

[teddylee777 (Teddy Lee)]

안녕하세요? 페이스북 PyTorch KR에는 공유 드렸던 내용인데요.
discuss.pytorch.kr 에도 공유 드려 봅니다.
(아마도 파이토치 입문자 분들께 도움이 될만한 내용입니다)

파이토치에서 Conv2d와 MaxPool2d 레이어를 쌓을때,
레이어를 통과해서 Image Output에 대한 출력사이즈에 대한 계산을 해야 하는데,
이 부분이 번거롭기도 하고, 처음 하시는 분들은 계산을 어려워 하실 수 있어서(제가 그랬거든요 ㅠ)
간단히 웹앱으로 Image Output에 대한 계산기를 만들어 봤습니다.

Streamlit이 프레임워크를 편하게 제공하더라구요~(처음 써봤는데, 정말 금방 훌륭한 데모 사이트를 만들어 줍니다!)

관련 코드와 데모 사이트를 공개해 놓았습니다. 혹시 도움이 되셨으면 좋겠습니다.

깃헙: https://github.com/teddylee777/pytorch-layer-calculator

사이트링크: https://bit.ly/torch-calc

사용법은

1. 이미지 input 사이즈 정의 후

2. Conv2d, MaxPool2d 를 추가해 주시면 실시간 반영되어 계산됩니다.

기타 건의사항이나 버그에 대한 리포팅은 github issue로 주시면 감사하겠습니다.

즐거운 하루 되십쇼!

[9bow (박정환)]

앗, 감사합니다! 제가 오늘 계속 밖이라 저녁에 올리려고 핬는데 직접 올려주셨군요
감사합니다!!

[teddylee777 (Teddy Lee)]

덕분에 파이토치 한국 사용자모임 커뮤니티를 알게 되었습니다^^
감사합니다!!