Comments (1)
I will suggest you change the current nms with this
def faster_nms(boxes, scores, nms_thresh):
x1 = boxes[:, 0]
y1 = boxes[:, 1]
x2 = boxes[:, 2]
y2 = boxes[:, 3]
areas = (x2 - x1 + 1) * (y2 - y1 + 1)
order = np.argsort(scores)[::-1]
num_detections = boxes.shape[0]
suppressed = np.zeros((num_detections,), dtype=np.bool)
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 <= nms_thresh)[0]
order = order[inds + 1]
return keep
from centerface.
Related Issues (20)
- landmark的groundtruth问题
- Don't support type Convolution on GPU, I need help?
- 请问 关于pytorch的代码有计划开源么,谢谢
- Test on widerface but stop at the third image. HOT 3
- How to get Landmarks Confidence
- Window10 onnx->tensorrt centerface
- [TensorRT] ERROR: ../rtSafe/safeRuntime.cpp (25) - Cuda Error in allocate: 2 (out of memory) --->Centerface-tensorrt
- 在哪里训练呢?还是只能使用现有的模型? HOT 1
- prj-opencv-cpp中C++调用的问题 HOT 1
- CMake error: include_directories(${CMAKE_CURRENT_LIST_DIR}/cpp)d
- how to find model_path in cpp version
- Video predictive acceleration HOT 1
- how to set minimum face size HOT 1
- 连续进行多张图片的人脸检测时,推理结果和单张图片检测的结果不一致 HOT 1
- How to preprocess image before using model with onnxruntime?
- tensorrt7 generate engine from onnx, output is not equal onnx output
- 不同任务的loss大小问题
- CenterFace-small指的是什么 HOT 1
- How to use it when image orientation is portrait?
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from centerface.