Как реализовать маску R-CNN?

Попытка выяснить конкретную маску объекта в изображении в Android. Я обучил модуль, используя маску R-CNN, используя тензор потока. Теперь я хочу поместить его в Android и хочу маску изображения.

Обнаружение тензорного объекта в Android доступно по ссылке.

а как получить маску изображения в андроиде???

для обнаружения маски в питоне используется код ниже.

if 'detection_masks' in tensor_dict:
    # The following processing is only for single image
    detection_boxes = tf.squeeze(tensor_dict['detection_boxes'], [0])
    detection_masks = tf.squeeze(tensor_dict['detection_masks'], [0])
    # Reframe is required to translate mask from box coordinates to image coordinates and fit the image size.
    real_num_detection = tf.cast(tensor_dict['num_detections'][0], tf.int32)
    detection_boxes = tf.slice(detection_boxes, [0, 0], [real_num_detection, -1])
    detection_masks = tf.slice(detection_masks, [0, 0, 0], [real_num_detection, -1, -1])
    detection_masks_reframed = utils_ops.reframe_box_masks_to_image_masks(
        detection_masks, detection_boxes, image.shape[0], image.shape[1])
    detection_masks_reframed = tf.cast(
        tf.greater(detection_masks_reframed, 0.5), tf.uint8)
    # Follow the convention by adding back the batch dimension
    tensor_dict['detection_masks'] = tf.expand_dims(detection_masks_reframed, 0)

но как я могу преобразовать его в Android??

может кто-нибудь помочь??

Благодарю.

0 ответов

Другие вопросы по тегам