Update README.md

dontLoveBugs · Dec 29, 2018 · 4e89441 · 4e89441
1 parent b6a56bd
commit 4e89441
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -12,3 +12,9 @@ Note: we modify the ordinal layer using matrix operation, making trianing faster
 - [ ] Results evaluation on nyu test set
 - [x] Calculate alpha and beta in nyu dataset
 - [x] Realize the ordinal loss in paper 
+
+### Datasets
+
+#### NYU Depth V2
+Some friends asked me about how to use the NYU Depth V2 dataset. The best choice is to use all the Images (about 120k) in the dataset, but if you just want to test the code, you can use the nyu_depth_v2_labeled.mat and turn it to a h5 file. The convert script is 'create_nyu_h5.py' and you need to change the file paths to yours.
+