Artificial Intelligence/Computer Vision2021. 7. 28. 16:07pretrained model layer 수정하기
Case 1. pretrained CNN model layer 직접 수정 pretrained CNN model layer 직접 수정하여 input channel을 변경해 보도록 하겠습니다. Problem 일반적인 classification task를 위한 cnn 기반 모델의 인풋은 3 channel (RGB)로 들어가게 됩니다. 하지만 데이터를 가공시키고 원하는 input channel 이 3채널이 아닐때(4채널 or 그 이상 채널이 들어가야 할 때) pretrained cnn 모델을 가져와서 사용하고 싶을때 본 방법은 유용합니다. 저는 se_resnext101 네트워크에 ImageNet dataset이 pretrained 모델을 가져와서 사용하였습니다. 그리고 제가 가공한 이미지 데이터는 9*256*2..