Ошибка словаря API обнаружения объектов Tensorflow - PullRequest
0 голосов
/ 04 апреля 2020

Я использую object_detection_tutorial.ipynb для запуска логического вывода.

После запуска логического вывода я получил эту ошибку на output_dict :

  # Run inference
  output_dict = model(input_tensor)

  # All outputs are batches tensors.
  # Convert to numpy arrays, and take index [0] to remove the batch dimension.
  # We're only interested in the first num_detections.
  num_detections = int(output_dict.pop('num_detections'))
  output_dict = {key:value[0, :num_detections].numpy() 
                 for key,value in output_dict.items()}
  output_dict['num_detections'] = num_detections

Журнал ошибок:

Traceback (последний вызов был последним): файл «detect.py», строка 140, в show_inference (обнаружения_модели, image_path), файл «detect.py», строка 109, в show_inference output_dict = run_inference_for_single_image (model, image_np) Файл "detect.py", строка 94, в run_inference_for_single_image num_detections = int (output_dict.pop ('num_detections')) Аргумент TypeError: int () строка, байтовидный объект или число, а не 'Тензор'

1 Ответ

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

Решено обновление до tenorflow 2.0

...