Skip to content

Commit 348fa1e

Browse files
authored
docs: update model download link via zenodo
1 parent c93edcd commit 348fa1e

1 file changed

Lines changed: 16 additions & 8 deletions

File tree

README.md

Lines changed: 16 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ DeFlow: Decoder of Scene Flow Network in Autonomous Driving
77
[![video](https://img.shields.io/badge/video-YouTube-FF0000?logo=youtube&logoColor=white)](https://youtu.be/bZ4uUv0nDa0)
88

99
Task: Scene Flow Estimation in Autonomous Driving.
10-
Pre-trained weights for models are available in [Onedrive link](https://hkustconnect-my.sharepoint.com/:f:/g/personal/qzhangcb_connect_ust_hk/Et85xv7IGMRKgqrVeJEVkMoB_vxlcXk6OZUyiPjd4AArIg?e=lqRGhx).
10+
Pre-trained weights for models are available in [Zenodo](https://zenodo.org/records/12173874) or [Onedrive link](https://hkustconnect-my.sharepoint.com/:f:/g/personal/qzhangcb_connect_ust_hk/Et85xv7IGMRKgqrVeJEVkMoB_vxlcXk6OZUyiPjd4AArIg?e=lqRGhx).
1111
Check usage in [2. Evaluation](#2-evaluation) or [3. Visualization](#3-visualization).
1212

1313
**Scripts** quick view in our scripts:
@@ -84,10 +84,15 @@ To help community benchmarking, we provide our weights including fastflow3d, def
8484

8585
You can view Wandb dashboard for the training and evaluation results or [run/submit to av2 leaderboard to get official results](assets/README.md#leaderboard-submission).
8686

87+
88+
8789
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.
8890
```bash
89-
python 2_eval.py checkpoint=/home/kin/model.ckpt av2_mode=val # it will directly prints all metric
90-
python 2_eval.py checkpoint=/home/kin/model.ckpt av2_mode=test # it will output the av2_submit.zip for you to submit to leaderboard
91+
# downloaded pre-trained weight, or train by yourself
92+
wget https://zenodo.org/records/12173874/files/deflow_best.ckpt
93+
94+
python 2_eval.py checkpoint=/home/kin/deflow_best.ckpt av2_mode=val # it will directly prints all metric
95+
python 2_eval.py checkpoint=/home/kin/deflow_best.ckpt av2_mode=test # it will output the av2_submit.zip for you to submit to leaderboard
9196
```
9297

9398
Check all detailed result files (presented in our paper Table 1) in [this discussion](https://github.com/KTH-RPL/DeFlow/discussions/2).
@@ -111,16 +116,19 @@ evalai challenge 2010 phase 4018 submit --file av2_submit.zip --large --private
111116
We provide a script to visualize the results of the model. You can specify the checkpoint path and the data path to visualize the results. The step is quickly similar to evaluation.
112117

113118
```bash
114-
python 3_vis.py checkpoint=/home/kin/model.ckpt dataset_path=/home/kin/data/av2/preprocess/sensor/vis
119+
# downloaded pre-trained weight, or train by yourself
120+
wget https://zenodo.org/records/12173874/files/deflow_best.ckpt
121+
122+
python 3_vis.py checkpoint=/home/kin/deflow_best.ckpt dataset_path=/home/kin/data/av2/preprocess/sensor/vis
115123

116124
# Then terminal will tell you the command you need run. For example here is the output of the above:
117-
Model: DeFlow, Checkpoint from: /home/kin/model_zoo/deflow.ckpt
118-
We already write the estimate flow: deflow into the dataset, please run following commend to visualize the flow. Copy and paste it to your terminal:
119-
python tests/scene_flow.py --flow_mode 'deflow' --data_dir /home/kin/data/av2/preprocess/sensor/mini
125+
Model: DeFlow, Checkpoint from: /home/kin/deflow_best.ckpt
126+
We already write the estimate flow: deflow_best into the dataset, please run following commend to visualize the flow. Copy and paste it to your terminal:
127+
python tests/scene_flow.py --flow_mode 'deflow_best' --data_dir /home/kin/data/av2/preprocess/sensor/mini
120128
Enjoy! ^v^ ------
121129

122130
# Then run the command in the terminal:
123-
python tests/scene_flow.py --flow_mode 'deflow' --data_dir /home/kin/data/av2/preprocess/sensor/mini
131+
python tests/scene_flow.py --flow_mode 'deflow_best' --data_dir /home/kin/data/av2/preprocess/sensor/mini
124132
```
125133

126134
Note: ego_motion already compensated, so the visualization is more clear.

0 commit comments

Comments
 (0)