VisionDK: ToolBox Of Image Classification & Face Recognition
一个开箱即用的Python机器学习工具箱,专注于人脸识别和图像分类。提供从数据增强可视化到模型训练、测试和推理的全流程工具,支持黑盒配置和模块化学习,适用于比赛、项目和学习场景。
- 项目地址:https://github.com/wuji3/visiondk- 类别:Python 机器学习- 项目标题:VisionDK: ToolBox Of Image Classification & Face Recognition- 项目描述:人脸识别和图像分类ToolBox,包含训练、测试和推理可视化,包含诸多强大功能,开箱即用。若希望作为黑盒使用,仅通过一个配置文件就可以启动人脸识别或图像分类模型训练;若希望学习使用,仓库已实现极致解耦,可以熟悉从dataset到training的全部细节。适用于比赛、项目和学习。- 亮点:训练前,提供数据增强可视化脚本,帮助算法工程师选择适合业务形态的数据增强方法;训练后,提供badcase自动归类、tensorboard和注意力heatmap可视化,帮助算法工程师回溯每一个优化细节。- 截图: