将 PyTorch 计算机视觉深度学习模型部署到 Windows 桌面应用程序中答案

【问题标题】：Deployment of a PyTorch computer vision deep learning model into a windows desktop app将 PyTorch 计算机视觉深度学习模型部署到 Windows 桌面应用程序中
【发布时间】：2020-11-03 18:04:24
【问题描述】：

我已经使用我的数据集在 PyTorch 中训练了一个 yolov3 模型，并且我还为它编写了一些实用程序代码，这些代码与模型一起运行，全部使用 python 语言。现在我想将此模型和我的实用程序部署到一个 Windows 桌面应用程序中，该应用程序可以拍摄视频并在其帧上运行模型。如何以最小的代码更改或用另一种语言重写它来完成这项任务？什么框架是设计应用程序 UI 的最佳选择？谢谢。

【问题讨论】：

标签： python windows deep-learning pytorch

【解决方案1】：

我会首先将您的模型与 PyTorch 一起使用来检测每一帧，然后使用 numpy ImageDraw 在您的对象周围绘制（待检测）。这是一篇关于此的文章：Drawing a rectangle inside a 2D numpy array
然后我将使用 OpenCV(cv2) 将所有帧附加在一起以制作您也可以使用 ffmpeg 的视频。这是一篇关于这个（OpenCV）的文章：How to make a movie out of images in python
然后对于您的 UI 框架，您可以使用 PyQt5 来显示您的视频：Load an opencv video frame by frame using PyQT。但是您也可以将 Kivy 与 Gstreamer 一起使用：Kivy VideoPlayer fullscreen, loop, and hide controls
最后将您的 .py 文件转换为 .exe（Windows 可执行文件）我会使用 PyInstaller：http://www.pyinstaller.org/

【讨论】：

感谢您的回答。第1步和第2步我已经做好了，毕竟这将是商业软件，PyQt5对windows的兼容性好吗？此外，据我所知，Pyinstaller 将所有库打包在一个 .exe 文件中，但是我希望拥有一个传统的 setup.exe 文件，并能够在将来锁定或许可该应用程序。我可以使用 Pyinstaller 做到这一点吗？
@Behnamkvl 1. 对于许可软件，有“Cryptolens”：youtube.com/watch?v=g3twJVAz-iI 但它不是免费服务。但是，如果您想自己免费制作许可申请，请查看 youtube.com/watch?v=IArt2Fgv644，但我不推荐它，因为它很长，而且您需要具有静态 ip 的服务器之类的东西，而且它的成本可能与 Cryptolens 帐户相同。
@Behnamkvl 2. 要将您的 .exe 转换为 setup.exe 以便用户轻松安装和卸载，我建议使用类似 'NSIS':youtube.com/watch?v=UZX5kH72Yx4 @ 5:30 或您可以使用 Inno：youtube.com/watch?v=KRMs9z6KoEU 我相信两者都是免费的，但请仔细检查。
@Behnamkvl 3. 对于 PyQt5 的兼容性，它在 windows 上运行得非常好，但它基于 Qt 框架。我也相信 PyQt5 可以在 Linux 和 macOS 上运行。 stackoverflow.com/questions/16846501/…