def pre_process_images(list_folders_scores, recalculate):
print('Start pre_processing images')
# Loading instances
instance_pose = ClassOpenPose()
for folder, min_score in list_folders_scores:
for file in os.listdir(folder):
full_path = os.path.join(folder, file)
extension = ClassUtils.get_filename_extension(full_path)
if extension != '.jpg':
print('Ignoring file {0}'.format(full_path))
continue
file_no_ext = ClassUtils.get_filename_no_extension(full_path)
arr_file_name = os.path.join(folder, '{0}.json'.format(file_no_ext))
# If image recalculation
if not recalculate:
if os.path.isfile(arr_file_name):
print('File already processed {0}'.format(full_path))
continue
# Processing file
print('Processing file {0}'.format(full_path))
image = cv2.imread(full_path)
arr, img_draw = instance_pose.recognize_image_tuple(image)
arr_pass = list()
# Checking vector integrity for all elements
# Verify there is at least one arm and one leg
for elem in arr:
if ClassUtils.check_vector_integrity_part(elem, min_score):
arr_pass.append(elem)
# If there is more than one person with vector integrity
if len(arr_pass) != 1:
for elem in arr_pass:
pt1, pt2 = ClassUtils.get_rectangle_bounds(elem, ClassUtils.MIN_POSE_SCORE)
cv2.rectangle(img_draw, pt1, pt2, (0, 0, 255), 3)
cv2.namedWindow('main_window')
cv2.imshow('main_window', img_draw)
print(arr)
print(arr_pass)
cv2.waitKey(0)
cv2.destroyAllWindows()
raise Exception('Invalid len: {0} file {1}'.format(len(arr_pass), full_path))
person_arr = arr_pass[0]
arr_str = json.dumps(person_arr.tolist())
with open(arr_file_name, 'w') as text_file:
text_file.write(arr_str)
print('Done!')
def test_color_compare_hist(perform_eq=False):
print('Test color comparision')
print('Loading image comparision')
# Loading instances
instance_pose = ClassOpenPose()
ignore_json_color = False
if perform_eq:
ignore_json_color = True
# Avoid to open two prompts
obj_img = ClassDescriptors.load_images_comparision_ext(
instance_pose,
min_score,
load_one_img=True,
perform_eq=perform_eq,
ignore_json_color=ignore_json_color)
# Extract color comparision from image
hist1 = obj_img['listPoints1']
# Generating examples folder
list_process = list()
for root, _, files in os.walk(EXAMPLES_FOLDER):
for file in files:
full_path = os.path.join(root, file)
extension = ClassUtils.get_filename_extension(full_path)
if extension == '.jpg':
list_process.append(full_path)
# Sorting list
list_process.sort()
"""
list_result = list()
score_max_pt = -1
"""
for full_path in list_process:
print('Processing file: {0}'.format(full_path))
json_path = full_path.replace('.jpg', '.json')
with open(json_path, 'r') as f:
obj_json = json.loads(f.read())
image2 = cv2.imread(full_path)
if perform_eq:
image2 = ClassUtils.equalize_hist(image2)
if 'vector' in obj_json:
pose2 = obj_json['vector']
elif 'vectors' in obj_json:
pose2 = obj_json['vectors']
else:
raise Exception('Invalid vector property for vector custom')
hist2 = ClassDescriptors.get_points_by_pose(image2, pose2, min_score)
diff = ClassDescriptors.get_kmeans_diff(hist1, hist2)
# Getting mean y from image 2 - discarding purposes
pt1, pt2 = ClassUtils.get_rectangle_bounds(pose2, min_score)
image2_crop = image2[pt1[1]:pt2[1], pt1[0]:pt2[0]]
image2_ycc = cv2.cvtColor(image2_crop, cv2.COLOR_BGR2YCrCb)
mean_y = np.mean(image2_ycc[:, :, 0])
print('Diff color: {0} - Mean y: {1}'.format(diff, mean_y))
"""
list_result.sort(key=lambda x: x['score'])
print('Printing list result')
print(list_result)
print('min_score: {0}'.format(score_max_pt))
"""
print('Done!')
def main():
print('Initializing main function')
# Initializing instances
instance_pose = ClassOpenPose()
# Withdrawing tkinter
Tk().withdraw()
# Loading folder images
folder_images = '/home/mauricio/PosesProcessed/folder_images'
folder_images_draw = '/home/mauricio/PosesProcessed/folder_images_draw'
pose_base_folder = '/home/mauricio/Pictures/PosesNew'
# Reading all elements in list
list_files = sorted(os.listdir(folder_images_draw))
cv2.namedWindow('main_window', cv2.WND_PROP_AUTOSIZE)
index = 0
while True:
file = list_files[index]
full_path_draw = os.path.join(folder_images_draw, file)
image_draw = cv2.imread(full_path_draw)
cv2.imshow('main_window', image_draw)
key = cv2.waitKey(0)
print('Key pressed: {0}'.format(key))
if key == 52:
# Left arrow
index -= 1
if index < 0:
index = 0
elif key == 54:
# Right arrow
index += 1
if index == len(list_files):
index = len(list_files) - 1
elif key == 27:
# Esc
# Ask are you sure question
result = messagebox.askyesno("Exit", "Are you sure to exit?")
print(result)
if result:
break
elif key == 115:
# Check JSON file
image_full_path = os.path.join(folder_images, file)
json_path = image_full_path.replace(".jpg", ".json")
if not os.path.exists(json_path):
# Generating JSON array
image = cv2.imread(image_full_path)
arr = instance_pose.recognize_image(image).tolist()
else:
with open(json_path, 'r') as file:
arr_str = file.read()
arr = json.load(arr_str)
arr_pass = []
min_score = 0.05
# Drawing candidates
for elem in arr:
if ClassUtils.check_vector_integrity_part(elem, min_score):
arr_pass.append(elem)
if len(arr_pass) == 0:
print('Arr pass size equals to zero')
continue
elif len(arr_pass) == 1:
selection_int = 0
person_arr = arr_pass[0]
arr_str = json.dumps(person_arr)
else:
# Draw rectangles for all candidates
for index_elem, person_arr in enumerate(arr_pass):
pt1, pt2 = ClassUtils.get_rectangle_bounds(person_arr, min_score)
cv2.rectangle(image_draw, pt1, pt2, (0, 0, 255), 5)
font = cv2.FONT_HERSHEY_SIMPLEX
bottom_left_corner = pt2
font_scale = 0.6
font_color = (255, 255, 255)
line_type = 2
cv2.putText(image_draw, '{0}'.format(index_elem),
bottom_left_corner,
font,
font_scale,
font_color,
line_type)
while True:
print('Select image to put')
cv2.imshow('main_window', image_draw)
wait_key = cv2.waitKey(0)
print('Wait Key: {0}'.format(wait_key))
selection_int = ClassUtils.key_to_number(wait_key)
print('Selection: {0}'.format(selection_int))
if 0 <= selection_int < len(arr_pass):
break
person_arr = arr_pass[selection_int]
arr_str = json.dumps(person_arr)
# Getting new image using numpy slicing
pt1, pt2 = ClassUtils.get_rectangle_bounds(person_arr, min_score)
image_draw = image_draw[pt1[1]:pt2[1], pt1[0]:pt2[0]]
# Selecting directory after processing image!
init_dir = '/home/mauricio/Pictures/PosesNew'
options = {'initialdir': init_dir}
dir_name = filedialog.askdirectory(**options)
if not dir_name:
print('Directory not selected')
else:
new_name = ClassUtils.get_filename_no_extension(file)
new_name = "{0}_{1}.{2}".format(new_name, selection_int, 'jpg')
new_image_path = os.path.join(dir_name, new_name)
new_json_path = os.path.join(dir_name, new_name.replace('.jpg', '.json'))
cv2.imwrite(new_image_path, image_draw)
with open(new_json_path, 'w') as file:
file.write(arr_str)
print('File copied from {0} to {1}'.format(full_path_draw, new_image_path))
index += 1
if index == len(list_files):
index = len(list_files) - 1
cv2.destroyAllWindows()
print('Done!')
def main():
print('Initializing main function')
print('Folder selection')
folder_images = '/home/mauricio/PosesProcessed/folder_images'
folder_images_draw = '/home/mauricio/PosesProcessed/folder_images_draw'
Tk().withdraw()
# Getting options
init_dir = '/home/mauricio/Videos/Oviedo'
options = {'initialdir': init_dir}
dir_name = askdirectory(**options)
if not dir_name:
raise Exception('Directory not selected')
# Create directory if does not exists
if not os.path.isdir(folder_images):
os.makedirs(folder_images)
# Create directory if does not exists
if not os.path.isdir(folder_images_draw):
os.makedirs(folder_images_draw)
# Initializing openpose instance
instance_pose = ClassOpenPose()
for root, subdirs, files in os.walk(dir_name):
for file in files:
full_path = os.path.join(root, file)
print('Processing {0}'.format(full_path))
extension = ClassUtils.get_filename_extension(full_path)
if extension == '.mjpeg':
file_info = ClassMjpegReader.process_video(full_path)
else:
print('Extension ignored: {0}'.format(extension))
continue
# Getting idcam from file
id_cam = full_path.split('/')[-2]
print('IdCam: {0}'.format(id_cam))
for index, info in enumerate(file_info):
print('Processing {0} of {1} from {2}'.format(index, len(file_info), full_path))
frame = info[0]
ticks = info[1]
image_np = np.frombuffer(frame, dtype=np.uint8)
image = cv2.imdecode(image_np, cv2.IMREAD_ANYCOLOR)
arr, image_draw = instance_pose.recognize_image_tuple(image)
min_score = 0.05
arr_pass = list()
for elem in arr:
if ClassUtils.check_vector_integrity_pos(elem, min_score):
arr_pass.append(elem)
if len(arr_pass) > 0:
# Draw rectangles for all candidates
for person_arr in arr_pass:
pt1, pt2 = ClassUtils.get_rectangle_bounds(person_arr, min_score)
cv2.rectangle(image_draw, pt1, pt2, (0, 0, 255), 5)
# Overwriting 1
full_path_images = os.path.join(folder_images, '{0}_{1}.jpg'.format(ticks, id_cam))
print('Writing image {0}'.format(full_path_images))
cv2.imwrite(full_path_images, image)
# Overwriting 2
full_path_draw = os.path.join(folder_images_draw, '{0}_{1}.jpg'.format(ticks, id_cam))
print('Writing image {0}'.format(full_path_images))
cv2.imwrite(full_path_draw, image_draw)
print('Done!')