1# Tencent is pleased to support the open source community by making ncnn available. 2# 3# Copyright (C) 2020 THL A29 Limited, a Tencent company. All rights reserved. 4# 5# Licensed under the BSD 3-Clause License (the "License"); you may not use this file except 6# in compliance with the License. You may obtain a copy of the License at 7# 8# https://opensource.org/licenses/BSD-3-Clause 9# 10# Unless required by applicable law or agreed to in writing, software distributed 11# under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR 12# CONDITIONS OF ANY KIND, either express or implied. See the License for the 13# specific language governing permissions and limitations under the License. 14 15import sys 16import cv2 17import numpy as np 18import ncnn 19from ncnn.model_zoo import get_model 20from ncnn.utils import draw_pose 21 22if __name__ == "__main__": 23 if len(sys.argv) != 2: 24 print("Usage: %s [imagepath]\n" % (sys.argv[0])) 25 sys.exit(0) 26 27 imagepath = sys.argv[1] 28 29 m = cv2.imread(imagepath) 30 if m is None: 31 print("cv2.imread %s failed\n" % (imagepath)) 32 sys.exit(0) 33 34 net = get_model("simplepose", num_threads=4, use_gpu=True) 35 36 keypoints = net(m) 37 38 draw_pose(m, keypoints) 39