Computer Vision/Vision Site
Netron
JackYoon
2021. 3. 14. 16:52
반응형
다음은 keras나 tensorflow, pytorch를 활용하여 모델 파일을 생성하고, 생성한 모델 arichitecture를 전반적으로 확인할 수 있는 시각화 사이트를 알려드리려합니다. Netron이라는 사이트 입니다.
Netron
netron.app
저는 keras로 model을 생성하였기 때문에 .h5 모델 파일이 생성되었습니다. model을 save하면 자동적으로 모델이 save되어 파일로 나오게됩니다. 이 .h5 파일을 netron 사이트에 드래그 하시면 모델에 대한 전반적인 구조를 시각적으로 알 수 있습니다.
반응형