colabで自分のファイル（データセット）を読み込ませたい

Question

前提

google colab上でpix2pixをやろうとしています。

実現したいこと

以下ではpix2pixの予め用意されたデータセットをロードしているのですが、自分で作成したデータセットを代わりに読み込ませたいです。
しかし、どの部分をどのように変更すればよいのかわからず教えていただけますでしょうか。
以下のコードの２段落目の学習データの確認で自作のデータセットがロードされているところまでいきたいです。

該当のソースコード

# facadesデータセットのダウンロード
# cityscapes, night2day, edges2handbags, edges2shoes, facades, mapsも同様
!wget http://efrosgans.eecs.berkeley.edu/pix2pix/datasets/facades.tar.gz
!mkdir -p ./datasets/facades
!tar -zxvf facades.tar.gz -C ./datasets/
!rm facades.tar.gz

# 学習データの確認
import matplotlib.pyplot as plt
img = plt.imread('datasets/facades/train/1.jpg')
plt.imshow(img)

試したこと

drive上にデータセットを入れたファイルをつくりましたが、読み込み方がわかりませんでした。

ボールドテキスト

Comments

[jbpb0]

> 自分で作成したデータセットを代わりに読み込ませたい

質問に、pix2pixのコード(あるいはコードがあるurl)が提示されてないので、どのような状態になればいいのかが分りませんが、おそらく質問に記載のコードの前半を行った後の状態に合わせればいいのでしょう

質問に記載のコードの前半(「# 学習データの確認」まで)を実行すると、「datasets」というディレクトリができて、その中に「facades」というディレクトリがあり、その中に「train」と「test」と「val」というディレクトリがあり、「train」と「test」と「val」のそれぞれの中に画像ファイルがたくさんある、という状態になりました

以下にディレクトリ構成を図示します
(行頭の「.....」は、スペースの代わりです)

datasets
└── facades
....├── test
....├── train
....└── val

「train」と「test」と「val」の中にある画像ファイルのファイル名は、どれも「1.jpg」から始まる「整数.jpg」でした

自分で作成したデータセットの画像ファイルの配置を、上記の構成に極力近くしたらいいと思います
(たぶん、「facades」というディレクトリ名は変えていいと思いますが)

[jbpb0]

> どの部分をどのように変更すればよいのかわからず

> 以下のコードの２段落目の学習データの確認で自作のデータセットがロードされているところまでいきたい

とは、

> img = plt.imread('datasets/facades/train/1.jpg')

の「datasets/facades/train」の部分をどう書けばいいのか？
という意味でしょうか？

> drive上にデータセットを入れたファイルをつくりました

google drive上のパスをgoogle colabのコード上で指定する書き方が分からない、ということでしょうか？

Answer 1

質問に記載のコードの前半(「# 学習データの確認」まで)を実行すると、「datasets」というディレクトリができて、その中に「facades」というディレクトリがあり、その中に「train」と「test」と「val」というディレクトリがあり、「train」と「test」と「val」のそれぞれの中に画像ファイルがたくさんある、という状態になりました
以下にディレクトリ構成を図示します

datasets
└── facades
    ├── test
    ├── train
    └── val

「train」と「test」と「val」の中にある画像ファイルのファイル名は、どれも「1.jpg」から始まる「整数.jpg」でした

自分で作成したデータセットの画像ファイルの配置を、上記の構成に極力近くしたらいいと思います
(たぶん、「facades」というディレクトリ名は変えていいと思いますが)

　

以下のコードの２段落目の学習データの確認で自作のデータセットがロードされているところまでいきたい

質問に記載のコードと同じことができればいいだけなら、

img = plt.imread('datasets/facades/train/1.jpg')

の画像ファイルのパス「datasets/facades/train/1.jpg」を、自分で作成したデータセットの画像ファイルのパスに書き換えるだけです