Skip to content

Commit

Permalink
update docs and README.
Browse files Browse the repository at this point in the history
  • Loading branch information
JubSteven committed Aug 15, 2024
1 parent cf5522e commit 8b5982d
Show file tree
Hide file tree
Showing 2 changed files with 18 additions and 3 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -70,8 +70,8 @@ $ python scripts/eval_single.py --cfg config/release/eval_single.yaml
-g ${gpu_id}
--reload ${PATH_TO_CKPT}
--dataset ${DATASET}
--view_min {MIN_VIEW}
--view_max {MAX_VIEW}
--view_min ${MIN_VIEW}
--view_max ${MAX_VIEW}
--model ${MODEL}
```

Expand Down
17 changes: 16 additions & 1 deletion docs/datasets.md
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,22 @@ The following datasets are used for training.

We have used [webdataset](https://github.com/webdataset/webdataset) to pack the datasets into shards for efficient training. In training POEM, we have also filtered samples with right hand only.

The packed dataset tars used for training and evaluation will be released soon.
The packed dataset tars used for training and evaluation is released [here](https://huggingface.co/datasets/JubSteven/POEM-v2/tree/main). After downloading the dataset, create a directory `data` in the root directory and organize the files as follows:

```
├── assets
├── config
├── data
│ ├── dataset_tars
│ ├── Oakink_mv
│ ├── HO3D_mv
│ ├── HO3D_mv_test
│ ├── ...
│ └── Arctic_mv
├── exp
├── lib
├── ...
```

 

Expand Down

0 comments on commit 8b5982d

Please sign in to comment.