整体步骤
步骤 1:创建一键启动脚本(替代打包后的 exe)
在/opt/darknet-master/train/目录下新建 ai_trainer.sh,内容:
#!/bin/bash set -e # 出错时立即退出,便于排查 # ==================== 1. 强制配置环境(和你的开发环境完全一致) ==================== # 激活虚拟环境 source /opt/auto-label-tool/envs/bin/activate # 配置cuda/darknet环境变量(和终端运行一致) export path="/usr/local/cuda-11.4/bin:$path" export ld_library_path="/usr/local/cuda-11.4/lib64:/opt/darknet-master:$ld_library_path" export cuda_home="/usr/local/cuda-11.4" export pythonpath="/opt/darknet-master/train:$pythonpath" # ==================== 2. 运行核心代码 ==================== echo "===== ai训练助手启动中 =====" python /opt/darknet-master/train/train.py # ==================== 3. 收尾(可选) ==================== deactivate echo "===== 程序运行结束 ====="
步骤 2:赋予脚本执行权限
chmod +x /opt/darknet-master/train/ai_trainer.sh
步骤 3:创建桌面快捷方式(模拟 exe 双击)
在 ubuntu 桌面新建 ai训练助手.desktop 文件(右键→新建文档→重命名),内容:
[desktop entry] version=1.0 type=application name=ai训练助手 comment=yolo训练工具(替代pyinstaller打包版) exec=/opt/darknet-master/train/ai_trainer.sh # 脚本绝对路径 icon=/opt/darknet-master/train/ai.png # 你的图标路径(可选) terminal=true # 显示终端日志(便于排查问题) categories=development;utility; encoding=utf-8 startupnotify=true
赋予桌面文件权限:
chmod +x ~/desktop/ai训练助手.desktop
步骤 4:运行验证
# 方式1:终端直接运行(推荐,查看完整日志) /opt/darknet-master/train/ai_trainer.sh # 方式2:双击桌面的「ai训练助手」图标(和exe双击体验一致)
创建.sh
步骤 1:进入目标目录
打开终端,执行以下命令切换到 /opt/darknet-master/train/ 目录:
cd /opt/darknet-master/train/
若提示 permission denied(权限不足),先提升权限:
sudo su # 切换到root用户(输入密码后回车) cd /opt/darknet-master/train/ # 再次进入目标目录
步骤 2:创建并编辑ai_trainer.sh
使用 nano 编辑器(简单易上手)创建并编辑脚本:
nano ai_trainer.sh
步骤 3:粘贴脚本内容
在打开的 nano 编辑器中,粘贴以下完整脚本内容(按 ctrl+v 粘贴):
#!/bin/bash set -e # 出错时立即退出,便于排查 # ==================== 1. 强制配置环境(和你的开发环境完全一致) ==================== # 激活虚拟环境 source /opt/auto-label-tool/envs/bin/activate # 配置cuda/darknet环境变量(和终端运行一致) export path="/usr/local/cuda-11.4/bin:$path" export ld_library_path="/usr/local/cuda-11.4/lib64:/opt/darknet-master:$ld_library_path" export cuda_home="/usr/local/cuda-11.4" export pythonpath="/opt/darknet-master/train:$pythonpath" # ==================== 2. 运行核心代码 ==================== echo "===== ai训练助手启动中 =====" python /opt/darknet-master/train/train.py # ==================== 3. 收尾(可选) ==================== deactivate echo "===== 程序运行结束 ====="
步骤 4:保存并退出nano编辑器
- 按
ctrl+o(保存文件),回车确认文件名; - 按
ctrl+x(退出编辑器)。
步骤 5:赋予脚本执行权限
执行以下命令让脚本具备可运行权限:
chmod +x ai_trainer.sh
若之前切换到了 root 用户,需确保权限对普通用户生效:
chmod 755 ai_trainer.sh # 所有用户可执行,所有者可读写 chown qt:qt ai_trainer.sh # 将文件归属改为你的普通用户(qt)
ubuntu 桌面新建 .desktop
步骤 1:切换到桌面目录
打开终端,执行命令进入桌面路径(ubuntu 默认桌面路径为 ~/desktop):
cd ~/desktop/
步骤 2:创建并编辑.desktop文件
使用 nano 编辑器创建文件(新手友好):
nano ai训练助手.desktop
步骤 3:粘贴完整配置内容
在 nano 编辑器中,粘贴以下内容(需根据你的实际路径修改exec和icon):
[desktop entry] version=1.0 type=application name=ai训练助手 comment=yolo训练工具(替代pyinstaller打包版) # 关键:替换为你的ai_trainer.sh脚本绝对路径 exec=/opt/darknet-master/train/ai_trainer.sh # 可选:替换为你的图标绝对路径(无图标可注释/删除该行) icon=/opt/darknet-master/train/ai.png terminal=true # 显示终端窗口(便于查看运行日志/报错) categories=development;utility;ai; encoding=utf-8 startupnotify=true # 启动时显示通知 name[zh_cn]=ai训练助手 comment[zh_cn]=yolo训练助手,调用darknet训练模型
核心参数说明:
exec:必须是ai_trainer.sh的绝对路径(之前创建的脚本路径);icon:可选,填你的图标图片路径(如无图标,删除该行即可);terminal=true:必须保留,否则脚本运行时无日志,出错无法排查。
步骤 4:保存并退出编辑器
- 按
ctrl+o→ 回车确认文件名; - 按
ctrl+x退出nano。
步骤 5:赋予可执行权限
chmod +x ~/desktop/ai训练助手.desktop
总结
以上为个人经验,希望能给大家一个参考,也希望大家多多支持代码网。
发表评论