mirror of
https://github.com/deepinsight/insightface.git
synced 2025-12-30 08:02:27 +00:00
17 lines
458 B
Python
17 lines
458 B
Python
import numpy as np
|
|
|
|
|
|
def unique_boxes(boxes, scale=1.0):
|
|
""" return indices of unique boxes """
|
|
v = np.array([1, 1e3, 1e6, 1e9])
|
|
hashes = np.round(boxes * scale).dot(v).astype(np.int)
|
|
_, index = np.unique(hashes, return_index=True)
|
|
return np.sort(index)
|
|
|
|
|
|
def filter_small_boxes(boxes, min_size):
|
|
w = boxes[:, 2] - boxes[:, 0]
|
|
h = boxes[:, 3] - boxes[:, 1]
|
|
keep = np.where((w >= min_size) & (h > min_size))[0]
|
|
return keep
|