diff --git a/rtmlib/tools/solution/body.py b/rtmlib/tools/solution/body.py index 9d152de..dcfc5fe 100644 --- a/rtmlib/tools/solution/body.py +++ b/rtmlib/tools/solution/body.py @@ -1,3 +1,45 @@ +''' +Example: + +import cv2 + +from rtmlib import Body, draw_skeleton + +device = 'cuda' +backend = 'onnxruntime' # opencv, onnxruntime + +cap = cv2.VideoCapture('./demo.mp4') + +openpose_skeleton = True # True for openpose-style, False for mmpose-style + +body = Body(to_openpose=openpose_skeleton, + backend=backend, + device=device) + +frame_idx = 0 + +while cap.isOpened(): + success, frame = cap.read() + frame_idx += 1 + + if not success: + break + + keypoints, scores = body(frame) + + img_show = frame.copy() + + img_show = draw_skeleton(img_show, + keypoints, + scores, + openpose_skeleton=openpose_skeleton, + kpt_thr=0.43) + + img_show = cv2.resize(img_show, (960, 540)) + cv2.imshow('img', img_show) + cv2.waitKey(10) + +''' import numpy as np from .. import YOLOX, RTMPose diff --git a/rtmlib/tools/solution/wholebody.py b/rtmlib/tools/solution/wholebody.py index 187fae0..ac62449 100644 --- a/rtmlib/tools/solution/wholebody.py +++ b/rtmlib/tools/solution/wholebody.py @@ -1,3 +1,45 @@ +''' +Example: + +import cv2 + +from rtmlib import Wholebody, draw_skeleton + +device = 'cuda' +backend = 'onnxruntime' # opencv, onnxruntime + +cap = cv2.VideoCapture('./demo.mp4') + +openpose_skeleton = True # True for openpose-style, False for mmpose-style + +wholebody = Wholebody(to_openpose=openpose_skeleton, + backend=backend, + device=device) + +frame_idx = 0 + +while cap.isOpened(): + success, frame = cap.read() + frame_idx += 1 + + if not success: + break + + keypoints, scores = wholebody(frame) + + img_show = frame.copy() + + img_show = draw_skeleton(img_show, + keypoints, + scores, + openpose_skeleton=openpose_skeleton, + kpt_thr=0.43) + + img_show = cv2.resize(img_show, (960, 540)) + cv2.imshow('img', img_show) + cv2.waitKey(10) + +''' from typing import List, Optional import numpy as np