[미니 프로젝트] DL.2 각 클래스의 이미지 확인하기

미니 프로젝트

문항 2 각 클래스의 이미지 확인하기

1. 라이브러리 호출

• 운영 체제와 연동하여 파일을 호출하기 위해 os 라이브러리
• 이미지를 읽어올 opencv 라이브러리 cv2
• 이미지를 표현해줄 matplotlib.pyplot 라이브러리 호출
• 랜덤으로 이미지를 선정하여 표시해주기 위해 random 라이브러리 호출

#%% 라이브러리 호출
import os
import matplotlib.pyplot as plt 
import cv2
import random

2. 폴더 경로와 카테고리 확인

하위 폴더 안에 저장되어있는 Rice_Image_Dataset 경로를 지정한 후, 카테고리의 이름을 미리 저장해두고, 개수를 세어둔다.

#%% 폴더 경로와 분류 카테고리 확인
image_path="./Rice_Image_Dataset/"
categories=["Arborio","Basmati","Ipsala","Jasmine","Karacadag"]
num_of_categories=len(categories)

3. 이미지 확인하기

각 카테고리마다 15000장의 사진이 있는 관계로 모든 사진을 확인하기는 어렵다. 따라서, 카테고리별로 랜덤으로 10장씩 이미지를 표시할 수 있도록 코드를 제작하였다.

#%% 각 카테고리 별로 이미지를 10장씩 랜덤으로 표시하기
for idx, cate in enumerate(categories):
    label=[0]*num_of_categories
    label[idx]=1
    image_dir=image_path+categories[idx]+"/"
    print(image_dir)
    plt.figure()
    for i in range(10):
        number=random.randint(1,15000) # 1부터 15000까지의 숫자중에서 하나를 선발
        img_dir = os.path.join(image_dir, f"{categories[idx]} ({number}).jpg")
        img=cv2.imread(img_dir,1)
        img=cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
        plt.subplot(2,5,i+1)
        plt.imshow(img)
        plt.title(f"{categories[idx]} ({number}).jpg")

결과

1) Arborio

2) Basmati

3) Ipsala

4) Jasmine

5) Karacadag