Skip to content

Commit 3e77e2b

Browse files
committed
docs(README): update model link to the latest code.
1 parent 3bf1121 commit 3e77e2b

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
@@ -8,10 +8,10 @@ SeFlow: A Self-Supervised Scene Flow Method in Autonomous Driving
88

99
![](assets/docs/seflow_arch.png)
1010

11-
2024/07/16 17:18: Most of codes already uploaded and tested. You can to try training directly by [downloading](https://zenodo.org/records/12751363) demo data or pretrained weight for evaluation.
11+
2024/07/16 17:18: Most of codes already uploaded and tested. You can to try training directly by [downloading](https://zenodo.org/records/13744999) demo data or pretrained weight for evaluation.
1212
The process script will be public when the paper is published officially.
1313

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

1616
Task: __Self-Supervised__ Scene Flow Estimation in Autonomous Driving. No human-label needed. Real-time inference (15-20Hz in RTX3090).
1717

@@ -79,7 +79,7 @@ Note: Prepare raw data and process train data only needed run once for the task.
7979
### Data Preparation
8080

8181
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.
82-
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).
82+
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).
8383

8484
```bash
8585
wget https://zenodo.org/record/12751363/files/demo_data.zip
@@ -112,7 +112,7 @@ Train SeFlow needed to specify the loss function, we set the config of our best
112112
python train.py model=deflow lr=2e-4 epochs=9 batch_size=16 loss_fn=seflowLoss "add_seloss={chamfer_dis: 1.0, static_flow_loss: 1.0, dynamic_chamfer_dis: 1.0, cluster_based_pc0pc1: 1.0}" "model.target.num_iters=2" "model.val_monitor=val/Dynamic/Mean"
113113
```
114114

115-
Or you can directly download the pre-trained weight from [Zenodo](https://zenodo.org/records/12751363/files/seflow_best.ckpt) and skip the training step.
115+
Or you can directly download the pre-trained weight from [Zenodo](https://zenodo.org/records/13744999/files/seflow_best.ckpt) and skip the training step.
116116

117117
### Other Benchmark Models
118118

@@ -135,7 +135,7 @@ Since in training, we save all hyper-parameters and model checkpoints, the only
135135

136136
```bash
137137
# downloaded pre-trained weight, or train by yourself
138-
wget https://zenodo.org/records/12751363/files/seflow_best.ckpt
138+
wget https://zenodo.org/records/13744999/files/seflow_best.ckpt
139139

140140
# it will directly prints all metric
141141
python eval.py checkpoint=/home/kin/seflow_best.ckpt av2_mode=val

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)