深度学习框架PyTorch因其易用性已经渗透到企业中。通过媒体巨头迪士尼、机器人公司Blue River Technology和矿业初创公司Datarock的案例，让我们看看他们为什么选择PyTorch而不是谷歌著名的TensorFlow框架。

深度学习是机器学习的一个子类，它利用多层神经网络自动处理历史悠久的机器任务，如图像识别、自然语言处理(NLP)和机器翻译等。

自2015年从谷歌推出以来，TensorFlow一直是研究和商业领域最受欢迎的开源深度学习框架。然而，在2016年，源于脸书的PyTorch由于社区驱动的易用性和部署改进，可以应用于广泛的用例，并很快赶上了TensorFlow。

PyTorch在汽车行业得到了广泛的应用，比如特斯拉和Lyft Level 5的飞行员自动驾驶系统。该框架还可以用于工业应用中媒体公司和机器人的内容分类和推荐。

脸书人工智能产品主管乔斯皮萨克告诉信息世界，尽管他对采用PyTorch的公司数量的增加感到高兴，但要在该行业获得更广泛的应用，还有很多工作要做。

“下一波采用将伴随着生命周期管理、MLOps和Kubeflow管道以及相关社区的出现。他说，“对于刚入门的人来说，这些工具相当不错，托管服务和一些开源软件都可以使用，比如Azure ML上的AWS或者SageMaker。」

一 迪士尼：精准识别动画片中的人脸

自2012年以来，媒体巨头迪士尼的工程师和数据科学家一直在构建公司所谓的“内容基因组”。“内容基因组”平台用内容元数据填充知识图谱，基于机器学习的搜索能力为迪士尼庞大的内容库和个性化应用提供了动力支持。

例如，如果在搜索引擎中搜索asjddy，AI程序可以使用这些数据来增强搜索，这可以更好地帮助动画师从迪士尼档案中找到特定的镜头和序列。

“这些元数据改进了迪士尼说书人用来制作内容的工具；激发了讲故事的迭代创造力；通过推荐引擎、数字导航和内容发现为用户体验提供动力；实现了商业智能。迪士尼开发者Miquelangel farr、苦苦挣扎的冷面Accardo、Marc Junyent、等待蓝天的Alfaro和Cesc Guitart在7月的一篇博文中写道。

在此之前，迪士尼不得不投入巨大的内容标注项目，求助数据科学家，利用图像识别的深度学习模型，训练一条自动化标签流水线，识别大量人物、人物、地点的图像。

迪士尼的工程师起初试图使用包括TensorFlow在内的各种框架，但在2019年，他们决定围绕PyTorch进行整合。工程师们已经从传统的面向梯度直方图(HOG)特征描述符和流行的支持向量机(SVM)模型转向称为区域和卷积神经网络(R-CNN)的对象检测架构版本。后者更有利于处理迪士尼内容中常见的真人、动画和视觉效果的结合。

“在漫画中很难定义人脸，所以我们转向使用对象检测器的深度学习方法，并使用了迁移学习。迪士尼研究工程师们正在等待的蓝天Alfaro向InfoWorld解释道。在只处理了数千张人脸后，新模型已经能够在所有三个用例中大致识别人脸。2020年1月投产。

“目前我们只用一个模型来处理三种类型的人脸，这对于像《复仇者联盟》这样的漫威漫画电影来说非常棒，因为它需要同时识别钢铁侠和振动发带，或者任何带口罩的人物。”她说。

因为工程师要处理大量的视频数据来并行训练和运行模型，所以在进入生产阶段时也要在昂贵的高性能GPU上运行。

从中央处理器的转变使工程师能够更快地重新培训和更新模型。它还加快了向迪士尼团队分发结果的速度，将整部电影的处理时间从大约一个小时减少到今天的5到10分钟。

“TensorFlow对象检测器在制造过程中会带来内存问题，很难更新，而PyTorch有相同的对象检测器和fast-RCNN，所以我们开始使用PyTorch来处理一切。”阿尔法罗说。

对于工程团队来说，从一个框架到另一个框架的转换出奇地简单。“切换到PyTorch很容易”，因为它都是内置的。只需要插入一些函数就可以快速上手，学习起来并不难。”阿尔法罗说。

