|
|
|
|
|
__author__ = 'cleardusk' |
|
|
|
import numpy as np |
|
import cv2 |
|
from math import sqrt |
|
import matplotlib.pyplot as plt |
|
|
|
RED = (0, 0, 255) |
|
GREEN = (0, 255, 0) |
|
BLUE = (255, 0, 0) |
|
|
|
|
|
def get_suffix(filename): |
|
"""a.jpg -> jpg""" |
|
pos = filename.rfind('.') |
|
if pos == -1: |
|
return '' |
|
return filename[pos:] |
|
|
|
|
|
def crop_img(img, roi_box): |
|
h, w = img.shape[:2] |
|
|
|
sx, sy, ex, ey = [int(round(_)) for _ in roi_box] |
|
dh, dw = ey - sy, ex - sx |
|
if len(img.shape) == 3: |
|
res = np.zeros((dh, dw, 3), dtype=np.uint8) |
|
else: |
|
res = np.zeros((dh, dw), dtype=np.uint8) |
|
if sx < 0: |
|
sx, dsx = 0, -sx |
|
else: |
|
dsx = 0 |
|
|
|
if ex > w: |
|
ex, dex = w, dw - (ex - w) |
|
else: |
|
dex = dw |
|
|
|
if sy < 0: |
|
sy, dsy = 0, -sy |
|
else: |
|
dsy = 0 |
|
|
|
if ey > h: |
|
ey, dey = h, dh - (ey - h) |
|
else: |
|
dey = dh |
|
|
|
res[dsy:dey, dsx:dex] = img[sy:ey, sx:ex] |
|
return res |
|
|
|
|
|
def calc_hypotenuse(pts): |
|
bbox = [min(pts[0, :]), min(pts[1, :]), max(pts[0, :]), max(pts[1, :])] |
|
center = [(bbox[0] + bbox[2]) / 2, (bbox[1] + bbox[3]) / 2] |
|
radius = max(bbox[2] - bbox[0], bbox[3] - bbox[1]) / 2 |
|
bbox = [center[0] - radius, center[1] - radius, center[0] + radius, center[1] + radius] |
|
llength = sqrt((bbox[2] - bbox[0]) ** 2 + (bbox[3] - bbox[1]) ** 2) |
|
return llength / 3 |
|
|
|
|
|
def parse_roi_box_from_landmark(pts): |
|
"""calc roi box from landmark""" |
|
bbox = [min(pts[0, :]), min(pts[1, :]), max(pts[0, :]), max(pts[1, :])] |
|
center = [(bbox[0] + bbox[2]) / 2, (bbox[1] + bbox[3]) / 2] |
|
radius = max(bbox[2] - bbox[0], bbox[3] - bbox[1]) / 2 |
|
bbox = [center[0] - radius, center[1] - radius, center[0] + radius, center[1] + radius] |
|
|
|
llength = sqrt((bbox[2] - bbox[0]) ** 2 + (bbox[3] - bbox[1]) ** 2) |
|
center_x = (bbox[2] + bbox[0]) / 2 |
|
center_y = (bbox[3] + bbox[1]) / 2 |
|
|
|
roi_box = [0] * 4 |
|
roi_box[0] = center_x - llength / 2 |
|
roi_box[1] = center_y - llength / 2 |
|
roi_box[2] = roi_box[0] + llength |
|
roi_box[3] = roi_box[1] + llength |
|
|
|
return roi_box |
|
|
|
|
|
def parse_roi_box_from_bbox(bbox): |
|
left, top, right, bottom = bbox[:4] |
|
old_size = (right - left + bottom - top) / 2 |
|
center_x = right - (right - left) / 2.0 |
|
center_y = bottom - (bottom - top) / 2.0 + old_size * 0.14 |
|
size = int(old_size * 1.58) |
|
|
|
roi_box = [0] * 4 |
|
roi_box[0] = center_x - size / 2 |
|
roi_box[1] = center_y - size / 2 |
|
roi_box[2] = roi_box[0] + size |
|
roi_box[3] = roi_box[1] + size |
|
|
|
return roi_box |
|
|
|
|
|
def plot_image(img): |
|
height, width = img.shape[:2] |
|
plt.figure(figsize=(12, height / width * 12)) |
|
|
|
plt.subplots_adjust(left=0, right=1, top=1, bottom=0) |
|
plt.axis('off') |
|
|
|
plt.imshow(img[..., ::-1]) |
|
plt.show() |
|
|
|
|
|
def draw_landmarks(img, pts, style='fancy', wfp=None, show_flag=False, **kwargs): |
|
"""Draw landmarks using matplotlib""" |
|
height, width = img.shape[:2] |
|
plt.figure(figsize=(12, height / width * 12)) |
|
plt.imshow(img[..., ::-1]) |
|
plt.subplots_adjust(left=0, right=1, top=1, bottom=0) |
|
plt.axis('off') |
|
|
|
dense_flag = kwargs.get('dense_flag') |
|
|
|
if not type(pts) in [tuple, list]: |
|
pts = [pts] |
|
for i in range(len(pts)): |
|
if dense_flag: |
|
plt.plot(pts[i][0, ::6], pts[i][1, ::6], 'o', markersize=0.4, color='c', alpha=0.7) |
|
else: |
|
alpha = 0.8 |
|
markersize = 4 |
|
lw = 1.5 |
|
color = kwargs.get('color', 'w') |
|
markeredgecolor = kwargs.get('markeredgecolor', 'black') |
|
|
|
nums = [0, 17, 22, 27, 31, 36, 42, 48, 60, 68] |
|
|
|
|
|
plot_close = lambda i1, i2: plt.plot([pts[i][0, i1], pts[i][0, i2]], [pts[i][1, i1], pts[i][1, i2]], |
|
color=color, lw=lw, alpha=alpha - 0.1) |
|
plot_close(41, 36) |
|
plot_close(47, 42) |
|
plot_close(59, 48) |
|
plot_close(67, 60) |
|
|
|
for ind in range(len(nums) - 1): |
|
l, r = nums[ind], nums[ind + 1] |
|
plt.plot(pts[i][0, l:r], pts[i][1, l:r], color=color, lw=lw, alpha=alpha - 0.1) |
|
|
|
plt.plot(pts[i][0, l:r], pts[i][1, l:r], marker='o', linestyle='None', markersize=markersize, |
|
color=color, |
|
markeredgecolor=markeredgecolor, alpha=alpha) |
|
if wfp is not None: |
|
plt.savefig(wfp, dpi=150) |
|
print(f'Save visualization result to {wfp}') |
|
|
|
if show_flag: |
|
plt.show() |
|
|
|
|
|
def cv_draw_landmark(img_ori, pts, box=None, color=GREEN, size=1): |
|
img = img_ori.copy() |
|
n = pts.shape[1] |
|
if n <= 106: |
|
for i in range(n): |
|
cv2.circle(img, (int(round(pts[0, i])), int(round(pts[1, i]))), size, color, -1) |
|
else: |
|
sep = 1 |
|
for i in range(0, n, sep): |
|
cv2.circle(img, (int(round(pts[0, i])), int(round(pts[1, i]))), size, color, 1) |
|
|
|
if box is not None: |
|
left, top, right, bottom = np.round(box).astype(np.int32) |
|
left_top = (left, top) |
|
right_top = (right, top) |
|
right_bottom = (right, bottom) |
|
left_bottom = (left, bottom) |
|
cv2.line(img, left_top, right_top, BLUE, 1, cv2.LINE_AA) |
|
cv2.line(img, right_top, right_bottom, BLUE, 1, cv2.LINE_AA) |
|
cv2.line(img, right_bottom, left_bottom, BLUE, 1, cv2.LINE_AA) |
|
cv2.line(img, left_bottom, left_top, BLUE, 1, cv2.LINE_AA) |
|
|
|
return img |
|
|
