File size: 3,785 Bytes
79cf6ef |
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 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 |
import cv2
import numpy as np
import matplotlib.pyplot as plt
def vis_detections(im, dets, thresh=0.5, show_text=True):
"""Draw detected bounding boxes."""
class_name = 'face'
inds = np.where(dets[:, -1] >= thresh)[0] if dets is not None else []
if len(inds) == 0:
return
im = im[:, :, (2, 1, 0)]
fig, ax = plt.subplots(figsize=(12, 12))
ax.imshow(im, aspect='equal')
for i in inds:
bbox = dets[i, :4]
score = dets[i, -1]
ax.add_patch(
plt.Rectangle((bbox[0], bbox[1]),
bbox[2] - bbox[0],
bbox[3] - bbox[1], fill=False,
edgecolor='red', linewidth=2.5)
)
if show_text:
ax.text(bbox[0], bbox[1] - 5,
'{:s} {:.3f}'.format(class_name, score),
bbox=dict(facecolor='blue', alpha=0.5),
fontsize=10, color='white')
ax.set_title(('{} detections with '
'p({} | box) >= {:.1f}').format(class_name, class_name,
thresh),
fontsize=10)
plt.axis('off')
plt.tight_layout()
plt.savefig('out.png')
plt.show()
def bbox_vote(det):
order = det[:, 4].ravel().argsort()[::-1]
det = det[order, :]
dets = None
while det.shape[0] > 0:
# IOU
area = (det[:, 2] - det[:, 0] + 1) * (det[:, 3] - det[:, 1] + 1)
xx1 = np.maximum(det[0, 0], det[:, 0])
yy1 = np.maximum(det[0, 1], det[:, 1])
xx2 = np.minimum(det[0, 2], det[:, 2])
yy2 = np.minimum(det[0, 3], det[:, 3])
w = np.maximum(0.0, xx2 - xx1 + 1)
h = np.maximum(0.0, yy2 - yy1 + 1)
inter = w * h
o = inter / (area[0] + area[:] - inter)
# get needed merge det and delete these det
merge_index = np.where(o >= 0.3)[0]
det_accu = det[merge_index, :]
det = np.delete(det, merge_index, 0)
if merge_index.shape[0] <= 1:
continue
det_accu[:, 0:4] = det_accu[:, 0:4] * np.tile(det_accu[:, -1:], (1, 4))
max_score = np.max(det_accu[:, 4])
det_accu_sum = np.zeros((1, 5))
det_accu_sum[:, 0:4] = np.sum(det_accu[:, 0:4], axis=0) / np.sum(det_accu[:, -1:])
det_accu_sum[:, 4] = max_score
try:
dets = np.row_stack((dets, det_accu_sum))
except:
dets = det_accu_sum
if dets is not None:
dets = dets[0:750, :]
return dets
def add_borders(curr_img, target_shape=(224, 224), fill_type=0):
curr_h, curr_w = curr_img.shape[0:2]
shift_h = max(target_shape[0] - curr_h, 0)
shift_w = max(target_shape[1] - curr_w, 0)
image = cv2.copyMakeBorder(curr_img, shift_h // 2, (shift_h + 1) // 2, shift_w // 2, (shift_w + 1) // 2, fill_type)
return image, shift_h, shift_w
def resize_image(image, target_size, resize_factor=None, is_pad=True, interpolation=3):
curr_image_size = image.shape[0:2]
if resize_factor is None and is_pad:
resize_factor = min(target_size[0] / curr_image_size[0], target_size[1] / curr_image_size[1])
elif resize_factor is None and not is_pad:
resize_factor = np.sqrt((target_size[0] * target_size[1]) / (curr_image_size[0] * curr_image_size[1]))
image = cv2.resize(image, None, None, fx=resize_factor, fy=resize_factor, interpolation=interpolation)
if is_pad:
image, shift_h, shift_w = add_borders(image, target_size)
else:
shift_h = shift_w = 0
scale = np.array([image.shape[1]/resize_factor, image.shape[0]/resize_factor,
image.shape[1]/resize_factor, image.shape[0]/resize_factor])
return image, shift_h/image.shape[0]/2, shift_w/image.shape[1]/2, scale |