近期NVIDIA开源了一个叫做JetNet的项目,可以让开发者轻松地在 NVIDIA Jetson 设备上尝试各种模型来完成图像分类、身体姿势、手部姿势和文本检测等任务。
JetNet项目具有如下特点:
- 由官方实现的端到端的基于TensorRT优化的算法高性能部署项目
- 端到端:包含从训练框架输出模型文件到live demo全流程,项目内部实现了模型的转换和部署
- 支持用户直接使用适配好的算法
- 支持用户对适配好的算法重新训练,直接替换重新训练后的模型文件,而不需要重新实现复杂的模型转换流程
- 支持用户扩展算法
- 基于python使用和开发
- 只支持pytorch DL训练框架,使用torch2trt转换模型-
项目主页:JetNet (nvidia-ai-iot.github.io)
这个项目对大家有多少价值?如果X3派开发类似的项目有什么建议,欢迎讨论~