English | 简体中文
这是一个基于 OpenMMLab 框架开发的一个适用于多种视觉任务主干网络的模型训练库。
同时可以使用MMDetection、MMClassification、MMPose等工具箱的模型并训练、测试和导出。
通过融合后的训练脚本我们可以很轻松地训练一个以上工具箱中已有的模型,只需要修改配置文件的数据集路径即可训练自定义数据集。、
可查看快速使用文档学习本项目的基本使用,同时也可使用我们提供的colab教程。
对于更多教程可以查看以下内容:
对于mmdetection、mmpos、mmclassify所支持的模型均可使用。
本项目直接实现的算法有以下:
PFLD
EAT
edgelab有关内容参考借鉴了以下项目:
对于在环境配置与训练过程中可能出现的问题可先查看相关问题解决文档 查看。若没能解决您的问题可提出issue, 我们会尽快为您解决。
edgelab 目前以 Apache 2.0 的许可证发布,但是其中有一部分功能并不是使用的 Apache2.0 许可证,我们在许可证 中详细地列出了这些功能以及他们对应的许可证,如果您正在从事盈利性活动,请谨慎参考此文档。