Skip to content

GitLab

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

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

parent 5c5ca2c7
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 2 deletions
README.md
View file @ 0b0143ed
...@@ -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
......
requirements.txt
View file @ 0b0143ed
...@@ -4,3 +4,5 @@ torch ...@@ -4,3 +4,5 @@ torch
torchvision torchvision
opencv-python opencv-python
huggingface_hub 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