Python 媒体识别包 Mediapipe - 物体空间（Objectron）

发表于 更新于 分类于

物体空间检测（Objectron）

import cv2
import mediapipe as mp

mp_drawing = mp.solutions.drawing_utils
mp_objectron = mp.solutions.objectron

with mp_objectron.Objectron(
    static_image_mode=False,
    max_num_objects=5,
    min_detection_confidence=0.5,
    min_tracking_confidence=0.5,
    model_name='Shoe'
) as objectron:
    cap = cv2.VideoCapture(0)
    while cap.isOpened():
        success, img = cap.read()
        if success:
            results = objectron.process(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))

            if results.detected_objects is not None:
                for obj in results.detected_objects:
                    mp_drawing.draw_landmarks(
                        img, obj.landmarks_2d, mp_objectron.BOX_CONNECTIONS)
                    mp_drawing.draw_axis(img, obj.rotation, obj.translation)

        cv2.imshow('物体空间识别', img)
        if cv2.waitKey(1) > 0:
            break

    cap.release()

其它功能