u-net이 이리 오래전에 나온 모델일줄이야

요새 이미지로부터 Dense Prediction을 해야하는 모델을 개발중에 있다.

 

그래서 이분야의 초석이라 할 수 있는 U-Net모델을 베이스라인으로 삼아서 테스트 해보고있다.

 

https://arxiv.org/abs/1505.04597

U-Net: Convolutional Networks for Biomedical Image Segmentation

 

 

무려 2015년에 나온 논문이였다.

 

무려 내가 학부 2학년때 나온 논문이다...

 

하도 유명한 네트워크고 Segmentation이라는 Classification 보다 고수준의 Task다보니 어렴풋이 생각했을때 어느정도 Classification Task 에서 많은 발전이 있고나서(내 생각엔 ResNet 나오고나서) 나온 줄 알았는데 그게 아니라 살짝 충격을 받았다.

 

우선 아래 프로젝트에서 모델 구조 그대로 갖다가 from scratch로 써보고 있다.

https://github.com/milesial/Pytorch-UNet

GitHub - milesial/Pytorch-UNet: PyTorch implementation of the U-Net for image semantic segmentation with high quality images