コード例 #1
0
-------------------------------- UNVALID MODELPOSES TEST CASE -------------------------------------------
Testjes voor unvalid model poses (single person)

'''

logging.basicConfig(level=logging.DEBUG)
logger = logging.getLogger("pose_match")
json_data_path = 'data/json_data/'
images_data_path = 'data/image_data/'

model = "1027"
input = "846"
model_json = json_data_path + model + '.json'
input_json = json_data_path + input + '.json'
model_image = images_data_path + model + '.png'
input_image = images_data_path + input + '.png'
model_features = parse_openpose_json.parse_JSON_single_person(model_json)
input_features = parse_openpose_json.parse_JSON_single_person(input_json)

model_features[2] = [0, 0]

# With valid input and valid model
match_result = pose_match.single_person(model_features, input_features, True)
logger.info("--Match or not: %s ", str(match_result.match_bool))

pose_match.plot_single_person(
    model_features, input_features, model_image, input_image, "valid input",
    "unvalid model", "match result: " + str(match_result.match_bool) + "(" +
    str(round(match_result.error_score, 5)) + ")")

plt.show()
コード例 #2
0
ファイル: test.py プロジェクト: hollylxj/SkeletonMatching

json_data_path = 'data/json_data/'
images_data_path = 'data/image_data/'

json_data_path = 'data/json_data/'
images_data_path = 'data/image_data/'

model = "coach"
input = "right"
model_json = json_data_path + model + '.json'
input_json = json_data_path + input + '.json'

model_image = images_data_path + model + '.png'
input_image = images_data_path + input + '.png'

model_features = parse_openpose_json.parse_JSON_single_person(model_json)
input_features = parse_openpose_json.parse_JSON_single_person(input_json)
# model_features = parse_openpose_json.parse_JSON_single_person('data/json_data/foto1.json')
# input_features = parse_openpose_json.parse_JSON_single_person('data/json_data/model1.json')



match_result = pose_match.single_person(model_features, input_features, True)
model_features = parse_openpose_json.parse_JSON_single_person(model_json)
input_features = parse_openpose_json.parse_JSON_single_person(input_json)
pose_match.plot_single_person(model_features, input_features, model_image, input_image)

#plt.show()

コード例 #3
0
def plot(model_points, input_features, model_image, input_image, torso_value):
    im = pose_match.plot_single_person(model_points, input_features,
                                       model_image, input_image, torso_value)
    #plt.show()
    return im