Spaces:
Running
Running
File size: 3,291 Bytes
95f8bbc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 |
import os
import sys
import cv2
dir_path = os.path.dirname(os.path.realpath(__file__))
sys.path.insert(0, dir_path)
import ipdb;
pdb = ipdb.set_trace
import argparse
from tqdm import tqdm
from utils import convert
import numpy as np
sys.path.remove(dir_path)
try:
from openpose import pyopenpose as op
except ImportError as e:
print('Error: OpenPose library could not be found. Did you enable `BUILD_PYTHON` in CMake and have this Python script in the right folder?')
raise e
# Flags
parser = argparse.ArgumentParser()
parser.add_argument("--image_path", default="../../examples/media/COCO_val2014_000000000192.jpg",
help="Process an image. Read all standard formats (jpg, png, bmp, etc.).")
args = parser.parse_known_args()
params = dict()
cur_dir = os.path.dirname(os.path.abspath(__file__))
params["model_folder"] = cur_dir + "/models/"
params['tracking'] = 5
params['number_people_max'] = 1
# params['num_gpu'] = 1
# params['num_gpu_start'] = 1
# import ipdb;ipdb.set_trace()
def load_model():
try:
opWrapper = op.WrapperPython()
opWrapper.configure(params)
opWrapper.start()
except Exception as e:
print(e)
sys.exit(-1)
return opWrapper
def test_video(model, video_name=0):
opWrapper = model
cam = cv2.VideoCapture(video_name)
# warm up
for i in range(5):
datum = op.Datum()
_, imageToProcess = cam.read()
datum.cvInputData = imageToProcess
opWrapper.emplaceAndPop([datum])
for i in tqdm(range(2000)):
datum = op.Datum()
_, imageToProcess = cam.read()
datum.cvInputData = imageToProcess
opWrapper.emplaceAndPop([datum])
# Display Image
# print("Body keypoints: \n" + str(datum.poseKeypoints))
# cv2.imshow("OpenPose 1.4.0 - Tutorial Python API", datum.cvOutputData)
# cv2.waitKey(10)
# cv2.destroyAllWindows()
def generate_kpts(video_name):
kpt_results = []
cap = cv2.VideoCapture(video_name)
length = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
opWrapper = load_model()
for i in tqdm(range(length)):
try:
datum = op.Datum()
_, imageToProcess = cap.read()
datum.cvInputData = imageToProcess
opWrapper.emplaceAndPop([datum])
results = datum.poseKeypoints
assert len(results) == 1, 'videopose3D only support one pserson restruction'
# 25 to 17
kpts = convert(results[0])
kpt_results.append(kpts)
except Exception as e:
print(e)
# pose processes
result = np.array(kpt_results)
# # save
# name = '/home/xyliu/experiments/VideoPose3D/data/tmp.npz'
# kpts = result.astype(np.float32)
# print('kpts npz save in ', name)
# np.savez_compressed(name, kpts=kpts)
return result
def generate_frame_kpt(frame, opWrapper):
'''
提供frame and model
'''
datum = op.Datum()
datum.cvInputData = frame
opWrapper.emplaceAndPop([datum])
re = datum.poseKeypoints
assert len(re) == 1, 'videopose3D only support one pserson restruction'
kpt = convert(re[0])
return kpt
if __name__ == "__main__":
generate_kpts(os.environ.get('VIDEO_PATH') + 'dance.mp4')
|