| | |
| | |
| | import sys |
| | import cv2 |
| | import os |
| | from sys import platform |
| | import argparse |
| |
|
| | 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_000000000192.jpg", help="Process an image. Read all standard formats (jpg, png, bmp, etc.).") |
| | args = parser.parse_known_args() |
| |
|
| | |
| | params = dict() |
| | params["model_folder"] = "../../../models/" |
| |
|
| | |
| | for i in range(0, len(args[1])): |
| | curr_item = args[1][i] |
| | if i != len(args[1])-1: next_item = args[1][i+1] |
| | else: next_item = "1" |
| | if "--" in curr_item and "--" in next_item: |
| | key = curr_item.replace('-','') |
| | if key not in params: params[key] = "1" |
| | elif "--" in curr_item and "--" not in next_item: |
| | key = curr_item.replace('-','') |
| | if key not in params: params[key] = next_item |
| |
|
| | |
| | |
| | |
| |
|
| | |
| | opWrapper = op.WrapperPython(op.ThreadManagerMode.Synchronous) |
| | opWrapper.configure(params) |
| | opWrapper.execute() |
| | except Exception as e: |
| | print(e) |
| | sys.exit(-1) |
| |
|