UNet++ 코드
저번에는 UNet++ 논문을 보고 공부를 했습니다. 이번에는 논문 내용을 토대로 UNet++를 구현하고 실행해보겠습니다. 논문에서는 총 4가지의 데이터 셋(cell nuclei, colon polyp, liver, lung nodule)으로 실험을 진행했습니다. 논문을 바탕으로 작성한 UNet++ model을 GAN 구조에서 generator의 network로 사용해보겠습니다. 먼저 UNet++ model을 구현해보겠습니다. class conv_block_nested(nn.Module): def __init__(self, in_ch, mid_ch, out_ch): super(conv_block_nested, self).__init__() self.activation = nn.LeakyReLU(negati..
연구실 공부
2022. 3. 30.