单目3D初始代码
This commit is contained in:
149
eval_tools/docs/QUICKSTART.md
Executable file
149
eval_tools/docs/QUICKSTART.md
Executable file
@@ -0,0 +1,149 @@
|
||||
# 快速开始:双ROI融合模型测试
|
||||
|
||||
## 5分钟快速上手
|
||||
|
||||
### 1. 检查依赖
|
||||
|
||||
```bash
|
||||
# 确保在项目根目录
|
||||
cd /path/to/yolov5-3d
|
||||
|
||||
# 检查Python包
|
||||
python -c "import numpy, cv2, torch; print('基础依赖OK')"
|
||||
|
||||
# 如果测试ONNX模型,检查onnxruntime
|
||||
python -c "import onnxruntime; print('ONNX Runtime已安装')"
|
||||
```
|
||||
|
||||
### 2. 准备模型和数据
|
||||
|
||||
确保以下文件存在:
|
||||
- 模型文件:`release/yolov5s-30w/merged_model.onnx` 或 `.torchscript`
|
||||
- 测试数据:视频bin文件或图像目录
|
||||
|
||||
### 3. 运行测试
|
||||
|
||||
#### 方式一:使用交互式脚本(推荐)
|
||||
|
||||
```bash
|
||||
./eval_tools/test_merged_model.sh
|
||||
# 然后按提示选择测试模式(1-7)
|
||||
```
|
||||
|
||||
#### 方式二:直接命令
|
||||
|
||||
```bash
|
||||
# ONNX模型
|
||||
python eval_tools/test_val_merged_model.py \
|
||||
--source /data1/dongying/Mono3d/G1M3/eval_dataset/20251210222153/sigmastar.1/camera4.bin \
|
||||
--weights release/yolov5s-30w/merged_model.onnx \
|
||||
--model-type onnx
|
||||
|
||||
# TorchScript模型
|
||||
python eval_tools/test_val_merged_model.py \
|
||||
--source /data1/dongying/Mono3d/G1M3/eval_dataset/20251210222153/sigmastar.1/camera4.bin \
|
||||
--weights release/yolov5s-30w/merged_model.torchscript \
|
||||
--model-type torchscript \
|
||||
--device cuda
|
||||
```
|
||||
|
||||
### 4. 查看结果
|
||||
|
||||
```bash
|
||||
# 结果保存在
|
||||
ls runs/val_viz_merged/exp/
|
||||
|
||||
# 用图像查看器打开
|
||||
eog runs/val_viz_merged/exp/000000_*.jpg
|
||||
# 或
|
||||
display runs/val_viz_merged/exp/000000_*.jpg
|
||||
```
|
||||
|
||||
## 常用配置
|
||||
|
||||
### 提高检测精度(减少误检)
|
||||
|
||||
```bash
|
||||
python eval_tools/test_val_merged_model.py \
|
||||
--source your/data/path \
|
||||
--weights your/model.onnx \
|
||||
--model-type onnx \
|
||||
--conf-thres 0.4 \
|
||||
--iou-thres 0.5
|
||||
```
|
||||
|
||||
### 获取更多检测(增加召回)
|
||||
|
||||
```bash
|
||||
python eval_tools/test_val_merged_model.py \
|
||||
--source your/data/path \
|
||||
--weights your/model.onnx \
|
||||
--model-type onnx \
|
||||
--conf-thres 0.15 \
|
||||
--iou-thres 0.6
|
||||
```
|
||||
|
||||
## 文件说明
|
||||
|
||||
| 文件 | 说明 |
|
||||
|------|------|
|
||||
| `test_val_merged_model.py` | 主测试脚本 |
|
||||
| `test_merged_model.sh` | 交互式测试脚本(推荐) |
|
||||
| `README_MERGED_MODEL.md` | 完整文档(详细说明) |
|
||||
| `QUICKSTART_MERGED.md` | 本文件(快速入门) |
|
||||
|
||||
## 故障排查
|
||||
|
||||
### 问题1:找不到模块
|
||||
|
||||
```
|
||||
ModuleNotFoundError: No module named 'utils'
|
||||
```
|
||||
|
||||
**解决**:确保在项目根目录运行
|
||||
```bash
|
||||
cd /deeplearning_team/ydong/dongying/projects/yolov5-3d
|
||||
python eval_tools/test_val_merged_model.py ...
|
||||
```
|
||||
|
||||
### 问题2:ONNX Runtime错误
|
||||
|
||||
```
|
||||
ImportError: onnxruntime not installed
|
||||
```
|
||||
|
||||
**解决**:
|
||||
```bash
|
||||
pip install onnxruntime-gpu # GPU版本
|
||||
# 或
|
||||
pip install onnxruntime # CPU版本
|
||||
```
|
||||
|
||||
### 问题3:CUDA内存不足
|
||||
|
||||
```
|
||||
RuntimeError: CUDA out of memory
|
||||
```
|
||||
|
||||
**解决**:使用CPU推理
|
||||
```bash
|
||||
# TorchScript模型
|
||||
python eval_tools/test_val_merged_model.py \
|
||||
--model-type torchscript \
|
||||
--device cpu \
|
||||
...
|
||||
```
|
||||
|
||||
## 下一步
|
||||
|
||||
- 阅读完整文档:`README_MERGED_MODEL.md`
|
||||
- 调整检测参数优化效果
|
||||
- 修改代码处理更多帧(默认10帧)
|
||||
- 集成到自己的推理流程
|
||||
|
||||
## 技术支持
|
||||
|
||||
遇到问题?请检查:
|
||||
1. 完整文档 `README_MERGED_MODEL.md`
|
||||
2. 脚本帮助 `python eval_tools/test_val_merged_model.py --help`
|
||||
3. 交互式脚本选项7查看详细帮助
|
||||
Reference in New Issue
Block a user