研究メモ 総集編
Anaconda Navigator ->deeplab2 ->terminal
cd C:\Users\Nialys\Downloads\school2\kenkyu\data\models\research\deeplab\datasets
もしくは
ターミナル起動して
conda activate deeplab2
cd
C:\Users\Nialys\Downloads\school2\kenkyu\data\models\research\deeplab\datasets
TFRecord 生成
python ./build_voc2012_data.py --image_folder="./original/img" --semantic_segmentation_folder="./original/lbl" --list_folder="./original/lst" --image_format="jpg" --output_dir="./original/tfrecord"
C:\Users\Nialys\Downloads\school2\kenkyu\data\models\research\deeplab\datasets\original\tfrecord
に生成されている
上の階層に戻る
cd ..
学習
python train.py --logtostderr --training_number_of_steps=300 --train_split="val" --model_variant="xception_65" --atrous_rates=6 --atrous_rates=12 --atrous_rates=18 --output_stride=16 --decoder_output_stride=4 --train_crop_size="513,513" --train_batch_size=1 --dataset="original" --tf_initial_checkpoint="./datasets/pascal_voc_seg/init_models/deeplabv3_pascal_train_aug/model.ckpt" --train_logdir="./datasets/original/log" --dataset_dir="./datasets/original/tfrecord" --fine_tune_batch_norm=false --initialize_last_layer=false --last_layers_contain_logits_only=false
C:\Users\Nialys\Downloads\school2\kenkyu\data\models\research\deeplab\datasets\original\log
にmodel.ckptがある
エクスポート
python export_model.py --checkpoint_path="./datasets/original/log/model.ckpt-301" --export_path="./datasets/original/log/frozen_inference_graph.pb" --num_classes=2 --model_variant="xception_65" --atrous_rates=6 --atrous_rates=12 --atrous_rates=18 --output_stride=16 --decoder_output_stride=4
あ
frozen_inference_graph.pbが上と同じファイルに生成されているはず
クラスは2にしないとエラーが出る
推論
cd datasets/original/log
python prediction.py
ファイル構成
この記事が気に入ったらサポートをしてみませんか?