인프런 영문 브랜드 로고
인프런 영문 브랜드 로고

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

Ann Kim님의 프로필 이미지
Ann Kim

작성한 질문수

딥러닝 CNN 완벽 가이드 - TFKeras 버전

AWS sagemaker을 이용하여 실습시 CIFAR10 데이터세트 문제점

작성

·

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))

 

감사합니다.

Ann Kim님의 프로필 이미지
Ann Kim

작성한 질문수

질문하기