ANSCENTER/ANSCORE
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use yuv and fix FR to use batch 4

Browse Source
This commit is contained in:
Tuan Nghia Nguyen
2026-04-22 16:01:44 +10:00
parent 24d6d2e0a5
commit fc47a42c6a
6 changed files with 65 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules/ANSFR/ANSFaceRecognizer.cpp
View File

@@ -255,8 +255,8 @@ namespace ANSCENTER {
if (engineType == EngineType::NVIDIA_GPU) {
optimizedModelFolder = GetParentFolder(_modelFilePath);
m_options.optBatchSize = 8;
m_options.maxBatchSize = 32;
m_options.optBatchSize = 2;
m_options.maxBatchSize = 4;
m_options.engineFileDir = optimizedModelFolder;
m_options.precision = fp16 ? Precision::FP16 : Precision::FP32;