immich-app/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2025-12-24 09:14:58 +03:00
Code Issues Packages Projects Releases 18 Wiki Activity

feat(server): optimize partial facial recognition (#6634)

Browse Source 
* optimize partial facial recognition

* add tests

* use map

* bulk insert faces
This commit is contained in:
Mert
2024-01-25 01:27:39 -05:00
committed by GitHub
parent 852effa998
commit bd87eb309c
9 changed files with 77 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
server/src/infra/repositories/person.repository.ts
View File

@@ -215,11 +215,11 @@ export class PersonRepository implements IPersonRepository {
return this.personRepository.save(entity);
}
async createFace(entity: AssetFaceEntity): Promise<void> {
if (!entity.embedding) {
throw new Error('Embedding is required to create a face');
}
await this.assetFaceRepository.insert({ ...entity, embedding: () => asVector(entity.embedding, true) });
async createFaces(entities: AssetFaceEntity[]): Promise<string[]> {
const res = await this.assetFaceRepository.insert(
entities.map((entity) => ({ ...entity, embedding: () => asVector(entity.embedding, true) })),
);
return res.identifiers.map((row) => row.id);
}
async update(entity: Partial<PersonEntity>): Promise<PersonEntity> {