yoloV5-打包成EXE可执行文件

最新推荐文章于 2024-08-05 21:20:17 发布

老三是只猫

最新推荐文章于 2024-08-05 21:20:17 发布

阅读量3k

点赞数 3

分类专栏：目标检测文章标签： pytorch 人工智能

本文链接：https://blog.csdn.net/zhonglongshen/article/details/126755272

版权

目标检测专栏收录该内容

54 篇文章 3 订阅

订阅专栏

前言

Python 项目打包是很多新手经常会问的问题，之前也有文章介绍过如何使用 pyinstaller 来打包生成可执行文件，只不过打包过程是基于命令行的。本文介绍的这个工具，auto-py-to-exe，它是 pyinstaller 的 GUI 版本，对新手更加友好，点点鼠标就可以轻松搞定，那么，快开始吧。

安装

使用 pip 直接安装

pip install auto-py-to-exe

或者拉取源码，解压后进入根目录，执行

python setup.py install

打包示例

本文使用之前的 yolov5-pyqt5，https://github.com/zlszhonglongshen/yolov5-pyqt5项目进行打包，这里需要注意一点，如果你是使用的 python 虚拟环境，那必须在这个虚拟环境中去安装 auto-py-to-exe

项目代码拉下来之后，修改 main.py，将

self.device = select_device(self.opt.device)

改为

# gpu
self.device = torch.device('cuda:0')

# 如果只有cpu的话，就改成
# self.device = torch.device('cpu')

为什么要这么改？这里主要是 select_device 方法中使用到了 subprocess，而 pyinstaller 在打包 subprocess 方法时会报错，如果去修改 subprocess 部分的代码就有点麻烦了，为简单起见，就不去调用这个方法了,具体的可以参考 PyQt5系列教程(四)pyinstaller打包exe - 迷途小书童的Note迷途小书童的Note

然后，打开 auto-py-to-exe, 脚本文件选择 yolov5-pyqt5 文件夹下的 main.py，选择单文件，也就是最后就只生成一个可执行文件

在高级选项卡里，点击 --hidden-import 右侧的加号，然后填上 modles.yolo，否则，生成的 exe 运行时会报如下的错误

Traceback (most recent call last):
  File "detect_image.py", line 405, in <module>
  File "detect_image.py", line 365, in main
  File "detect_image.py", line 293, in detection_alertarea
  File "detect_image.py", line 79, in __init__
  File "models\experimental.py", line 137, in attempt_load
  File "torch\serialization.py", line 594, in load
    return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
  File "torch\serialization.py", line 853, in _load
    result = unpickler.load()
ModuleNotFoundError: No module named 'models.yolo'
[12976] Failed to execute script detect_image