-
질문 & 답변
카테고리
-
세부 분야
컴퓨터 비전
-
해결 여부
미해결
AWS sagemaker을 이용하여 실습시 CIFAR10 데이터세트 문제점
22.09.24 10:14 작성 조회수 186
0
안녕하세요~~
친절하고 자세한 강의 잘 듣고 있습니다~ ㅎㅎ
CIFAR10 데이터세트를 이용하여 CNN 모델 구현 실습 - 01 강의에서 저는 AWS sagemaker를 이용하여 실습코드를 따라해 보고 있는데 거기서는 CIFAR10 데이터세트를 download하면 아래와 같이 됩니다 ㅠㅠ
어떻게 해야 할까요??
train dataset shape: (50000, 3, 32, 32) (50000, 1)
답변을 작성해보세요.
1
권 철민
지식공유자2022.09.24
안녕하십니까,
학습 이미지 데이터 세트가 (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))
감사합니다.
답변 1