研究メモ 総集編

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

ファイル構成


この記事が気に入ったらサポートをしてみませんか?