Merge pull request #85 from hmorimitsu/v041

V041

Author: hmorimitsu

README.md

@@ -24,6 +24,17 @@ This is still under development, so some things may not work as intended. I plan
 
 ## What's new
 
+###  - v0.4.1
+
+- Added new models:
+  - DPFlow [https://arxiv.org/abs/2503.14880](https://arxiv.org/abs/2503.14880)
+  - StreamFlow [https://arxiv.org/abs/2311.17099](https://arxiv.org/abs/2311.17099)
+- Additions / fixes:
+  - Add 4K read mode for Spring dataset
+  - Add missing arguments for Middlebury-ST and VIPER datasets
+  - Fix RAFT config files
+  - Fix RAPIDFlow config files
+
 ###  - v0.4.0
 
 Major update to support Lightning 2 (finally!). However, it also introduces breaking changes from the previous v0.3 code. See the details below.
@@ -92,6 +103,7 @@ Major update to support Lightning 2 (finally!). However, it also introduces brea
 - CSFlow [https://arxiv.org/abs/2202.00909](https://arxiv.org/abs/2202.00909)
 - DICL-Flow [https://arxiv.org/abs/2010.14851](https://arxiv.org/abs/2010.14851)
 - DIP [https://arxiv.org/abs/2204.00330](https://arxiv.org/abs/2204.00330)
+- DPFlow [https://arxiv.org/abs/2503.14880](https://arxiv.org/abs/2503.14880)
 - FastFlowNet [https://arxiv.org/abs/2103.04524](https://arxiv.org/abs/2103.04524)
 - Flow1D [https://arxiv.org/abs/2103.04524](https://arxiv.org/abs/2103.04524)
 - FlowFormer [https://arxiv.org/abs/2203.16194](https://arxiv.org/abs/2203.16194)
@@ -126,6 +138,7 @@ Major update to support Lightning 2 (finally!). However, it also introduces brea
 - SKFlow [https://arxiv.org/abs/2205.14623](https://arxiv.org/abs/2205.14623)
 - SplatFlow [https://arxiv.org/abs/2306.08887](https://arxiv.org/abs/2306.08887)
 - STaRFlow [https://arxiv.org/abs/2007.05481](https://arxiv.org/abs/2007.05481)
+- StreamFlow [https://arxiv.org/abs/2311.17099](https://arxiv.org/abs/2311.17099)
 - VCN [https://papers.nips.cc/paper/2019/file/bbf94b34eb32268ada57a3be5062fe7d-Paper.pdf](https://papers.nips.cc/paper/2019/file/bbf94b34eb32268ada57a3be5062fe7d-Paper.pdf)
 - VideoFlow [https://arxiv.org/abs/2303.08340](https://arxiv.org/abs/2303.08340)
 

configs/results/validate_all.yaml

@@ -4,7 +4,7 @@
 all: true
 select: null
 exclude: null
-ckpt_path: things
+ckpt_path: null
 output_path: outputs/validate
 write_outputs: false
 show: false
@@ -18,12 +18,14 @@ fp16: false
 seq_val_mode: all
 write_individual_metrics: false
 epe_clip: 5.0
+metric_exclude:
+  - wauc
 seed_everything: true
 data:
   predict_dataset: null
   test_dataset: null
   train_dataset: null
-  val_dataset: sintel-clean+sintel-final+kitti-2015
+  val_dataset: sintel-clean-occ+sintel-final-occ+kitti-2012+kitti-2015
   train_batch_size: null
   train_num_workers: 4
   train_crop_size: null

configs/results/validate_all_things.yaml

@@ -18,6 +18,7 @@ fp16: false
 seq_val_mode: all
 write_individual_metrics: false
 epe_clip: 5.0
+metric_exclude: null
 seed_everything: true
 data:
   predict_dataset: null

docs/source/_static/plot-memory(gb)-fp32-sintel-final-occ-val_epe.html

@@ -65,6 +65,13 @@ Paper: **Kubric: A scalable dataset generator** - `https://arxiv.org/abs/2203.03
 
 Dataset link: `https://github.com/google-research/kubric <https://github.com/google-research/kubric>`_
 
+Kubric-NK
+---------
+
+Paper: **DPFlow: Adaptive Optical Flow Estimation with a Dual-Pyramid Framework** - `https://arxiv.org/abs/2503.14880 <https://arxiv.org/abs/2503.14880>`_
+
+Dataset link: `https://github.com/hmorimitsu/kubric-nk <https://github.com/hmorimitsu/kubric-nk>`_
+
 Middlebury
 ----------