Как получить входные данные поезда для обучения API обнаружения объектов Tensorflow?

Question

При модели поезда fast_rcnn_resnet101 потери отображаются на терминале на каждом этапе.

Я хочу знать, какие данные вводятся на каждом этапе.когда потеря увеличивается, я не знаю, почему потеря увеличивается.

Есть ли кто-то, кто знает, как видеть входные данные на каждом шаге?

Answer 1

Вы не можете проверить результат каждого шага, но в вашем каталоге object_detection / training создан обученный префикс контрольной точки, который обновляется после определенного номера шага.Вы можете проверить обнаружение объекта, используя текущую обученную модель.

Например:

python3 export_inference_graph.py \
    --input_type image_tensor \
    --pipeline_config_path training/ssd_mobilenet_v1_pets.config \
    --trained_checkpoint_prefix training/model.ckpt-25000 \
    --output_directory latest_dataset

здесь model.ckpt-25000 - это число шагов (25000), обученных до сих пор.