2019-01-10から1日間の記事一覧

PyTorch C+ APIで自作データセットを定義する.

PyTorchのC++ APIで自作のデータセットの定義の仕方を紹介. 今回は以下のようなデータセットを作る, データセットクラス:ImageDataset(root, labelfile) 画像が格納されたルートディレクトリ:root 画像は以下のように格納 $ tree test_data/ test_data/ …