Как добавить фоновые изображения в TensorFlow Object Detection API - PullRequest
Новая
       18

Как добавить фоновые изображения в TensorFlow Object Detection API

1 голос
/ 27 марта 2019

Я пытаюсь добавить фоновые изображения к своим тренировочным данным. Я использую TensorFlow 1.4. Я пытаюсь добавить эти файлы непосредственно в каталог изображений и файл CSV train_labels.csv. Тем не менее, я получаю ошибки, когда я запускаю оба: rapid_rcnn_inception_resnet_v2_atrous_coco.config & ssd_mobilenet_v1_coco.config. Мой train_labels.csv имеет следующие поля:

имя файла, ширина, высота, класс, Xmin, Ymin, Xmax, утах

Для фонового изображения соответствующий ввод: pattern.jpg, 200200, фон ,,,,

Я пытался: 

python3 train.py --logtostderr --train_dir=training/ --pipeline_config_path=training/ssd_mobilenet_v1_coco.config

And got the error: **tensorflow.python.framework.errors_impl.InvalidArgumentError: All bounding box coordinates must be in [0.0, 1.0]: nan**

I have also tried python3 train.py --logtostderr --train_dir=training/ --pipeline_config_path=training/faster_rcnn_inception_resnet_v2_atrous_coco.config

INFO:tensorflow:Recording summary at step 33.
INFO:tensorflow:global step 34: loss = 0.7557 (24.100 sec/step)
INFO:tensorflow:global step 35: loss = 0.7095 (15.436 sec/step)
INFO:tensorflow:global step 36: loss = 1.4553 (16.423 sec/step)
INFO:tensorflow:global step 37: loss = 1.1022 (15.649 sec/step)
INFO:tensorflow:Error reported to Coordinator: <class **'tensorflow.python.framework.errors_impl.InvalidArgumentError'>, LossTensor is inf or nan. : Tensor had NaN values
[[Node: CheckNumerics = CheckNumericsT=DT_FLOAT, message="LossTensor is inf or nan.", _device="/job:localhost/replica:0/task:0/device:CPU:0"]]**

Бот fast-rcnn и ssd работают нормально, если я удаляю фоновые изображения из файла CSV, а также из каталога images. Для своих фоновых изображений я просто использую изображение, которое является полностью белым, поскольку оно является фоном во всех моих тренировочных изображениях. Я совершенно уверен, что неправильно вводю фоновые изображения. Как мне это сделать?

...