作業フォルダ jetson-inference/python/training/classification
| ラベルファイル | data/cat_dog/labels.txt |
| ラベル | cat dog |
| データセット | 対 象 | 猫 | 犬 |
| cat_dog | ラベル名 | cat | dog |
| 訓練用 | data/cat_dog/train | 2500 | 2500 |
| 評価用 | data/cat_dog/val | 500 | 500 |
| 試験用 | data/cat_dog/test | 100 | 100 |
| 作業 | 転移学習 | Transfer Learning with PyTorch |
| 分野 | 分類 | Classification |
| データセット | cat_dog | data/cat_dog（下記よりＤＬ） https://nvidia.box.com/shard/static/～/cat_dog.tar.gz |
| 学習コマンド | python3 train.py | – – model-dir=models/cat_dog ( – – batch-size=) ( – – works=) ( – – epochs=) data/cat_dog |
| ネットワーク | resnet18 | for PyTorch |
| 作業 | 学習済みデータの onnx への転送 | resnet18.onnx（for TensorRT） |
| 転送コマンド | python3 onnx_export.py | – – model-dir=models/cat_dog |
| 作業ディレクトリ | データ出力先（要作成） |
| 猫用 | /data/cat_dog/test_cat_output |
| 犬用 | /data/cat_dog/test_dog_output |
| 作業 | 推論 | |
| 分野 | 分類 | Classification |
| 学習済みデータ | resnet18.onnx | |
| 推論コマンド | imagenet | – – model=models/cat_dog/resnet18.onnx – – labels=data/cat_dog/labels.txt – – input-blob=input_0 – – output-blob=output_0 |
| 入出力データ （猫の推論） | 入力：data/cat_dog /test/cat | 出力：data/cat_dog /test_cat_output |
| 入出力データ （犬の推論） | 入力：data/cat_dog /test/dog | 出力：data/cat_dog /test_dog_output |