目前 TF 和 PyTorch 都是非常流行的 AI 深度学习框架,TF 更通用一些,也有 Google 的强大支持。目前地平线的算法工具链支持 TF 和 MXNET。欢迎阅读我们的入门课程,了解更多详细信息。
除了可以直接用TF和MXNET训练之外,我们5月份还会有浮点模型转定点工具发布,这个工具支持我们caffe和ONNX的浮点模型转换为直接运行在板子上的定点模型。 其中ONNX本身是一个转换框架,其他比如PyTorch的模型可以先转为ONNX,在转换为板子上执行的模型。如果需要做一个硬件产品项目,可以联系我们商务同学,可以尽快开展前期工作和算法讨论。 关于可以用地平线哪些模型,我们后续会专门有一个页面介绍。敬请期待。