AKA-Sim 模拟器

基于前视视角的自动驾驶数据采集与ACT模型训练系统。

功能

• 前视视角模拟器（俯视图 + 第一人称伪3D视角）
• 键盘控制（WASD/方向键）
• 数据采集用于ACT训练
• 模型训练与推理

快速启动

1. 安装依赖

# 后端依赖
cd backend
pip install -r requirements.txt

# 前端依赖
cd ../ui
npm install

2. 启动后端

cd backend
python main.py

后端运行在 http://localhost:8000

3. 启动前端

cd ui
npm run dev

前端运行在 http://localhost:5173

使用流程

数据采集

1. 在模拟器中使用 WASD 或方向键控制小车
2. 点击"开始采集"按钮开始记录数据
3. 采集完成后点击"停止采集"（自动导出到 output/dataset/）

模型训练

1. 确保已采集足够数据（至少100个样本）
2. 点击"开始训练"按钮
3. 训练完成后模型保存在 output/train/model.pt

模型推理

1. 点击"加载模型"加载训练好的模型
2. 点击"自动推理"让小车自主行动

目录结构

├── backend/           # 后端代码
│   ├── main.py       # 入口
│   ├── api.py        # API接口
│   ├── training.py   # 训练模块
│   └── data_export.py # 数据导出
├── ui/               # 前端代码
├── output/           # 输出目录
│   ├── dataset/     # 采集的数据
│   └── train/       # 训练模型
│       └── model.pt
└── README.md

技术栈

• 后端: FastAPI, Socket.IO, PyTorch
• 前端: React, TypeScript, Socket.IO Client