Spaces:
Runtime error
Runtime error
File size: 1,830 Bytes
7652882 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
from matplotlib.patches import Circle
import os
import sys
import matplotlib.pyplot as plt
import pylab
sys.path.append(os.getcwd())
def vis_face(im_array, dets, landmarks, face_size, save_name):
"""Visualize detection results
Parameters:
----------
im_array: numpy.ndarray, shape(1, c, h, w)
test image in rgb
dets1: numpy.ndarray([[x1 y1 x2 y2 score]])
detection results before calibration
dets2: numpy.ndarray([[x1 y1 x2 y2 score]])
detection results after calibration
thresh: float
boxes with scores > thresh will be drawn in red otherwise yellow
Returns:
-------
"""
pylab.imshow(im_array)
for i in range(dets.shape[0]):
bbox = dets[i, :5]
rect = pylab.Rectangle((bbox[0], bbox[1]),
bbox[2] - bbox[0],
bbox[3] - bbox[1], fill=False,
edgecolor='red', linewidth=0.9)
score = bbox[4]
plt.gca().text(bbox[0], bbox[1] - 2,
'{:.5f}'.format(score),
bbox=dict(facecolor='red', alpha=0.5), fontsize=8, color='white')
pylab.gca().add_patch(rect)
if landmarks is not None:
for i in range(landmarks.shape[0]):
landmarks_one = landmarks[i, :]
landmarks_one = landmarks_one.reshape((5, 2))
for j in range(5):
cir1 = Circle(xy=(landmarks_one[j, 0], landmarks_one[j, 1]), radius=face_size/12, alpha=0.4, color="red")
pylab.gca().add_patch(cir1)
#pylab.savefig(save_name)
#只保存图片内容，不保存坐标轴
pylab.axis('off')
pylab.savefig(save_name, bbox_inches='tight', pad_inches=0.0)
pylab.show()
# 返回图片对象
return pylab |