search

analogdevicesinc / ai8x-training

Model Training for ADI's MAX78000 and MAX78002 Edge AI Devices
Apache License 2.0
89 stars 80 forks source link

Add improved FaceID model and new datasets #276

Closed oguzhanbsolak closed 9 months ago

oguzhanbsolak commented 10 months ago

FaceID part 2 related updates and items.

Old faceID dataset, and embedding generation scripts are removed. Added, dimensionality reduction with SubcenterArcFace Loss. Updated relationbasedKD policy for 8 bit mode support New MAX78000, and MAX78002 models New combined dataset for Facedetection, FaceID dimensionality reduction, and FaceID tasks

This version includes compression scheduler support for loss optimizer, and it needs to work together with the distiller version at: https://github.com/MaximIntegratedAI/distiller/pull/6

oguzhanbsolak commented 9 months ago

File "datasets/vggface2.py", line 26, in import kornia.geometry.transform as GT ModuleNotFoundError: No module named 'kornia'

Added kornia to requirements.txt.

oguzhanbsolak commented 9 months ago

In addition to the comments, this PR also needs the updated distiller submodule

Updated the distiller version with the latest commit