Skip to content

release v0.0.1

Latest
Compare
Choose a tag to compare
@yakhyo yakhyo released this 31 Aug 14:19
· 25 commits to main since this release

✨ Features

Date Feature Description
- πŸ”„ Updated Pre-trained Models: New weights for existing ResNet (18, 34, 50) and MobileNet (v2, v3) backbones, offering improved accuracy and efficiency.
- πŸš€ Multi-GPU Training Support: Enabled distributed training across multiple GPUs for faster model training and improved scalability.
2024-08-31 🧠 Face Detection Integration: Sample and Computation Redistribution employed for efficient inference and processing in face detection tasks.
2024-08-31 πŸ”„ Pre-trained Models: Support for ResNet (18, 34, 50) and MobileNet (v2, v3) backbones for feature extraction and optimized performance across diverse applications.
2024-08-31 🎯 Head Pose Estimation: Enhanced model architecture for precise head pose estimation, utilizing pre-trained backbones for robust and efficient performance.

Evaluation results on AFLW2000

Model Size Best Epochs Yaw Pitch Roll MAE
ResNet-18 43 MB 53 4.5027 5.8261 4.2188 4.8492
ResNet-34 81.6 MB 73 4.4538 5.2690 3.8855 4.5361
ResNet-50 91.3 MB 77 3.5529 4.9962 3.4986 4.0159
MobileNet V2 9.59 MB 73 5.6880 6.0391 4.4433 5.3901
MobileNet V3 small 6 MB 17 8.6926 7.7089 6.0035 7.4683
MobileNet V3 large 17 MB 71 5.6068 6.6022 4.9959 5.7350