test/AIAS
1
0
Fork 0
mirror of https://gitee.com/mymagicpower/AIAS.git synced 2024-11-29 18:58:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

no message

Browse Source
This commit is contained in:
Calvin 2022-11-01 16:59:43 +08:00
parent 8d30de3c53
commit 6850d3347b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
4_video_sdks/rtsp_face_sdk/src/main/java/me/aias/example/utils/FaceDetection.java
View File

@ -77,7 +77,7 @@ public final class FaceDetection {
Shape shape = array.getShape();
array =
NDImageUtils.resize(array, (int) (shape.get(1) * shrink), (int) (shape.get(0) * shrink));
array = array.transpose(2, 0, 1).flip(0); // HWC -> CHW BGR -> RGB
array = array.transpose(2, 0, 1).flip(0); // HWC -> CHW RGB -> BGR
NDArray mean = manager.create(new float[] {104f, 117f, 123f}, new Shape(3, 1, 1));
array = array.sub(mean).mul(0.007843f); // normalization
array = array.expandDims(0); // make batch dimension