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

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

parent 5c5ca2c7
...@@ -88,7 +88,8 @@ python main_processing.py # 预处理数据集 ...@@ -88,7 +88,8 @@ python main_processing.py # 预处理数据集
``` ```
python train.py --config configs/default.txt # 注意修改configs文件以及config.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 cd ./metric_depth
......
...@@ -3,4 +3,6 @@ gradio==4.14.0 ...@@ -3,4 +3,6 @@ gradio==4.14.0
torch torch
torchvision torchvision
opencv-python opencv-python
huggingface_hub huggingface_hub
\ No newline at end of file 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