gitee/face-search
1
0
Fork 0
mirror of https://gitee.com/open-visual/face-search.git synced 2025-07-10 19:37:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

update:PCN压测用例

Browse Source
This commit is contained in:
divenswu 2022-11-30 11:35:53 +08:00
parent 1524934754
commit 6007220504
1 changed files with 47 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
face-search-core/src/test/java/com/visual/face/search/core/test/models/PcnNetworkFaceDetectionStress.java Normal file
View File

@ -0,0 +1,47 @@
package com.visual.face.search.core.test.models;
import ai.onnxruntime.OrtEnvironment;
import com.visual.face.search.core.domain.FaceInfo;
import com.visual.face.search.core.domain.ImageMat;
import com.visual.face.search.core.models.PcnNetworkFaceDetection;
import com.visual.face.search.core.test.base.BaseTest;
import org.opencv.core.Mat;
import org.opencv.imgcodecs.Imgcodecs;
import java.util.List;
import java.util.Map;
public class PcnNetworkFaceDetectionStress extends BaseTest {
static{ nu.pattern.OpenCV.loadShared(); }
private OrtEnvironment env = OrtEnvironment.getEnvironment();
private static String model1Path = "face-search-core/src/main/resources/model/onnx/detection_face_pcn/pcn1_sd.onnx";
private static String model2Path = "face-search-core/src/main/resources/model/onnx/detection_face_pcn/pcn2_sd.onnx";
private static String model3Path = "face-search-core/src/main/resources/model/onnx/detection_face_pcn/pcn3_sd.onnx";
private static String imagePath = "face-search-test/src/main/resources/image/validate/noface";
public static void main(String[] args) {
Map<String, String> map = getImagePathMap(imagePath);
PcnNetworkFaceDetection infer = new PcnNetworkFaceDetection(new String[]{model1Path, model2Path, model3Path}, 1);
int num = 0;
for (int i = 0; i < 10000; i++) {
for (String fileName : map.keySet()) {
num = num + 1;
String imageFilePath = map.get(fileName);
System.out.println(num+":"+imageFilePath);
Mat image = Imgcodecs.imread(imageFilePath);
ImageMat imageMat = ImageMat.fromCVMat(image);
List<FaceInfo> faceInfos = infer.inference(imageMat, PcnNetworkFaceDetection.defScoreTh, PcnNetworkFaceDetection.defIouTh, null);
faceInfos.clear();
imageMat.release();
image.release();
}
}
}
}