|
|
|
|
|
import sys |
|
import cv2 |
|
import os |
|
from sys import platform |
|
import argparse |
|
import numpy as np |
|
|
|
try: |
|
|
|
dir_path = os.path.dirname(os.path.realpath(__file__)) |
|
try: |
|
|
|
if platform == "win32": |
|
|
|
sys.path.append(dir_path + '/../../python/openpose/Release'); |
|
os.environ['PATH'] = os.environ['PATH'] + ';' + dir_path + '/../../x64/Release;' + dir_path + '/../../bin;' |
|
import pyopenpose as op |
|
else: |
|
|
|
sys.path.append('../../python'); |
|
|
|
|
|
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 |
|
|
|
|
|
parser = argparse.ArgumentParser() |
|
parser.add_argument("--image_path", default="../../../examples/media/COCO_val2014_000000000294.jpg", help="Process an image. Read all standard formats (jpg, png, bmp, etc.).") |
|
args = parser.parse_known_args() |
|
|
|
|
|
imageToProcess = cv2.imread(args[0].image_path) |
|
|
|
def get_sample_heatmaps(): |
|
|
|
params = dict() |
|
params["model_folder"] = "../../../models/" |
|
params["heatmaps_add_parts"] = True |
|
params["heatmaps_add_bkg"] = True |
|
params["heatmaps_add_PAFs"] = True |
|
params["heatmaps_scale"] = 3 |
|
params["upsampling_ratio"] = 1 |
|
params["body"] = 1 |
|
|
|
|
|
opWrapper = op.WrapperPython() |
|
opWrapper.configure(params) |
|
opWrapper.start() |
|
|
|
|
|
datum = op.Datum() |
|
imageToProcess = cv2.imread(args[0].image_path) |
|
datum.cvInputData = imageToProcess |
|
opWrapper.emplaceAndPop(op.VectorDatum([datum])) |
|
poseHeatMaps = datum.poseHeatMaps.copy() |
|
opWrapper.stop() |
|
|
|
return poseHeatMaps |
|
|
|
|
|
poseHeatMaps = get_sample_heatmaps() |
|
|
|
|
|
params = dict() |
|
params["model_folder"] = "../../../models/" |
|
params["body"] = 2 |
|
params["upsampling_ratio"] = 0 |
|
opWrapper = op.WrapperPython() |
|
opWrapper.configure(params) |
|
opWrapper.start() |
|
|
|
|
|
datum = op.Datum() |
|
datum.cvInputData = imageToProcess |
|
datum.poseNetOutput = poseHeatMaps |
|
opWrapper.emplaceAndPop(op.VectorDatum([datum])) |
|
|
|
|
|
print("Body keypoints: \n" + str(datum.poseKeypoints)) |
|
cv2.imshow("OpenPose 1.7.0 - Tutorial Python API", datum.cvOutputData) |
|
cv2.waitKey(0) |
|
except Exception as e: |
|
print(e) |
|
sys.exit(-1) |
|
|