Как улучшить точность предварительно обученных моделей Deeplab - PullRequest
Новая
       31

Как улучшить точность предварительно обученных моделей Deeplab

0 голосов
/ 08 апреля 2020

Мне нужно улучшить точность от существующей модели. В частности, mobilenetv2_coco_voc_trainval , который может быть расположен по следующей ссылке:

https://github.com/tensorflow/models/blob/master/research/deeplab/g3doc/model_zoo.md

Достаточно ли продолжать обучение на выходящей контрольной точке ? Я уже пытался увеличить тренировочные шаги, но не видел никакой разницы.

Я использую команду обучения: 

python .research\deeplab\train.py \
    --logtostderr \
    --train_split="trainval" \
    --model_variant="mobilenet_v2" \
    --output_stride=16 \
    --train_crop_size="513,513" \
    --train_batch_size=8 \
    --fine_tune_batch_norm=true \
    --dataset="pascal_voc_seg" \
    --tf_initial_checkpoint=".\research\deeplab\weights\model.ckpt-30000" \
    --train_logdir=".\research\deeplab\datasets\pascal_voc_seg\exp\train_on_train_set\train" \
    --dataset_dir=".\research\deeplab\datasets\pascal_voc_seg\tfrecord" \
    --num_clones=4 
    --training_number_of_steps=10000
...