EASY EAI NetHub is a repository aiming to help build and train model.
EASY EAI NetHub 是一个神经网络模型的搭建及训练的仓库,旨在收纳适用于边缘端进行前向预测的模型。主要优化 模块复用、数据集迁移、回溯训练记录 、调试参数 等方面的体验。主要包含 图像分类、物体检测、图像分割、关键点定位 等模型类别。不定期更新新模型的训练配置及效果对比,通过提供模型训练脚本和经典数据集上的模型效果对比参考,方便用户参考搭建合适模型。
目前还在努力完善中,完善过程中可能会出现不少bug或缺陷,欢迎提意见、参与到完善中~
技术交流QQ群:810456486
相关rk产品链接:null
- 此仓库为了方便模块复用,采用动态导入的形式链接各函数,示例文件
- 动态导入的解析规则说明
- 模型适配边缘端处理。目前主要适配瑞星微硬件上的 npu 处理器。
- 主要包含 图像分类、物体检测、图像分割、关键点定位 的神经网络模型。
- 提供模型训练结果参考。
- 提供常用模块,如 框型标注、关键点标注、分割图标注的数据增强等功能。
-
训练结果可选导出所使用文件,方便后期对以往的训练结果进行追溯复现,同时亦可用作分发。
-
增加物体跟踪模型。