CIFAR-10をダウンロードし、pngかjpgの形式に書き換え、ファイル構造を変える

Question

実現したい事

CIFAR-10をダウンロードし、pngかjpgの形式に書き換え

ファイル構造を変える事ってできますかね？

CIFAR10-master
｜
｜—クラス1 — クラス1のデータ
｜—クラス2 — クラス2のデータ
｜—クラス3 — クラス3のデータ
｜—クラス4 — クラス4のデータ
｜　　・
｜ 　・
｜　　・
｜—クラス10 — クラス10のデータ

上記の構造が理想です。
ファイル/フォルダ名はテキトーです。

通常であれば以下のコードが一般的だと思いますが、
実現したい事を実現できるコードが見つからないため、質問しました。

# データセットの準備
from tensorflow.keras.datasets import cifar10

(train_images, train_labels), (test_images, test_labels) = cifar10.load_data()

ホームページからCIFAR-10をダウンロードし、ファイルを解凍したのですが、
画像ファイルではなかったです。

Answer 1

Pillow で保存すればいいと思います。

from tensorflow.keras.datasets import cifar10
from pathlib import Path
from PIL import Image

(train_images, train_labels), (test_images, test_labels) = cifar10.load_data()

output_dir = Path("cifar10/train")

for i, (img, label) in enumerate(zip(train_images, train_labels)):
    save_dir = output_dir / str(label[0])
    save_dir.mkdir(exist_ok=True, parents=True)
    save_path = save_dir / f"{i}.jpg"

    img = Image.fromarray(img)
    img.save(save_path)

Comments

[kane_study]

ありがとうございます。
コードを読めば意味は分かるのですが、自力で0から書く事は中々難しいです…
本当に助かりました。