EfficientDet-Custom Dataset -StringToNumberOp не может правильно преобразовать строку

Question

Как новичок, я пытаюсь обучить свои собственные наборы данных с помощью TensorFlow, но при запуске обучения получаю следующую ошибку:

вот моя командная строка:

python main.py --mode=train_and_eval
--training_file_pattern=tfrecord/train.record
--validation_file_pattern=tfrecord/test.record
--model_name=efficientdet-d0
--model_dir=/tmp/efficientdet-d0-scratch
--backbone_ckpt=efficientnet-b0
--train_batch_size=8
--eval_batch_size=8 --eval_samples=512
--num_examples_per_epoch=5717 --num_epochs=1
--hparams="num_classes=4,moving_average_decay=0"
--use_tpu=False

Answer 1

Ответил здесь на github. Это проблема с созданием tfrecord. В вашем скрипте создания tf-записи измените source_id

'image/source_id': dataset_util.bytes_feature(input_image_filename.encode('utf8')),

к

'image/source_id': dataset_util.bytes_feature('0'.encode('utf8')),

Вам не нужно изменять имена файлов. Оригинальная ссылка ниже.

https://github.com/google/automl/issues/307