aboutsummaryrefslogtreecommitdiffstats
path: root/anime-face-detector/nms/py_cpu_nms.py
diff options
context:
space:
mode:
authorcarp <25677564+carp@users.noreply.github.com>2020-07-13 13:40:11 -0400
committercarp <25677564+carp@users.noreply.github.com>2020-07-13 13:40:11 -0400
commit50e411320563894d411b0c37d37cb16105a908af (patch)
tree01e1a43e11f9e51cf6a46d9a3d40814b76ad6d7b /anime-face-detector/nms/py_cpu_nms.py
parentf457064bb15a00010959e664492d87f3bfe82537 (diff)
downloadyaoi-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.py38
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