709 B
709 B
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文件