0

タイプ、クラス、パスの 3 つの列を持つ reference.csv ファイルがあります。最初の 5 行の例を次に示します。

"Type","Class","Path"
"train","A","./path1/001.jpg"
"train","A","./path2/002.jpg"
"test","C","./path3/003.jpg"
"train","B","./path4/001.jpg"
"test","B","./path5/002.jpg"
...

より見やすい形式で:

|----------------------|------------------|------------------|
|         Type         |       Class      |       Path       |
|----------------------|------------------|------------------|
|        train         |         A        | ./path1/001.jpg  |
|----------------------|------------------|------------------|
|        train         |         A        | ./path2/002.jpg  |
|----------------------|------------------|------------------|
|        train         |         C        | ./path3/003.jpg  |
|----------------------|------------------|------------------|
|        test          |         B        | ./path4/001.jpg  |
|----------------------|------------------|------------------|
|        test          |         B        | ./path5/002.jpg  |
|----------------------|------------------|------------------|

DataLoader (torch.utils.data.DataLoader) を使用できるように、データセット クラス (torch.utils.data.Dataset) を作成して画像を読み取りたいと考えています。

参照テーブルを使用してカスタム データセットを作成する正しい方法は何ですか?

4

0 に答える 0