OpenAIglasses_for_Navigation/setup.bat at main · TemsSwift/OpenAIglasses_for_Navigation · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
@echo off
REM AI Glass System - Windows 快速安装脚本

echo ==========================================
echo   AI Glass System - 自动安装脚本
echo ==========================================
echo.

REM 检查 Python
echo 正在检查 Python...
python --version >nul 2>&1
if errorlevel 1 (
    echo [错误] 未找到 Python
    echo 请从 https://www.python.org/downloads/ 下载并安装 Python 3.9-3.11
    pause
    exit /b 1
)

python --version
echo [成功] Python 已安装

REM 检查 CUDA
echo.
echo 正在检查 CUDA...
nvidia-smi >nul 2>&1
if errorlevel 1 (
    echo [警告] 未检测到 NVIDIA GPU，将使用 CPU 模式（速度较慢）
    set HAS_GPU=0
) else (
    echo [成功] 检测到 NVIDIA GPU
    nvidia-smi --query-gpu=name,driver_version,memory.total --format=csv,noheader
    set HAS_GPU=1
)

REM 创建虚拟环境
echo.
echo 正在创建虚拟环境...
if exist venv (
    echo [警告] 虚拟环境已存在
    set /p RECREATE="是否删除并重新创建? (y/n): "
    if /i "%RECREATE%"=="y" (
        rmdir /s /q venv
        python -m venv venv
        echo [成功] 虚拟环境已重新创建
    )
) else (
    python -m venv venv
    echo [成功] 虚拟环境已创建
)

REM 激活虚拟环境
echo.
echo 正在激活虚拟环境...
call venv\Scripts\activate.bat

REM 升级 pip
echo.
echo 正在升级 pip...
python -m pip install --upgrade pip -q
echo [成功] pip 已升级

REM 安装 PyTorch
echo.
echo 正在安装 PyTorch...
if %HAS_GPU%==1 (
    echo 安装 GPU 版本 PyTorch ^(CUDA 11.8^)...
    pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 --index-url https://download.pytorch.org/whl/cu118 -q
) else (
    echo 安装 CPU 版本 PyTorch...
    pip install torch torchvision -q
)
echo [成功] PyTorch 已安装

REM 验证 PyTorch
echo.
echo 验证 PyTorch 安装...
python -c "import torch; print(f'PyTorch 版本: {torch.__version__}'); print(f'CUDA 可用: {torch.cuda.is_available()}')"

REM 安装 PyAudio
echo.
echo 正在安装 PyAudio...
echo [警告] PyAudio 在 Windows 上可能需要手动安装
echo 如果自动安装失败，请从以下地址下载 wheel 文件:
echo https://www.lfd.uci.edu/~gohlke/pythonlibs/#pyaudio
echo.
pip install pyaudio -q
if errorlevel 1 (
    echo [警告] PyAudio 自动安装失败，请手动安装
) else (
    echo [成功] PyAudio 已安装
)

REM 安装其他依赖
echo.
echo 正在安装 Python 依赖...
pip install -r requirements.txt -q
echo [成功] Python 依赖已安装

REM 创建 .env 文件
echo.
if not exist .env (
    echo 正在创建 .env 配置文件...
    copy .env.example .env >nul
    echo [成功] .env 文件已创建
    echo [提示] 请编辑 .env 文件，填入您的 DASHSCOPE_API_KEY
) else (
    echo [跳过] .env 文件已存在
)

REM 创建必要的目录
echo.
echo 正在创建目录结构...
if not exist recordings mkdir recordings
if not exist model mkdir model
if not exist music mkdir music
if not exist voice mkdir voice
echo [成功] 目录结构已创建

REM 检查模型文件
echo.
echo 正在检查模型文件...
set MISSING=0
if exist model\yolo-seg.pt (echo [成功] yolo-seg.pt) else (echo [缺失] yolo-seg.pt & set MISSING=1)
if exist model\yoloe-11l-seg.pt (echo [成功] yoloe-11l-seg.pt) else (echo [缺失] yoloe-11l-seg.pt & set MISSING=1)
if exist model\shoppingbest5.pt (echo [成功] shoppingbest5.pt) else (echo [缺失] shoppingbest5.pt & set MISSING=1)
if exist model\trafficlight.pt (echo [成功] trafficlight.pt) else (echo [缺失] trafficlight.pt & set MISSING=1)
if exist model\hand_landmarker.task (echo [成功] hand_landmarker.task) else (echo [缺失] hand_landmarker.task & set MISSING=1)

if %MISSING%==1 (
    echo.
    echo [警告] 部分模型文件缺失，请将模型文件放入 model\ 目录
)

REM 完成
echo.
echo ==========================================
echo [成功] 安装完成!
echo ==========================================
echo.
echo 下一步:
echo 1. 编辑 .env 文件，填入您的 API 密钥:
echo    notepad .env
echo.
echo 2. 确保所有模型文件已放入 model\ 目录
echo.
echo 3. 启动系统:
echo    venv\Scripts\activate
echo    python app_main.py
echo.
echo 4. 访问 http://localhost:8081
echo.
echo [提示] 每次使用前请激活虚拟环境:
echo   venv\Scripts\activate
echo.

pause