작성
·
246
0
안녕하세요~~
친절하고 자세한 강의 잘 듣고 있습니다~ ㅎㅎ
CIFAR10 데이터세트를 이용하여 CNN 모델 구현 실습 - 01 강의에서 저는 AWS sagemaker를 이용하여 실습코드를 따라해 보고 있는데 거기서는 CIFAR10 데이터세트를 download하면 아래와 같이 됩니다 ㅠㅠ
어떻게 해야 할까요??
train dataset shape: (50000, 3, 32, 32) (50000, 1)
답변 1
1
안녕하십니까,
학습 이미지 데이터 세트가 (50000, 3, 32, 32)면 channel first 방식으로 이미지가 만들어졌군요.
numpy의 transpose()를 이용해서 축을 변경하면 될 것 같습니다.
import numpy as np
train_image_np = train_dataset[0]
channel_last_train_image = np.transpose(train_image_np, (0,2, 3, 1))
감사합니다.