Python_11_문제풀이(Dataset과 Dataloader)

TODO

• CIFAR10 Built-in dataset 을 LOADING 후 다음을 확인하시오.
      1. Dataset loading
      2. train, test dataset의 데이터 개수
      3. class index - class name
      4. train set의 이미지 5장을 출력. label의 이름을 title로 출력.

 

1. Dataset loading

DATASET_ROOT_PATH = 'datasets'

# 1. Dataset loading
cifar10_train_set = datasets.CIFAR10(root = DATASET_ROOT_PATH
                                    ,train=True
                                    ,download=True
                                    )
cifar10_test_set = datasets.CIFAR10(root = DATASET_ROOT_PATH
                                    ,train=False
                                    ,download=True
                                    )

 

 

2. train, test dataset의 데이터 개수

# 2. train, test dataset의 데이터 개수
print(len(cifar10_train_set), len(cifar10_test_set))

 

3. class index - class name

# 3. class index(정답) -> class name(정답의 의미)
cifar10_train_set.classes
# class name -> class
cifar10_train_set.class_to_idx

 

4. train set의 이미지 5장을 출력. label의 이름을 title로 출력.

# 4. train set의 이미지 5장을 출력. label의 이름을 title로 출력.
for i in range(5):
    plt.subplot(1, 5, i+1)
    x, y = cifar10_train_set[i] # 튜플대입
    img = np.array(x) # PIL.Image -> ndarray
    plt.imshow(img)
    plt.title(cifar10_train_set.classes[y]) # label: int -> 문자열로 변환.

plt.tight_layout()
plt.show()