You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: README.md
+8-6Lines changed: 8 additions & 6 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -9,11 +9,13 @@ DeFlow: Decoder of Scene Flow Network in Autonomous Driving
9
9
10
10
Task: Scene Flow Estimation in Autonomous Driving.
11
11
12
+
🤗 2024/11/18 16:17: Update model and demo data download link through HuggingFace, Personally I found `wget` from HuggingFace link is much faster than Zenodo.
🔥 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.
15
17
16
-
Pre-trained weights for models are available in [Zenodo](https://zenodo.org/records/13744999).
18
+
Pre-trained weights for models are available in [Zenodo](https://zenodo.org/records/13744999)/[HuggingFace](https://huggingface.co/kin-zhang/OpenSceneFlow).
17
19
Check usage in [2. Evaluation](#2-evaluation) or [3. Visualization](#3-visualization).
18
20
19
21
**Scripts** quick view in our scripts:
@@ -81,10 +83,10 @@ Note: Prepare raw data and process train data only needed run once for the task.
81
83
### Data Preparation
82
84
83
85
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/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).
86
+
You can download it from [Zenodo](https://zenodo.org/records/13744999/files/demo_data.zip)/[HuggingFace](https://huggingface.co/kin-zhang/OpenSceneFlow/blob/main/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).
> 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.
115
117
> We suggest afterward researchers or users to use the setting here (larger lr and smaller epoch) for faster converge and better performance.
116
118
117
-
To help community benchmarking, we provide our weights including fastflow3d, deflow in [Zendo](https://zenodo.org/records/13744999).
119
+
To help community benchmarking, we provide our weights including fastflow3d, deflow in [HuggingFace](https://huggingface.co/kin-zhang/OpenSceneFlow).
118
120
These checkpoints also include parameters and status of that epoch inside it. If you are interested in weights of ablation studies, please contact us.
119
121
120
122
## 2. Evaluation
@@ -124,7 +126,7 @@ You can view Wandb dashboard for the training and evaluation results or run/subm
124
126
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.
125
127
```bash
126
128
# downloaded pre-trained weight, or train by yourself
0 commit comments