인프런 커뮤니티 질문&답변

 현재 VGG 모델 학습 중입니다. 이전 AlexNet까지는 정상적으로 (h w c) 순으로 모델이 만들어졌는데, VGG 모델부터는 갑자기 (c h w) 순으로 입력이 되고 있습니다. 그러다보니, 실습코드대로 작업을 하면 계속 제일 앞에 있는 차원을 channel로 인식하게 되어 차원이 맞지 않아 모델이 정상적으로 만들어지지 않습니다. (지금까지 모두 동일한 가상환경에서 작업하고 있었고, 업데이트 등을 한 적이 없습니다.)

 혹시 몰라 이전에 작업한 AlexNet, CIFAR10 등의 파일들도 다시 실행해보니 모두 동일한 현상을 보였으며, CIFAR10의 경우에는 처음 dataset을 불러올 때 역시 (c h w) 순으로 불러오는 것을 확인하였습니다. 강제로 data_format을 channels_last로 설정할 수는 있지만, pretrained model의 경우 Conv2D, MaxPooling 등을 제 마음대로 변경할 수 없어서 현재 학습에 어려움을 겪고 있는 상황입니다.

 혹시 코드 내부에서, 혹은 conda에 추가적으로 설치해서 강제로 마지막 차원을 channel로 인식하도록 하는 방법이 있는지 알고 싶습니다. 감사합니다.