Skip to content

Commit e1165aa

Browse files
committed
docs(README): update model link to the latest code.
1 parent d9475f7 commit e1165aa

2 files changed

Lines changed: 6 additions & 6 deletions

File tree

README.md

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ Task: Scene Flow Estimation in Autonomous Driving.
1313

1414
🔥 2024/07/02: Check the self-supervised version in our new ECCV'24 [SeFlow](https://github.com/KTH-RPL/SeFlow). The 1st ranking in new leaderboard among self-supervise methods.
1515

16-
Pre-trained weights for models are available in [Zenodo](https://zenodo.org/records/12751363).
16+
Pre-trained weights for models are available in [Zenodo](https://zenodo.org/records/13744999).
1717
Check usage in [2. Evaluation](#2-evaluation) or [3. Visualization](#3-visualization).
1818

1919
**Scripts** quick view in our scripts:
@@ -81,7 +81,7 @@ Note: Prepare raw data and process train data only needed run once for the task.
8181
### Data Preparation
8282

8383
Check [dataprocess/README.md](dataprocess/README.md#argoverse-20) for downloading tips for the raw Argoverse 2 dataset. Or maybe you want to have the **mini processed dataset** to try the code quickly, We directly provide one scene inside `train` and `val`. It already converted to `.h5` format and processed with the label data.
84-
You can download it from [Zenodo](https://zenodo.org/records/12751363/files/demo_data.zip) and extract it to the data folder. And then you can skip following steps and directly run the [training script](#train-the-model).
84+
You can download it from [Zenodo](https://zenodo.org/records/13744999/files/demo_data.zip) and extract it to the data folder. And then you can skip following steps and directly run the [training script](#train-the-model).
8585

8686
```bash
8787
wget https://zenodo.org/record/12751363/files/demo_data.zip
@@ -114,7 +114,7 @@ python train.py model=fastflow3d lr=4e-5 epochs=20 batch_size=16 loss_fn=ff3dLos
114114
> However, we kept the setting on lr=2e-6 and 50 epochs in (SeFlow & DeFlow) paper experiments for the fair comparison with ZeroFlow where we directly use their provided weights.
115115
> We suggest afterward researchers or users to use the setting here (larger lr and smaller epoch) for faster converge and better performance.
116116
117-
To help community benchmarking, we provide our weights including fastflow3d, deflow in [Zendo](https://zenodo.org/records/12751363).
117+
To help community benchmarking, we provide our weights including fastflow3d, deflow in [Zendo](https://zenodo.org/records/13744999).
118118
These checkpoints also include parameters and status of that epoch inside it. If you are interested in weights of ablation studies, please contact us.
119119

120120
## 2. Evaluation
@@ -124,7 +124,7 @@ You can view Wandb dashboard for the training and evaluation results or run/subm
124124
Since in training, we save all hyper-parameters and model checkpoints, the only thing you need to do is to specify the checkpoint path. Remember to set the data path correctly also.
125125
```bash
126126
# downloaded pre-trained weight, or train by yourself
127-
wget https://zenodo.org/records/12751363/files/deflow_best.ckpt
127+
wget https://zenodo.org/records/13744999/files/deflow_best.ckpt
128128

129129
python eval.py checkpoint=/home/kin/deflow_best.ckpt av2_mode=val # it will directly prints all metric
130130
# it will output a command with absolute path of a zip file for you to submit to leaderboard
@@ -157,7 +157,7 @@ We provide a script to visualize the results of the model. You can specify the c
157157

158158
```bash
159159
# downloaded pre-trained weight, or train by yourself
160-
wget https://zenodo.org/records/12751363/files/deflow_best.ckpt
160+
wget https://zenodo.org/records/13744999/files/deflow_best.ckpt
161161

162162
python save.py checkpoint=/home/kin/deflow_best.ckpt dataset_path=/home/kin/data/av2/preprocess/sensor/vis
163163

dataprocess/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ And flowlabel data can be downloaded here with ground segmentation by HDMap foll
8888
You can download the processed map folder here to free yourself downloaded another type of data again:
8989

9090
```bash
91-
wget https://zenodo.org/records/12751363/files/waymo_map.tar.gz
91+
wget https://zenodo.org/records/13744999/files/waymo_map.tar.gz
9292
tar -xvf waymo_map.tar.gz -C /home/kin/data/waymo/flowlabel
9393
# you will see there is a `map` folder in the `flowlabel` folder now.
9494
```

0 commit comments

Comments
 (0)