jochen_deel.py

import sys
import cv2
import common
from urbanscene import features
import logging
import matching
from matplotlib import pyplot as plt
import timer
logging.basicConfig(level=logging.DEBUG)
logger = logging.getLogger("main")
import plot_vars


feature_name = 'orb-flann'
#path_img = '../img/galabal2018/fotos/'    #'../img/'  #'posesGeoteam/fotos/'
#path_json = '../img/galabal2018/json/' #'../json_data/'   #'posesGeoteam/json/'
schijfnaam = "/media/jochen/2FCA69D53AB1BFF43/"
poses = schijfnaam+'dataset/kever/poses/'
urban_json = schijfnaam+'dataset/kever/json/'
urban_fotos = schijfnaam+'dataset/kever/fotos/'


path_img = urban_fotos  # 'posesGeoteam/fotos/'
path_json = urban_json  # 'posesGeoteam/json/'

#path_img = "../img/"
#path_json = '../json_data/'  # 'posesGeoteam/json/'

model_name = '17.jpg' #'bulb14.jpg'  # goeie : "pisa9"  taj3  # trap1     trap1
input_name = '70.jpg' #'bulb16.jpg' # goeie : "pisa10"  taj4  # trap2     trap3

#model_name="duo21.jpg"
#input_name="duo39.jpg"

model_image = cv2.imread(path_img + model_name, cv2.IMREAD_GRAYSCALE)
input_image = cv2.imread(path_img + input_name, cv2.IMREAD_GRAYSCALE)

detector, matcher = features.init_feature(feature_name)

if model_image is None:
    print('Failed to load fn1:', model_name)
    sys.exit(1)

if input_image is None:
    print('Failed to load fn2:', input_image)
    sys.exit(1)

if detector is None:
    print('unknown feature:', feature_name)
    sys.exit(1)

logger.debug(" using %s", feature_name)

include_keypoints = False
plot_us = True  # plot urban scene
plot_mp = False # plot multi pose
plot_vars.input_name = input_name
plot_vars.model_name = model_name
plot_vars.model_path = path_img + model_name
plot_vars.input_path = path_img + input_name
plot_vars.write_img = False
global crop
global correction
crop = True
correction = True


logger.debug("---Starting pose matching --")
if include_keypoints:
    model_pose_features = common.parse_JSON_multi_person(path_json + model_name.split('.')[0] + '_keypoints' +  '.json')  # + '_keypoints'
    input_pose_features = common.parse_JSON_multi_person(path_json + input_name.split('.')[0]  +'_keypoints' + '.json')
else:
    model_pose_features = common.parse_JSON_multi_person(path_json + model_name.split('.')[0].replace('fotos','json') +  '.json')  # + '_keypoints'
    input_pose_features = common.parse_JSON_multi_person(path_json + input_name.split('.')[0].replace('fotos','json')  +'.json')



result_whole = matching.match_whole(model_pose_features, input_pose_features, detector, matcher, model_image, input_image,plot_us, plot_mp)

if plot_us or plot_mp:
    plt.show()