英伟达发布迁移学习工具包，现在可以申请早期试用
圆栗子 发自 凹非寺 量子位 报道 | 公众号 QbitAI
最近，英伟达发布了一个迁移学习工具包 (Transfer Learning Toolkit) 。
这个基于Python的工具包，打包了许多预训练的模型：
常用的ResNet-10，ResNet-18，ResNet-50，GoogLeNet，VGG-16和VGG-19等等都包含在内。
英伟达说，开发者不用费时费力自己去搭建深度神经网络 (DNN) ，直接对预训练模型做些修改，就可以做出自己的应用。
并且，工具包里的许多模型，都是为某一领域定制的。比如视频分析 (IVA) ，比如医学影像分析。
六大功能
所谓迁移学习，是指预训练的模型已经学习到一些特征，我们要把它学到的东西，通过权重，迁移给另外一个神经网络。
用户还可以在工具包提供的原有神经网络上，增加数据，或者增加特征。然后让它们重新训练，适应变化。
工具包还提供多GPU支持。
英伟达官方总结了六大关键功能：
- 许多预训练的优化过的领域特定DNN，预先打包在里面；
- 有计算机视觉中，物体分类、检测的应用示例；
- 在异构的多GPU环境中，易于做模型适应 (Model Adaptation) ，易于重新训练；
- 可以轻松修改配置文件，增加新类别、新特征，压缩模型大小；
- Model Export API可以把模型轻松部署在英伟达的DeepStream SDK 3.0上，做智能视频分析 (IVA) 应用；
- Model Export API可以把模型部署到Clara平台上，来做医学影像相关应用。
两种主要应用
英伟达介绍的两种应用，一是智能视频分析 (IVA) 。
比如，停车管理、物流管理、零售分析等方面都有这样的应用。
工具包里的模型，都是专门用来做IVA推理、并经过透彻训练的。
以上这些栗子，是预训练的图像分类和物体识别模型，都是工具包附带的。
英伟达提供了端到端的深度学习工作流，来加速模型训练和部署：如用DeepStream SDK 3.0，部署在Tesla GPU上。
另外一种应用，是医学影像分析。
工具包里的模型，依然是专为这类应用而训练的。
比如，AI辅助人类进行数据标注的SDK，可以加速医学影像的标注过程。
还有，英伟达团队曾经开发过一个3D脑瘤分割模型，叫做BraTS。它获得了2018多模态脑瘤分割挑战赛的冠军。
这个模型也打包在工具包里了。
工具包用的是朴素的命令行界面，开发者可以轻松修改原有的模型。
可以申请了
现在，迁移学习工具包还没有正式发布，不过可以申请早期访问 (Early Access) 了：
另外，官方表示发布已经临近，各位也可以注册一下，以便第一时间收到消息：
申请/注册传送门： https://developer.nvidia.com/transfer-learning-toolkit
— 完 —