TFRecordが生成できない

conda activate deeplab2
cd 
C:\Users\Nialys\Downloads\school2\kenkyu\data\models\research\deeplab\datasets


python ./build_voc2012_data.py --image_folder="./pascal_voc_seg/VOCdevkit/VOC2012/JPEGImages" --semantic_segmentation_folder="./pascal_voc_seg/VOCdevkit/VOC2012/SegmentationClassRaw" --list_folder="./pascal_voc_seg/VOCdevkit/VOC2012/ImageSets/Segmentation" --image_format="jpg" --output_dir="./pascal_voc_seg/tfrecord/"

これは成功する

python ./build_voc2012_data.py --image_folder="./original/img/train" --semantic_segmentation_folder="./original/lbl/train" --list_folder="./original/lst/train" --image_format="jpg" --output_dir="./original/tfrecord"

これは失敗する

なんでじゃ???

⇒ 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"

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"


学習

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=true --last_layers_contain_logits_only=false

うまくいかず

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

うまくいった!

モデルのエクスポート

python export_model.py --checkpoint_path="./datasets/original/log/model.ckpt-300" --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

いけるかな?

いけたわ。


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