Можете ли вы опубликовать результаты обработки из результатов прогноза Cloud ML?

Question

У меня есть модель для обнаружения объектов (Faster RCNN из API обнаружения объектов Tensorflow), работающая в Google Cloud ML. У меня также есть некоторый код для фильтрации получающихся ограничивающих рамок по размеру, соотношению сторон и т. Д.

1. Возможно ли запустить этот код как часть процесса прогнозирования, поэтому мне не нужно запускать отдельный процесс, чтобы сделать это потом.

2. Можно ли ограничить число ограничивающих рамок, прогнозируемых моделью, на основе некоторого доверительного порога, поскольку в настоящее время выводится много посторонних данных.

Answer 1

1. Вы можете просто добавить логику фильтра в код процесса прогнозирования и развернуть его обратно.
2. Да, вы можете использовать min_score_thresh аргумент в visualize_boxes_and_labels_on_image_array . используйте код ниже

vis_util.visualize_boxes_and_labels_on_image_array(
        image_np,
        np.squeeze(boxes),
        np.squeeze(classes).astype(np.int32),
        np.squeeze(scores),
        category_index,
        use_normalized_coordinates=True,
        line_thickness=4,
        min_score_thresh=.9)

Это приведет к тому, что ограничивающие прямоугольники будут иметь более чем 90% доверительной вероятности. измените его на любое значение от 0 до 1.

Answer 2

Я отвечу (1): у нас есть Alpha API, который позволит это. Пожалуйста, свяжитесь с cloudml-feedback@google.com для получения дополнительной информации.