Skip to content

Commit c0ebc92

Browse files
committed
conf: update deltaflow conf files
update README to show the progress.
1 parent 99b1103 commit c0ebc92

3 files changed

Lines changed: 49 additions & 13 deletions

File tree

README.md

Lines changed: 33 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -6,28 +6,50 @@ DeltaFlow
66
<!-- [![poster](https://img.shields.io/badge/ECCV24|Poster-6495ed?style=flat&logo=Shotcut&logoColor=wihte)](https://hkustconnect-my.sharepoint.com/:b:/g/personal/qzhangcb_connect_ust_hk/EWyWD-tAX4xIma5U7ZQVk9cBVjsFv0Y_jAC2G7xAB-w4cg?e=c3FbMg) -->
77
<!-- [![video](https://img.shields.io/badge/video-YouTube-FF0000?logo=youtube&logoColor=white)](https://youtu.be/fQqx2IES-VI) -->
88

9-
Status: Under Review 🚀
10-
11-
We will release the full codebase, trained models, supplementary materials and all baselines upon acceptance and publication.
12-
13-
Stay tuned and feel free to star ⭐ this repository to get notified when we publish.
14-
15-
16-
🔗 While you wait, explore our codebase: [KTH-RPL/OpenSceneFlow](https://github.com/KTH-RPL/OpenSceneFlow).
17-
189
**News w. TBD**:
1910

11+
Note (2025/09/18): We got accepted by NeurIPS 2025 and it's **spotlighted**! 🎉🎉🎉 Working on release the code here.
12+
2013
- 2025/08/24: I'm updating some codes for early release.
2114
- [x] 2025/08/24: Updating train data augmentation as illustrated in the DeltaFlow paper.
2215
- [x] 2025/08/25: Updating paper preprint link.
2316
- [x] 2025/09/05: Merged the latest commit from OpenSceneFlow codebase to DeltaFlow for afterward unified merged.
24-
- [ ] DeltaFlow Model python file.
17+
- [x] 2025/09/25: DeltaFlow Model python file and config file.
2518
- [ ] pre-trained weights upload.
2619
- [ ] DeltaFlow Loss fn.
2720
- [ ] Merged into [OpenSceneFlow](https://github.com/KTH-RPL/OpenSceneFlow)
2821

29-
## Cite & Acknowledgements
22+
## Quick Run
23+
24+
### Training
25+
26+
1. Prepare the demo train and val data:
27+
```bash
28+
# around 1.3G
29+
wget https://huggingface.co/kin-zhang/OpenSceneFlow/resolve/main/demo-data-v2.zip
30+
unzip demo-data-v2.zip -d /home/kin/data/av2/h5py # to your data path
31+
```
32+
33+
2. Follow the [OpenSceneFlow](https://github.com/KTH-RPL/OpenSceneFlow/tree/main?tab=readme-ov-file#0-installation) to setup the environment.
3034

35+
3. Run the training with the following command (modify the data path accordingly):
36+
```bash
37+
python train.py model=deltaflow batch_size=4 num_frames=5 voxel_size="[0.15,0.15,0.15]" point_cloud_range="[-38.4,-38.4,-3,38.4,38.4,3]" optimizer.lr=2e-4 train_data=${demo_train_data_path} val_data=${demo_val_data_path}
38+
```
39+
### Evaluation
40+
41+
I will provide the pre-trained weights soon. Then you can run the evaluation with the following command:
42+
```bash
43+
python eval.py checkpoint=${path_to_pretrained_weights} dataset_path=${demo_data_path}
44+
```
45+
46+
### Visualization
47+
Please refer to the [OpenSceneFlow](https://github.com/KTH-RPL/OpenSceneFlow/tree/main?tab=readme-ov-file#4-visualization) for visualization instructions.
48+
49+
50+
51+
52+
## Cite & Acknowledgements
3153
```
3254
@article{zhang2025deltaflow,
3355
title={{DeltaFlow}: An Efficient Multi-frame Scene Flow Estimation Method},
@@ -36,5 +58,4 @@ Stay tuned and feel free to star ⭐ this repository to get notified when we pub
3658
journal={arXiv preprint arXiv:2508.17054},
3759
}
3860
```
39-
4061
This work was partially supported by the Wallenberg AI, Autonomous Systems and Software Program (WASP) funded by the Knut and Alice Wallenberg Foundation and Prosense (2020-02963) funded by Vinnova.

conf/config.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ gradient_clip_val: 5.0
2828
# optimizer ==> Adam
2929
optimizer:
3030
name: Adam # [Adam, AdamW]
31-
lr: 1e-3
31+
lr: 2e-4
3232
loss_fn: deflowLoss # choices: [ff3dLoss, zeroflowLoss, deflowLoss, seflowLoss]
3333
# add_seloss: {chamfer_dis: 1.0, static_flow_loss: 1.0, dynamic_chamfer_dis: 1.0, cluster_based_pc0pc1: 1.0}
3434
# ssl_label:

conf/model/deltaflow.yaml

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
name: deltaflow
2+
3+
target:
4+
_target_: src.models.DeltaFlow
5+
voxel_size: ${voxel_size}
6+
point_cloud_range: ${point_cloud_range}
7+
num_frames: ${num_frames}
8+
planes: [16, 32, 64, 128, 256, 256, 128, 64, 32, 16] # 1st is #input channel, last is #output channel
9+
# num_layer: [1, 1, 1, 1, 1, 1, 1, 1, 1] # the smallest model
10+
num_layer: [2, 2, 2, 2, 2, 2, 2, 2, 2] # MinkUnet 18
11+
# num_layer: [2, 3, 4, 6, 2, 2, 2, 2, 2] # MinkUnet 34
12+
decay_factor: 0.4
13+
decoder_option: default # choices: [default, deflow]
14+
15+
val_monitor: val/Dynamic/Mean

0 commit comments

Comments
 (0)