Я обучил детектор объектов с помощью Tensorflow 1.15, и теперь я пытаюсь запустить его на отдельных изображениях, чтобы убедиться в его работоспособности. Однако при запуске моего тестового кода я получаю эту ошибку:
InvalidArgumentError: Shape must be rank 4 but is rank 3 for 'DrawBoundingBoxes' (op: 'DrawBoundingBoxes') with input shapes: [235,339,3], [1,300,4].
Это происходит в этой строке кода:
tf.image.draw_bounding_boxes(image, boxes)
Я думаю, что моя проблема заключается в том, что мне нужно либо изменить форму мои коробки или мое изображение, но я не уверен, что мне следует изменить, и как я должен это делать, поскольку я новичок в Tensorflow. Я могу предоставить больше кода при необходимости. Вот ссылка на API обнаружения объектов Tensorflow , за которым я следовал как можно ближе в процессе обучения моей сети.