当他们遇到任何问题或瓶颈时，充满活力的PyTorch社区随时准备帮助他们。

二 蓝河科技：除杂草机器人

蓝河科技有限公司设计了一款机器人，它结合了数字寻路、集成摄像头和计算机视觉技术，能够识别农作物和杂草，并对杂草精准喷洒除草剂。

2017年，重型设备制造商John Deere以3.05亿美元收购了Blue River Technology，目的是转化这项技术

整合到其农业设备中。

蓝河科技的研究人员在训练计算机视觉模型识别杂草和作物之间的区别时，尝试了各种深度学习框架。

他们让一些农学家手动标记图像，并使用PyTorch训练了卷积神经网络（CNN），以分析每帧图像并绘制出精确的农作物和杂草位置图。蓝河科技计算机视觉和机器学习技术总监Chris Padwick8月在一篇博客文章中写道。

「和其他公司一样，我们尝试了Caffe、TensorFlow，然后尝试了PyTorch，」Padwick接受媒体采访时说，「对我们来说，PyTorch的作用非常大。我们没有收到过任何bug报告，也没有出现过阻塞bug。在分布式计算中，它确实比TensorFlow更好，并且更易于使用，因为TensorFlow的数据并行性相当复杂。」

Padwick表示，PyTorch框架的流行和简单性让他在快速增加新员工时具有优势。尽管如此，Padwick梦想着一个世界，「人们用他们喜欢的任何东西进行开发。有些人喜欢用Apache MXNet或Darknet或Caffe进行研究，但在生产中必须用单一语言，而PyTorch拥有我们成功所需的一切。」

三 采矿创企Datarock：基于深度学习的岩心钻探

澳大利亚初创公司Datarock由一群地球科学家创立，致力于将计算机视觉技术应用于采矿业。更具体地说，其深度学习模型正在帮助地质学家比以前更快地分析钻芯样品图像。

通常，地质学家会一厘米一厘米地仔细研究这些样本，以评估矿物学和结构，而工程师则会寻找诸如断层、裂缝和岩石质量等物理特征。这个过程既缓慢又容易出现人为错误。

Datarock的首席运营官细心的过客 Crawford接受InfoWorld采访时表示:「电脑能像工程师一样看到岩石。」「如果你能在图像中看到它，我们就可以训练一个模型来像人类一样分析它。」

与Blue River类似，Datarock在生产中使用RCNN模型的一种变体，研究人员转向数据增强技术，以在早期阶段收集足够的训练数据。

在儒雅的早晨的发现期之后，该团队着手结合各种技术来创建用于钻探岩心图像的图像处理工作流程。研究人员在博客文章中写道：「这涉及开发一系列深度学习模型，这些模型可以将原始图像处理为结构化格式并分割重要的地质信息。」

使用Datarock的技术，客户可以在半小时内获得结果，而手动记录发现结果需要五到六个小时。Crawford说，这使地质学家从繁重的工作中解放了出来。但是，「当我们将更困难的事情自动化时，我们确实会遇到一些压力，并且必须说明它们是该系统的一部分，可以训练模型并获得反馈循环。」

像许多训练深度学习计算机视觉模型的公司一样，Datarock始于TensorFlow，但不久后便转移到PyTorch。

Datarock机器学习负责人Duy Tin Truong对InfoWorld表示：「起初，我们使用TensorFlow时，它会因神秘原因而崩溃。」他还表示，「PyTorch和Detecton2当时发布了，很符合我们的需求，因此在进行一些测试后，我们发现调试和工作更加容易，占用的内存更少，因此我们进行了转换。」

Datarock还表示，在GPU上运行模型时，从TensorFlow到PyTorch和Detectron2的推理性能提高了4倍，在CPU上提高了3倍。

Truong指出，转换使用PyTorch的原因在于，不断发展的社区、设计良好的界面、易用性和更好的调试等优点。他还认为，虽然从界面角度上看，它们有很大的不同，但是如果了解TensorFlow，切换起来就相当容易，尤其如果更了解Python，感受就会不同。

参考链接：

https://www.infoworld.com/article/3597904/why-enterprises-are-turning-from-tensorflow-to-pytorch.html