Merge pull request #64 from hmorimitsu/neuflow

Add model NeuFlow

README.md

@@ -85,6 +85,7 @@ This is still under development, so some things may not work as intended. I plan
 - MaskFlownet [https://arxiv.org/abs/2003.10955](https://arxiv.org/abs/2003.10955)
 - MatchFlow [https://arxiv.org/abs/2303.08384](https://arxiv.org/abs/2303.08384)
 - MS-RAFT+ [https://arxiv.org/abs/2210.16900](https://arxiv.org/abs/2210.16900)
+- NeuFlow [https://arxiv.org/abs/2403.10425](https://arxiv.org/abs/2403.10425)
 - PWCNet [https://arxiv.org/abs/1709.02371](https://arxiv.org/abs/1709.02371)
 - RAFT [https://arxiv.org/abs/2003.12039](https://arxiv.org/abs/2003.12039)
 - RAPIDFlow [https://hmorimitsu.com/publication/2024-icra-rapidflow/](https://hmorimitsu.com/publication/2024-icra-rapidflow/)

docs/source/models/checkpoint_list.csv

@@ -100,6 +100,8 @@ matchflow_gma-sintel-683422f4.ckpt,matchflow-g-sintel.pth,
 matchflow_gma-things-49295bd8.ckpt,matchflow-g-things.pth,
 matchflow_raft-things-bf560032.ckpt,matchflow-r-things.pth,
 ms_raft_plus-mixed-2bb01f62.ckpt,mixed.pth,https://github.com/cv-stuttgart/MS_RAFT_plus
+neuflow-things-c402aa7a.ckpt,neuflow_things.pth,https://github.com/neufieldrobotics/neuflow
+neuflow-sintel-0d969ea2.ckpt,neuflow_sintel.pth,
 pwcdcnet-sintel-c7d08a46.ckpt,pwc_net.pth.tar,https://github.com/NVlabs/PWC-Net/tree/master/PyTorch
 pwcdcnet-things-cc223701.ckpt,pwc_net_chairs.pth.tar,
 pwcnet-sintel-533815e5.ckpt,pwc_net.pth.tar,

docs/source/models/models_list.rst

@@ -277,6 +277,17 @@ MS-RAFT+
 
 - Model names: ``ms_raft+``
 
+NeuFlow
+-------
+
+`[source code] <https://github.com/hmorimitsu/ptlflow/tree/main/ptlflow/models/neuflow>`__
+
+- Paper: **NeuFlow: Real-time, High-accuracy Optical Flow Estimation on Robots Using Edge Devices** - `https://arxiv.org/abs/2403.10425 <https://arxiv.org/abs/2403.10425>`_
+
+- Reference code: `https://github.com/neufieldrobotics/neuflow <https://github.com/neufieldrobotics/neuflow>`_
+
+- Model names: ``neuflow``
+
 PWCNet
 ------
 

docs/source/results/metrics_all.csv

@@ -109,6 +109,8 @@ matchflow,sintel,0.494,0.941,0.977,0.985,2.012,2.061,0.245,0.704,0.967,0.868,0.9
 matchflow,things,1.095,0.918,0.964,0.974,3.367,4.178,0.499,0.635,0.947,0.818,0.983,0.87,0.989,16.884,1.557,2.416,0.866,0.926,0.944,7.016,6.368,1.663,0.564,0.894,0.753,0.947,0.811,0.961,23.506,4.976,1.738,0.753,0.905,0.941,8.001,4.075,0.662,0.828,0.872,15.363,,,,,,
 matchflow_raft,things,1.15,0.915,0.963,0.973,3.502,4.356,0.532,0.624,0.944,0.812,0.982,0.865,0.989,17.509,1.662,2.633,0.864,0.925,0.943,7.131,6.951,1.79,0.558,0.892,0.748,0.946,0.809,0.961,23.967,5.069,1.814,0.765,0.915,0.945,7.042,4.159,0.679,0.846,0.888,13.317,,,,,,
 ms_raft+,mixed,0.648,0.961,0.979,0.984,1.939,2.664,0.243,0.775,0.985,0.881,0.993,0.911,0.995,10.58,0.627,1.107,0.913,0.954,0.967,4.172,3.525,0.654,0.696,0.936,0.836,0.97,0.878,0.98,15.02,2.64,1.015,0.87,0.956,0.972,3.054,0.679,0.877,0.964,0.981,1.724,,,,,,
+neuflow,things,1.66,0.837,0.935,0.956,6.148,5.449,0.985,0.472,0.87,0.734,0.959,0.816,0.975,25.104,3.818,3.126,0.787,0.892,0.921,10.424,7.954,2.244,0.415,0.817,0.664,0.917,0.751,0.942,32.279,7.933,5.056,0.525,0.789,0.848,20.281,11.111,0.45,0.682,0.747,30.918,,,,,,
+neuflow,sintel,0.988,0.871,0.952,0.97,4.411,3.414,0.609,0.552,0.901,0.787,0.973,0.856,0.985,19.69,2.49,1.294,0.838,0.93,0.955,6.463,3.933,0.889,0.517,0.865,0.756,0.95,0.831,0.971,22.786,4.53,2.297,0.673,0.887,0.931,9.489,2.06,0.645,0.868,0.922,9.539,,,,,,
 pwcnet,things,2.643,0.801,0.909,0.934,8.811,7.733,1.697,0.369,0.835,0.65,0.937,0.749,0.958,33.673,6.064,4.059,0.749,0.871,0.904,12.592,9.855,2.943,0.337,0.78,0.599,0.898,0.7,0.927,38.921,9.872,4.042,0.545,0.788,0.854,20.243,10.288,0.452,0.662,0.731,33.014,,,,,,
 pwcnet,sintel,1.849,0.872,0.936,0.952,6.164,6.205,1.12,0.502,0.906,0.724,0.96,0.795,0.972,26.298,3.858,2.338,0.836,0.912,0.935,8.387,6.74,1.597,0.47,0.866,0.694,0.936,0.772,0.955,29.229,6.096,1.93,0.76,0.915,0.946,6.921,3.122,0.696,0.869,0.911,10.184,,,,,,
 pwcnet_nodc,things,3.329,0.724,0.88,0.917,11.639,9.577,2.164,0.271,0.758,0.549,0.915,0.678,0.947,43.948,8.154,4.769,0.675,0.843,0.885,15.416,11.502,3.454,0.257,0.704,0.516,0.874,0.636,0.913,47.481,12.212,5.29,0.503,0.743,0.811,25.053,12.274,0.414,0.633,0.704,36.11,,,,,,

docs/source/results/model_benchmark-all.csv

@@ -44,6 +44,7 @@ maskflownet_s,10.514,162.96,500,1000,500000,40.171,0.936,50.655,1.2
 matchflow,15.446,2994.565,500,1000,500000,,,440.889,2.196
 matchflow_raft,14.824,2284.576,500,1000,500000,,,373.69,2.022
 ms_raft+,16.177,9658.707,500,1000,500000,795.76,1.709,1197.577,2.965
+neuflow,3.847,76.246,500,1000,500000,9.22,0.52,13.435,0.508
 pwcnet,9.374,157.505,500,1000,500000,39.595,0.917,47.103,1.209
 pwcnet_nodc,8.243,96.719,500,1000,500000,36.401,0.874,41.83,1.147
 raft,5.258,1780.45,500,1000,500000,140.326,1.233,256.396,1.543

docs/source/results/result_plots.rst

@@ -16,7 +16,7 @@ Memory usage x Sintel Final EPE
 
 .. raw:: html
 
-  <iframe src="../_static/plot-memory(gb)-fp32-sintel-final-occ-val_outlier.html" height="600px" width="100%"></iframe>
+  <iframe src="../_static/plot-memory(gb)-fp32-sintel-final-occ-val_epe.html" height="600px" width="100%"></iframe>
 
 Sintel Final EPE x KITTI 2015 Fl-All
 ====================================

docs/source/results/summarized_metrics-epe.csv

@@ -82,17 +82,17 @@ irr_pwcnet,things,3.128,4.413,6.103,14.855
 irr_pwcnet_irr,things,2.748,4.019,4.854,12.417
 lcv_raft,things,1.83,2.876,2.432,5.831
 lcv_raft,chairs,2.331,4.135,4.647,9.61
-liteflownet,things,2.583,4.062,4.275,10.64
 liteflownet,sintel,1.498,1.894,3.47,9.993
+liteflownet,things,2.583,4.062,4.275,10.64
 liteflownet,kitti,4.594,5.945,1.101,1.737
 liteflownet2,sintel,1.557,1.946,1.596,3.12
 liteflownet2_pseudoreg,kitti,4.578,5.712,0.98,1.355
 liteflownet3,sintel,1.451,1.917,1.619,3.302
 liteflownet3_pseudoreg,kitti,4.455,5.6,0.978,1.388
 liteflownet3s,sintel,1.569,2.075,1.78,3.602
 liteflownet3s_pseudoreg,kitti,4.275,5.518,1.026,1.524
-llaflow,kitti,3.815,5.773,1.326,0.642
 llaflow,sintel,0.554,0.846,1.317,1.31
+llaflow,kitti,3.815,5.773,1.326,0.642
 llaflow,chairs,2.296,4.52,4.479,9.748
 llaflow,things,1.275,2.725,1.882,4.702
 llaflow_raft,chairs,2.176,4.832,5.868,12.314
@@ -101,70 +101,72 @@ llaflow_raft,sintel,0.623,0.991,1.257,1.364
 llaflow_raft,kitti,5.635,8.36,1.292,0.579
 maskflownet,kitti,3.576,4.81,0.965,1.626
 maskflownet,sintel,1.443,1.922,1.462,1.846
-maskflownet_s,things,2.535,3.884,4.383,10.773
 maskflownet_s,sintel,1.502,2.002,1.708,2.445
+maskflownet_s,things,2.535,3.884,4.383,10.773
 matchflow,things,1.095,2.416,1.738,4.075
-matchflow,sintel,0.494,0.796,1.158,1.24
-matchflow,chairs,2.375,4.581,5.635,12.956
 matchflow,kitti,2.984,4.585,1.125,0.531
+matchflow,chairs,2.375,4.581,5.635,12.956
+matchflow,sintel,0.494,0.796,1.158,1.24
 matchflow_raft,things,1.15,2.633,1.814,4.159
 ms_raft+,mixed,0.648,1.107,1.015,0.679
+neuflow,things,1.66,3.126,5.056,11.111
+neuflow,sintel,0.988,1.294,2.297,2.06
 pwcnet,things,2.643,4.059,4.042,10.288
 pwcnet,sintel,1.849,2.338,1.93,3.122
-pwcnet_nodc,things,3.329,4.769,5.29,12.274
 pwcnet_nodc,sintel,2.231,2.765,3.234,6.093
-raft,kitti,4.544,6.157,1.283,0.643
-raft,sintel,0.763,1.218,1.338,1.555
+pwcnet_nodc,things,3.329,4.769,5.29,12.274
 raft,chairs,2.286,4.536,4.717,9.88
 raft,things,1.458,2.684,2.135,5.018
+raft,sintel,0.763,1.218,1.338,1.555
+raft,kitti,4.544,6.157,1.283,0.643
 raft_small,things,2.123,3.256,3.075,7.472
-rapidflow,chairs,2.502,3.672,4.087,10.42
 rapidflow,things,1.575,2.893,2.323,5.872
 rapidflow,sintel,0.949,1.424,1.361,1.594
+rapidflow,chairs,2.502,3.672,4.087,10.42
 rapidflow,kitti,2.435,3.502,1.379,1.069
+rapidflow_it1,kitti,4.947,5.723,3.554,3.37
 rapidflow_it1,sintel,2.918,3.274,3.886,5.106
-rapidflow_it1,chairs,5.039,5.904,11.006,20.895
 rapidflow_it1,things,4.089,5.188,8.508,15.699
-rapidflow_it1,kitti,4.947,5.723,3.554,3.37
-rapidflow_it2,kitti,3.683,4.631,2.329,2.289
-rapidflow_it2,sintel,1.959,2.375,2.374,3.254
+rapidflow_it1,chairs,5.039,5.904,11.006,20.895
 rapidflow_it2,chairs,3.684,4.73,6.8,15.684
 rapidflow_it2,things,2.762,3.963,5.078,11.138
-rapidflow_it3,chairs,3.011,4.202,5.469,13.685
-rapidflow_it3,things,2.026,3.355,3.692,9.169
+rapidflow_it2,sintel,1.959,2.375,2.374,3.254
+rapidflow_it2,kitti,3.683,4.631,2.329,2.289
 rapidflow_it3,sintel,1.286,1.815,1.777,2.347
 rapidflow_it3,kitti,3.063,4.142,1.768,1.504
+rapidflow_it3,chairs,3.011,4.202,5.469,13.685
+rapidflow_it3,things,2.026,3.355,3.692,9.169
 rapidflow_it6,chairs,2.61,3.806,4.47,11.497
 rapidflow_it6,things,1.685,2.995,2.616,6.768
 rapidflow_it6,sintel,1.014,1.492,1.431,1.688
 rapidflow_it6,kitti,2.582,3.624,1.45,1.112
 rpknet,things,1.12,2.449,1.674,3.783
-rpknet,sintel,0.552,0.813,1.192,1.257
 rpknet,chairs,2.406,3.588,3.965,10.878
 rpknet,kitti,1.478,2.37,1.171,0.946
-scopeflow,chairs,2.529,3.908,3.793,10.747
-scopeflow,things,1.847,3.291,2.76,8.268
+rpknet,sintel,0.552,0.813,1.192,1.257
 scopeflow,kitti,8.259,10.676,0.911,1.308
 scopeflow,sintel,1.615,2.477,1.984,5.795
-separableflow,universal,0.685,1.016,1.027,1.103
-separableflow,kitti,3.1,5.13,0.761,0.62
-separableflow,sintel,0.642,0.956,1.509,1.468
+scopeflow,chairs,2.529,3.908,3.793,10.747
+scopeflow,things,1.847,3.291,2.76,8.268
 separableflow,things,1.295,2.593,2.094,4.733
+separableflow,sintel,0.642,0.956,1.509,1.468
+separableflow,kitti,3.1,5.13,0.761,0.62
+separableflow,universal,0.685,1.016,1.027,1.103
 skflow,kitti,2.614,4.003,1.201,0.519
 skflow,sintel,0.524,0.782,1.2,1.227
 skflow,things,1.205,2.441,1.758,4.238
-starflow,things,1.817,3.452,3.868,9.316
-starflow,sintel,1.599,2.056,3.736,8.155
 starflow,kitti,5.07,7.097,1.78,2.951
-vcn,kitti,5.052,7.499,0.937,1.424
-vcn,sintel,1.671,2.44,1.981,4.213
-vcn,things,2.207,3.694,3.151,8.119
+starflow,sintel,1.599,2.056,3.736,8.155
+starflow,things,1.817,3.452,3.868,9.316
 vcn,chairs,2.816,4.013,3.976,10.268
+vcn,things,2.207,3.694,3.151,8.119
+vcn,sintel,1.671,2.44,1.981,4.213
+vcn,kitti,5.052,7.499,0.937,1.424
 vcn_small,chairs,3.22,4.395,4.966,11.863
 vcn_small,things,2.759,4.306,3.757,8.879
+videoflow_bof,kitti,1.847,3.139,6.329,6.73
 videoflow_bof,things_288960,1.155,2.394,5.172,17.009
 videoflow_bof,sintel,0.37,0.54,1.598,2.25
-videoflow_bof,kitti,1.847,3.139,6.329,6.73
 videoflow_mof,sintel,0.432,0.621,1.839,2.893
 videoflow_mof,things,1.057,2.474,7.524,23.696
 videoflow_mof,kitti,2.621,3.71,4.493,7.27

docs/source/results/summarized_metrics-epe_outlier.csv

@@ -82,17 +82,17 @@ irr_pwcnet,things,3.128,9.144,4.413,12.918,6.103,22.715,14.855,36.72
 irr_pwcnet_irr,things,2.748,8.808,4.019,12.846,4.854,22.949,12.417,35.509
 lcv_raft,things,1.83,5.069,2.876,8.541,2.432,11.102,5.831,18.928
 lcv_raft,chairs,2.331,7.311,4.135,11.401,4.647,29.915,9.61,38.444
-liteflownet,things,2.583,7.496,4.062,11.906,4.275,18.746,10.64,30.425
 liteflownet,sintel,1.498,5.039,1.894,7.4,3.47,12.322,9.993,23.174
+liteflownet,things,2.583,7.496,4.062,11.906,4.275,18.746,10.64,30.425
 liteflownet,kitti,4.594,11.889,5.945,15.726,1.101,3.849,1.737,6.243
 liteflownet2,sintel,1.557,5.421,1.946,7.671,1.596,5.853,3.12,9.625
 liteflownet2_pseudoreg,kitti,4.578,11.928,5.712,15.728,0.98,3.182,1.355,4.482
 liteflownet3,sintel,1.451,5.094,1.917,7.328,1.619,6.355,3.302,11.126
 liteflownet3_pseudoreg,kitti,4.455,11.439,5.6,15.297,0.978,3.289,1.388,4.717
 liteflownet3s,sintel,1.569,5.289,2.075,7.619,1.78,7.267,3.602,12.277
 liteflownet3s_pseudoreg,kitti,4.275,10.951,5.518,14.851,1.026,3.486,1.524,5.279
-llaflow,kitti,3.815,8.789,5.773,12.722,1.326,4.4,0.642,1.598
 llaflow,sintel,0.554,2.258,0.846,3.809,1.317,4.458,1.31,4.581
+llaflow,kitti,3.815,8.789,5.773,12.722,1.326,4.4,0.642,1.598
 llaflow,chairs,2.296,6.791,4.52,11.233,4.479,27.4,9.748,36.588
 llaflow,things,1.275,3.724,2.725,7.314,1.882,8.544,4.702,16.33
 llaflow_raft,chairs,2.176,6.862,4.832,12.643,5.868,38.753,12.314,44.8
@@ -101,70 +101,72 @@ llaflow_raft,sintel,0.623,2.502,0.991,4.353,1.257,4.285,1.364,4.806
 llaflow_raft,kitti,5.635,10.822,8.36,15.016,1.292,4.352,0.579,1.327
 maskflownet,kitti,3.576,9.122,4.81,12.526,0.965,2.953,1.626,6.088
 maskflownet,sintel,1.443,4.736,1.922,6.915,1.462,4.938,1.846,6.292
-maskflownet_s,things,2.535,8.117,3.884,12.012,4.383,20.73,10.773,32.764
 maskflownet_s,sintel,1.502,5.291,2.002,7.475,1.708,6.071,2.445,9.4
+maskflownet_s,things,2.535,8.117,3.884,12.012,4.383,20.73,10.773,32.764
 matchflow,things,1.095,3.367,2.416,7.016,1.738,8.001,4.075,15.363
-matchflow,sintel,0.494,2.012,0.796,3.587,1.158,3.937,1.24,4.307
-matchflow,chairs,2.375,6.867,4.581,11.613,5.635,29.183,12.956,39.803
 matchflow,kitti,2.984,8.547,4.585,13.2,1.125,3.71,0.531,1.111
+matchflow,chairs,2.375,6.867,4.581,11.613,5.635,29.183,12.956,39.803
+matchflow,sintel,0.494,2.012,0.796,3.587,1.158,3.937,1.24,4.307
 matchflow_raft,things,1.15,3.502,2.633,7.131,1.814,7.042,4.159,13.317
 ms_raft+,mixed,0.648,1.939,1.107,4.172,1.015,3.054,0.679,1.724
+neuflow,sintel,0.988,4.411,1.294,6.463,2.297,9.489,2.06,9.539
+neuflow,things,1.66,6.148,3.126,10.424,5.056,20.281,11.111,30.918
 pwcnet,things,2.643,8.811,4.059,12.592,4.042,20.243,10.288,33.014
 pwcnet,sintel,1.849,6.164,2.338,8.387,1.93,6.921,3.122,10.184
-pwcnet_nodc,things,3.329,11.639,4.769,15.416,5.29,25.053,12.274,36.11
 pwcnet_nodc,sintel,2.231,8.422,2.765,11.29,3.234,15.438,6.093,21.75
-raft,kitti,4.544,10.511,6.157,14.794,1.283,4.272,0.643,1.58
-raft,sintel,0.763,2.855,1.218,4.936,1.338,4.587,1.555,5.573
+pwcnet_nodc,things,3.329,11.639,4.769,15.416,5.29,25.053,12.274,36.11
 raft,chairs,2.286,6.984,4.536,12.069,4.717,29.809,9.88,36.698
 raft,things,1.458,4.173,2.684,7.721,2.135,9.257,5.018,17.046
+raft,sintel,0.763,2.855,1.218,4.936,1.338,4.587,1.555,5.573
+raft,kitti,4.544,10.511,6.157,14.794,1.283,4.272,0.643,1.58
 raft_small,things,2.123,6.684,3.256,9.933,3.075,14.645,7.472,24.658
-rapidflow,chairs,2.502,7.502,3.672,11.028,4.087,22.748,10.42,33.46
 rapidflow,things,1.575,4.73,2.893,8.573,2.323,9.557,5.872,17.72
 rapidflow,sintel,0.949,3.405,1.424,5.631,1.361,4.973,1.594,5.827
+rapidflow,chairs,2.502,7.502,3.672,11.028,4.087,22.748,10.42,33.46
 rapidflow,kitti,2.435,6.909,3.502,10.481,1.379,4.842,1.069,3.533
+rapidflow_it1,kitti,4.947,23.814,5.723,25.253,3.554,25.472,3.37,21.354
 rapidflow_it1,sintel,2.918,15.477,3.274,17.263,3.886,28.26,5.106,33.041
-rapidflow_it1,chairs,5.039,24.549,5.904,26.49,11.006,58.75,20.895,63.759
 rapidflow_it1,things,4.089,20.213,5.188,22.722,8.508,50.773,15.699,56.991
-rapidflow_it1,kitti,4.947,23.814,5.723,25.253,3.554,25.472,3.37,21.354
-rapidflow_it2,kitti,3.683,14.23,4.631,16.869,2.329,11.845,2.289,11.957
-rapidflow_it2,sintel,1.959,9.047,2.375,11.183,2.374,11.915,3.254,16.803
+rapidflow_it1,chairs,5.039,24.549,5.904,26.49,11.006,58.75,20.895,63.759
 rapidflow_it2,chairs,3.684,14.409,4.73,17.53,6.8,37.237,15.684,49.075
 rapidflow_it2,things,2.762,11.123,3.963,14.561,5.078,26.636,11.138,37.792
-rapidflow_it3,chairs,3.011,9.192,4.202,12.967,5.469,24.867,13.685,37.487
-rapidflow_it3,things,2.026,6.365,3.355,10.346,3.692,15.431,9.169,25.49
+rapidflow_it2,sintel,1.959,9.047,2.375,11.183,2.374,11.915,3.254,16.803
+rapidflow_it2,kitti,3.683,14.23,4.631,16.869,2.329,11.845,2.289,11.957
 rapidflow_it3,sintel,1.286,4.683,1.815,7.203,1.777,7.066,2.347,9.141
 rapidflow_it3,kitti,3.063,9.33,4.142,12.819,1.768,7.133,1.504,5.997
+rapidflow_it3,chairs,3.011,9.192,4.202,12.967,5.469,24.867,13.685,37.487
+rapidflow_it3,things,2.026,6.365,3.355,10.346,3.692,15.431,9.169,25.49
 rapidflow_it6,chairs,2.61,7.815,3.806,11.566,4.47,23.403,11.497,35.051
 rapidflow_it6,things,1.685,5.117,2.995,9.009,2.616,10.722,6.768,19.857
 rapidflow_it6,sintel,1.014,3.647,1.492,5.915,1.431,5.457,1.688,6.38
 rapidflow_it6,kitti,2.582,7.485,3.624,10.983,1.45,5.308,1.112,3.815
 rpknet,things,1.12,3.516,2.449,7.08,1.674,6.817,3.783,12.998
-rpknet,sintel,0.552,2.19,0.813,3.659,1.192,4.112,1.257,4.273
 rpknet,chairs,2.406,7.341,3.588,10.697,3.965,22.812,10.878,33.835
 rpknet,kitti,1.478,4.605,2.37,7.86,1.171,3.817,0.946,2.904
-scopeflow,chairs,2.529,7.188,3.908,10.923,3.793,17.146,10.747,30.812
-scopeflow,things,1.847,5.997,3.291,10.02,2.76,12.263,8.268,24.893
+rpknet,sintel,0.552,2.19,0.813,3.659,1.192,4.112,1.257,4.273
 scopeflow,kitti,8.259,15.355,10.676,18.921,0.911,2.971,1.308,4.309
 scopeflow,sintel,1.615,5.026,2.477,7.797,1.984,6.946,5.795,16.019
-separableflow,universal,0.685,2.862,1.016,4.828,1.027,3.346,1.103,3.816
-separableflow,kitti,3.1,9.298,5.13,15.2,0.761,2.218,0.62,1.455
-separableflow,sintel,0.642,2.747,0.956,4.516,1.509,4.889,1.468,5.337
+scopeflow,chairs,2.529,7.188,3.908,10.923,3.793,17.146,10.747,30.812
+scopeflow,things,1.847,5.997,3.291,10.02,2.76,12.263,8.268,24.893
 separableflow,things,1.295,4.103,2.593,7.697,2.094,8.673,4.733,15.977
+separableflow,sintel,0.642,2.747,0.956,4.516,1.509,4.889,1.468,5.337
+separableflow,kitti,3.1,9.298,5.13,15.2,0.761,2.218,0.62,1.455
+separableflow,universal,0.685,2.862,1.016,4.828,1.027,3.346,1.103,3.816
 skflow,kitti,2.614,8.148,4.003,12.221,1.201,3.897,0.519,1.036
 skflow,sintel,0.524,2.261,0.782,3.62,1.2,4.036,1.227,4.292
 skflow,things,1.205,3.744,2.441,7.168,1.758,7.727,4.238,15.127
-starflow,things,1.817,6.123,3.452,10.417,3.868,16.539,9.316,27.384
-starflow,sintel,1.599,5.442,2.056,8.081,3.736,13.214,8.155,23.316
 starflow,kitti,5.07,14.814,7.097,18.895,1.78,6.607,2.951,12.217
-vcn,kitti,5.052,12.651,7.499,17.272,0.937,2.713,1.424,4.893
-vcn,sintel,1.671,6.111,2.44,8.892,1.981,8.601,4.213,16.951
-vcn,things,2.207,7.3,3.694,11.394,3.151,13.986,8.119,24.956
+starflow,sintel,1.599,5.442,2.056,8.081,3.736,13.214,8.155,23.316
+starflow,things,1.817,6.123,3.452,10.417,3.868,16.539,9.316,27.384
 vcn,chairs,2.816,8.552,4.013,12.0,3.976,18.566,10.268,31.823
+vcn,things,2.207,7.3,3.694,11.394,3.151,13.986,8.119,24.956
+vcn,sintel,1.671,6.111,2.44,8.892,1.981,8.601,4.213,16.951
+vcn,kitti,5.052,12.651,7.499,17.272,0.937,2.713,1.424,4.893
 vcn_small,chairs,3.22,11.211,4.395,14.54,4.966,28.116,11.863,40.397
 vcn_small,things,2.759,9.877,4.306,13.872,3.757,20.483,8.879,31.81
+videoflow_bof,kitti,1.847,6.512,3.139,10.45,6.329,19.59,6.73,21.226
 videoflow_bof,things_288960,1.155,3.669,2.394,7.196,5.172,15.489,17.009,31.361
 videoflow_bof,sintel,0.37,1.58,0.54,2.503,1.598,5.416,2.25,8.935
-videoflow_bof,kitti,1.847,6.512,3.139,10.45,6.329,19.59,6.73,21.226
 videoflow_mof,sintel,0.432,1.845,0.621,2.964,1.839,6.695,2.893,10.771
 videoflow_mof,things,1.057,3.383,2.474,6.973,7.524,25.455,23.696,45.155
 videoflow_mof,kitti,2.621,8.23,3.71,12.518,4.493,12.347,7.27,16.91

ptlflow/__init__.py

@@ -63,6 +63,7 @@
 from ptlflow.models.maskflownet.maskflownet import MaskFlownet, MaskFlownet_S
 from ptlflow.models.matchflow.matchflow import MatchFlow, MatchFlowRAFT
 from ptlflow.models.ms_raft_plus.ms_raft_plus import MSRAFTPlus
+from ptlflow.models.neuflow.neuflow import NeuFlow
 from ptlflow.models.pwcnet.pwcnet import PWCNet, PWCDCNet
 from ptlflow.models.raft.raft import RAFT, RAFTSmall
 from ptlflow.models.rapidflow.rapidflow import (
@@ -144,6 +145,7 @@
     "matchflow": MatchFlow,
     "matchflow_raft": MatchFlowRAFT,
     "ms_raft+": MSRAFTPlus,
+    "neuflow": NeuFlow,
     "pwcnet": PWCDCNet,
     "pwcnet_nodc": PWCNet,
     "raft": RAFT,