|
|
||
|---|---|---|
| cnn | ||
| data/nerf/lego | ||
| docs | ||
| nerf | ||
| README.md | ||
| requirements.txt | ||
| 实验4+三维重建.docx | ||
NeRF和CNN实验项目
项目结构
nerf/:NeRF算法实现代码cnn/:CNN图像识别算法代码data/:数据集存储目录results/:训练结果保存目录docs/:实验报告文档
环境依赖
# 安装PyTorch(根据CUDA版本选择对应命令)
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
# 安装其他依赖
pip install numpy opencv-python matplotlib argparse
使用说明
运行NeRF实验
cd nerf
python nerf/train_nerf.py
python nerf/test_nerf.py
运行CNN实验
cd cnn
python cnn.py --data_path ../data
实验报告
详见docs/experiment_report.md文件