【开源自荐】人脸识别和图像分类ToolBox，开箱即用

[wuji3]

推荐项目

• 项目地址：https://github.com/wuji3/visiondk

• 类别：Python 机器学习

• 项目标题：VisionDK: ToolBox Of Image Classification & Face Recognition

• 项目描述：人脸识别和图像分类ToolBox，包含训练、测试和推理可视化，包含诸多强大功能，开箱即用。若希望作为黑盒使用，仅通过一个配置文件就可以启动人脸识别或图像分类模型训练；若希望学习使用，仓库已实现极致解耦，可以熟悉从dataset到training的全部细节。适用于比赛、项目和学习。

• 亮点：训练前，提供数据增强可视化脚本，帮助算法工程师选择适合业务形态的数据增强方法；训练后，提供badcase自动归类、tensorboard和注意力heatmap可视化，帮助算法工程师回溯每一个优化细节。

• 截图：

• 后续更新计划：持续优化仓库，预计上线知识蒸馏，支持移动端小模型优化。欢迎star、fock和pull request