Skip to content

Commit 22897f2

Browse files
committed
docs(link): hugging face download link
* the download speed is much faster then Zenodo.
2 parents 2b4b272 + da14393 commit 22897f2

5 files changed

Lines changed: 12 additions & 10 deletions

File tree

.github/issue_stale.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# Number of days of inactivity before an issue becomes stale
2-
daysUntilStale: 60
2+
daysUntilStale: 7
33
# Number of days of inactivity before a stale issue is closed
44
daysUntilClose: 7
55
# Issues with these labels will never be considered stale

README.md

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -9,11 +9,13 @@ DeFlow: Decoder of Scene Flow Network in Autonomous Driving
99

1010
Task: Scene Flow Estimation in Autonomous Driving.
1111

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.
13+
1214
📜 2024/07/24: Merging SeFlow & DeFlow code together, lighter setup and easier running.
1315

1416
🔥 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.
1517

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).
1719
Check usage in [2. Evaluation](#2-evaluation) or [3. Visualization](#3-visualization).
1820

1921
**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.
8183
### Data Preparation
8284

8385
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).
8587

8688
```bash
87-
wget https://zenodo.org/record/12751363/files/demo_data.zip
89+
wget https://huggingface.co/kin-zhang/OpenSceneFlow/resolve/main/demo_data.zip
8890
unzip demo_data.zip -p /home/kin/data/av2
8991
```
9092

@@ -114,7 +116,7 @@ python train.py model=fastflow3d lr=4e-5 epochs=20 batch_size=16 loss_fn=ff3dLos
114116
> 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.
115117
> We suggest afterward researchers or users to use the setting here (larger lr and smaller epoch) for faster converge and better performance.
116118
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).
118120
These checkpoints also include parameters and status of that epoch inside it. If you are interested in weights of ablation studies, please contact us.
119121

120122
## 2. Evaluation
@@ -124,7 +126,7 @@ You can view Wandb dashboard for the training and evaluation results or run/subm
124126
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.
125127
```bash
126128
# downloaded pre-trained weight, or train by yourself
127-
wget https://zenodo.org/records/13744999/files/deflow_best.ckpt
129+
wget https://huggingface.co/kin-zhang/OpenSceneFlow/resolve/main/deflow_best.ckpt
128130

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

158160
```bash
159161
# downloaded pre-trained weight, or train by yourself
160-
wget https://zenodo.org/records/13744999/files/deflow_best.ckpt
162+
wget https://huggingface.co/kin-zhang/OpenSceneFlow/resolve/main/deflow_best.ckpt
161163

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

dataprocess/README.md

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

9797
```bash
98-
wget https://zenodo.org/records/13744999/files/waymo_map.tar.gz
98+
wget https://huggingface.co/kin-zhang/OpenSceneFlow/resolve/main/waymo_map.tar.gz
9999
tar -xvf waymo_map.tar.gz -C /home/kin/data/waymo/flowlabel
100100
# you will see there is a `map` folder in the `flowlabel` folder now.
101101
```

tools/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,5 +32,5 @@ python tools/zero2ours.py --model_path /home/kin/nsfp_distilatation_3x_49_epochs
3232
```
3333

3434
- model_path, you can download from: [kylevedder/zeroflow_weights](https://github.com/kylevedder/zeroflow_weights/tree/master/argo)
35-
- reference_path, you can download fastflow3d model from: [zendo](https://zenodo.org/records/12632962)
35+
- reference_path, you can download fastflow3d model from: [HuggingFace](https://huggingface.co/kin-zhang/OpenSceneFlow/blob/main/fastflow3d_best.ckpt)
3636
- output_path, the converted model path. You can then run any evaluation script and visualization script with the converted model.

tools/zerof2ours.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
def main(
77
# download from: https://github.com/kylevedder/zeroflow_weights/tree/master/argo
88
model_path: str = "/home/kin/nsfp_distilatation_3x_49_epochs.ckpt",
9-
# download from: https://zenodo.org/records/12632962
9+
# download from: https://huggingface.co/kin-zhang/OpenSceneFlow/blob/main/fastflow3d_best.ckpt
1010
reference_path: str = "/home/kin/fastflow3d.ckpt",
1111
# new output weight file
1212
output_path: str = "/home/kin/zeroflow3x.ckpt",

0 commit comments

Comments
 (0)