Я использую API обнаружения объектов Tensorflow для оценки модели SSD_mobilenet в наборе данных OpenImageV5. Я использовал предустановленную модель, представленную в их модельный зоопарк.
Для конфигурации конвейера я использую конвейер по умолчанию (ssd_mobilenet_v2_oid_v4
), доступный здесь.
Когда я запускаю оценку, используя следующую команду:
python object_detection/model_main.py --alsologtostderr --run_once --checkpoint_dir=/Path/TO/ssd_mobilenet_v2_oid_v4_2018_12_12/ --model_dir=eval/ --pipeline_config_path=PATH/TO/ssd_mobilenet_v2_oid_v4.config
Я получаю эту ошибку:
2019-05-28 12:26:09.658946: W tensorflow/core/framework/op_kernel.cc:1306] Invalid argument: ValueError: Image with id 1407414044 already added.
Traceback (most recent call last):
File "/bigSSD/tensorflow_model/v_env_tf_1.8/lib/python3.6/site-packages/tensorflow/python/ops/script_ops.py", line 157, in __call__
ret = func(*args)
File "/bigSSD/tensorflow_model/models/research/object_detection/utils/object_detection_evaluation.py", line 432, in update_op
self.add_single_ground_truth_image_info(image_id, single_example_dict)
File "/bigSSD/tensorflow_model/models/research/object_detection/utils/object_detection_evaluation.py", line 600, in add_single_ground_truth_image_info
raise ValueError('Image with id {} already added.'.format(image_id))
ValueError: Image with id 1407414044 already added.
Я изменил sample_1_of_n_examples
с 10 на 100 и 1000, но все равно получил ту же ошибку.
num_examples
устарел, и у меня его нет в файле конфигурации. Я использовал образец файла конфигурации, предоставленный API обнаружения объектов (здесь).
для более быстрой_rcnn_inception_resnet_v2_atrous_oid_v4.config я изменил
eval_config
{
metrics_set: "open_images_V2_detection_metrics"
}
к
eval_config {
metrics_set: "open_images_V2_detection_metrics"
num_examples: 383 #correspond to my test images
}
затем запустите оценку с помощью eval.py, используя следующую команду:
python eval.py --logtostderr --checkpoint_dir=path/to/checkpoint --eval_dir=path/to/eval_dir --pipeline_config_path=path/to/pretrained_model.config
и моя ошибка решена.
не могли бы вы вставить сюда свой файл конфигурации, особенно в
eval_config
,num_examples
должен быть равен размеру вашего оценочного набора данных.