tf.image.sample_distorted_bounding_box ValueError

Я пытаюсь дополнить существующее помеченное ограничивающее изображение в рамке для получения дополнительных обучающих данных по обнаружению объектов с помощью функции tf.image.sample_distorted_bounding_box, но я продолжаю обнаруживать эти ошибки здесь. Я почти уверен, что моя ограничительная рамка установлена ​​правильно, потому что она работает, когда я рисую ограничительную рамку.

img = mpimg.imread('bPawn0.jpg')
img = img.reshape(1,300,300,3)
boxes = [100,88,253,209]
box = np.ones([1,1,4])
for i in range(4):
    box[:,:,i] = boxes[i]/300
box = tf.convert_to_tensor(box, np.float32)

begin, size, bbox_for_draw = tf.image.sample_distorted_bounding_box(tf.shape(img),bounding_boxes=box)

ValueError: Tried to convert 'min_object_covered' to a tensor and failed. Error: None values not supported.

Любые предложения относительно того, что я делаю неправильно здесь?


tensorflow valueerror bounding-box
person Justin Francis    schedule 20.11.2017    source источник

Ответы (1)


arrow_upward
0
arrow_downward

Мне интересно, является ли это ошибкой в ​​​​коде sample_distorted_bounding_box(), поскольку я не вижу теста, в котором этот аргумент явно не указывается для функции.

Можете ли вы попробовать установить этот аргумент явно, что-то вроде этого?

sample_distorted_bounding_box = tf.image.sample_distorted_bounding_box(
    tf.shape(image),
    bounding_boxes=bbox,
    min_object_covered=0.1,
    aspect_ratio_range=[0.75, 1.33],
    area_range=[0.05, 1.0],
    max_attempts=100,
    use_image_if_no_bounding_boxes=True)

https://github.com/tensorflow/models/blob/master/research/inception/inception/image_processing.py#L235

person Pete Warden    schedule 05.12.2017
comment
После преобразования изображения обратно в [3] элементы это сработало. Код обновлен здесь. Спасибо! - person Justin Francis; 10.12.2017
comment
Что вы подразумеваете под [3] элементами? Ваша ссылка дает 404 :( - person snazziii; 25.02.2018