pythorchを使って画像分類をする人工知能を作りたいと考えています。
自分で集めた画像でデータセットを作りたいのですが方法がわからずお力を貸していただきたいです。
データセットのテンプレート的なものは調べると出てきますが一様にファイルから画像を参照してるように思います。
ファイルからではなく変数に代入されているものを使いたい場合、特にですがgetitemのところをどう書いたらよいのかが思いつきません。
例えばx=pd.read_csv(~~~)と直前にしたとき、xを使ってデータセットを作りたいです。
下記のコードはデータセット 作り方で調べるとこのようなものが多かったです。
該当のソースコード
python
1#???のところはどう書いたらよいかわからなかったです。 2class MyDataset(Dataset): 3 4 def __init__(self, csv_file_path, root_dir, transform=None): 5 self.image_dataframe = pd.read_csv(csv_file_path) 6 self.root_dir = root_dir 7 self.transform = transform 8 9 def __len__(self): 10 return len(self.image_dataframe) 11 12 def __getitem__(self, idx): 13 label = self.image_dataframe.iat[idx, 1] 14 img_name = os.path.join(self.root_dir, ???, self.image_dataframe.iat[idx, 2]) 15 image = io.imread(img_name) 16 if self.transform: 17 image = self.transform(image) 18 19 return image, label
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2019/09/24 00:02