Commit 0b0143ed authored by bailuo's avatar bailuo
Browse files

update requirements 训练数据集配置文件说明

parent 5c5ca2c7
......@@ -88,7 +88,8 @@ python main_processing.py # 预处理数据集
```
python train.py --config configs/default.txt # 注意修改configs文件以及config.py文件
``` -->
训练只采用了部分数据集NYUv2,在GPU和DCU上完全保持一致
训练只采用了部分数据集`NYUv2`,在GPU和DCU上完全保持一致。\
注意:修改配置文件 metric_depth/zoedepth/utils/comfig.py 中对应字段数据地址路径。
### 单机单卡
```
cd ./metric_depth
......
......@@ -4,3 +4,5 @@ torch
torchvision
opencv-python
huggingface_hub
wandb
h5py
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment