TerribleDev/ente
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[mob][photos] More cleanup

Browse Source
This commit is contained in:
laurenspriem
2024-06-22 00:33:26 +05:30
parent 2855181de9
commit c8effce8fc
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
mobile/lib/services/machine_learning/face_ml/face_embedding/face_embedding_service.dart
View File

@@ -71,13 +71,11 @@ class FaceEmbeddingService {
..setIntraOpNumThreads(1)
..setSessionGraphOptimizationLevel(GraphOptimizationLevel.ortEnableAll);
try {
// _logger.info('Loading face embedding model');
final session =
OrtSession.fromFile(File(args["modelPath"]), sessionOptions);
// _logger.info('Face embedding model loaded');
return session.address;
} catch (e, _) {
// _logger.severe('Face embedding model not loaded', e, s);
} catch (e, s) {
_logger.severe('Face embedding model not loaded', e, s);
}
return -1;
}
@@ -104,7 +102,7 @@ class FaceEmbeddingService {
final int numberOfFaces = input.length ~/ (kInputSize * kInputSize * 3);
final inputOrt = OrtValueTensor.createTensorWithDataList(
input,
[numberOfFaces, kInputSize, kInputSize, 3],
[numberOfFaces, kInputSize, kInputSize, kNumChannels],
);
final inputs = {'img_inputs': inputOrt};
final session = OrtSession.fromAddress(sessionAddress);