diff options
author | carp <25677564+carp@users.noreply.github.com> | 2020-07-13 13:40:11 -0400 |
---|---|---|
committer | carp <25677564+carp@users.noreply.github.com> | 2020-07-13 13:40:11 -0400 |
commit | 50e411320563894d411b0c37d37cb16105a908af (patch) | |
tree | 01e1a43e11f9e51cf6a46d9a3d40814b76ad6d7b /anime-face-detector/nms/py_cpu_nms.py | |
parent | f457064bb15a00010959e664492d87f3bfe82537 (diff) | |
download | yaoi-communism-50e411320563894d411b0c37d37cb16105a908af.tar.gz yaoi-communism-50e411320563894d411b0c37d37cb16105a908af.zip |
removing submodule
Diffstat (limited to 'anime-face-detector/nms/py_cpu_nms.py')
-rw-r--r-- | anime-face-detector/nms/py_cpu_nms.py | 38 |
1 files changed, 0 insertions, 38 deletions
diff --git a/anime-face-detector/nms/py_cpu_nms.py b/anime-face-detector/nms/py_cpu_nms.py deleted file mode 100644 index 54e7b25..0000000 --- a/anime-face-detector/nms/py_cpu_nms.py +++ /dev/null @@ -1,38 +0,0 @@ -# -------------------------------------------------------- -# Fast R-CNN -# Copyright (c) 2015 Microsoft -# Licensed under The MIT License [see LICENSE for details] -# Written by Ross Girshick -# -------------------------------------------------------- - -import numpy as np - -def py_cpu_nms(dets, thresh): - """Pure Python NMS baseline.""" - x1 = dets[:, 0] - y1 = dets[:, 1] - x2 = dets[:, 2] - y2 = dets[:, 3] - scores = dets[:, 4] - - areas = (x2 - x1 + 1) * (y2 - y1 + 1) - order = scores.argsort()[::-1] - - keep = [] - while order.size > 0: - i = order[0] - keep.append(i) - xx1 = np.maximum(x1[i], x1[order[1:]]) - yy1 = np.maximum(y1[i], y1[order[1:]]) - xx2 = np.minimum(x2[i], x2[order[1:]]) - yy2 = np.minimum(y2[i], y2[order[1:]]) - - w = np.maximum(0.0, xx2 - xx1 + 1) - h = np.maximum(0.0, yy2 - yy1 + 1) - inter = w * h - ovr = inter / (areas[i] + areas[order[1:]] - inter) - - inds = np.where(ovr <= thresh)[0] - order = order[inds + 1] - - return keep |