Merge pull request #30 from hmorimitsu/fastflownet

Fastflownet

Author: hmorimitsu

README.md

@@ -24,6 +24,7 @@ This is still under development, so some things may not work as intended. I plan
 ## Available models
 
 - DICL-Flow [https://arxiv.org/abs/2010.14851](https://arxiv.org/abs/2010.14851)
+- FastFlowNet [https://arxiv.org/abs/2103.04524](https://arxiv.org/abs/2103.04524)
 - FlowNet - [https://arxiv.org/abs/1504.06852](https://arxiv.org/abs/1504.06852)
 - FlowNet2 - [https://arxiv.org/abs/1612.01925](https://arxiv.org/abs/1612.01925)
 - GMA - [https://arxiv.org/abs/2104.02409](https://arxiv.org/abs/2104.02409)

docs/source/_static/kitti_2012_epe_outlier-drop_kitti_sintel.html

@@ -3,6 +3,11 @@ dicl-chairs-fdc24e2f.ckpt,ckpt_chairs.pth.tar,https://github.com/jytime/DICL-Flo
 dicl-kitti-4813ccab.ckpt,ckpt_kitti.pth.tar,
 dicl-sintel-fa9fc259.ckpt,ckpt_sintel.pth.tar,
 dicl-things-20bc573d.ckpt,ckpt_things.pth.tar,
+fastflownet-chairs-89e7a48e.ckpt,fastflownet_chairs.pth,https://github.com/ltkong218/FastFlowNet
+fastflownet-kitti-6d3526a8.ckpt,fastflownet_ft_kitti.pth,
+fastflownet-mix-fd9b8c0d.ckpt,fastflownet_ft_mix.pth,
+fastflownet-sintel-6475ea96.ckpt,fastflownet_ft_sintel.pth,
+fastflownet-things3d-fc093d29.ckpt,fastflownet_things.pth,
 flownet2-things-d63b53a7.ckpt,FlowNet2_checkpoint.pth.tar,https://github.com/NVIDIA/flownet2-pytorch
 flownetc-things-cc8ac7fd.ckpt,FlowNet2-C_checkpoint,
 flownetcs-things-4bdecffa.ckpt,FlowNet2-CS_checkpoint,

docs/source/_static/kitti_2015_epe_outlier-drop_kitti_sintel.html

@@ -18,6 +18,17 @@ DICL-Flow
 
 - Model names: ``dicl``
 
+FastFlownet
+-----------
+
+`[source code] <https://github.com/hmorimitsu/ptlflow/tree/main/ptlflow/models/fastflownet>`__
+
+- Paper: **FastFlowNet: A Lightweight Network for Fast Optical Flow Estimation** - `https://arxiv.org/abs/2103.04524 <https://arxiv.org/abs/2103.04524>`_
+
+- Reference code: `https://github.com/ltkong218/FastFlowNet <https://github.com/ltkong218/FastFlowNet>`_
+
+- Model names: ``fastflownet``
+
 Flownet
 -------
 

docs/source/_static/sintel_clean_epe_outlier-drop_kitti_sintel.html

@@ -1,35 +1,34 @@
 Model,Params,Time(ms)
-dicl,11226036,98.89
-flownet2,162518834,134.466
-flownetc,39175298,50.588
-flownetcs,77870620,61.348
-flownetcss,116565942,78.742
-flownets,38676506,11.257
-flownetsd,45371666,18.575
-gma,5879873,172.245
-hd3,39561975,73.798
-hd3_ctxt,39942647,74.583
-irr_pwc,6362092,164.908
-irr_pwcnet,8639230,31.728
-irr_pwcnet_irr,3354146,35.898
-lcv_raft,5323328,133.956
-lcv_raft_small,1006674,45.429
-liteflownet,5379613,82.569
-liteflownet2,6429120,32.58
-liteflownet2_pseudoreg,6492907,40.102
-liteflownet3,7524188,55.053
-liteflownet3_pseudoreg,7587975,62.607
-liteflownet3s,8005810,56.656
-liteflownet3s_pseudoreg,8069597,64.378
-maskflownet,20655716,66.67
-maskflownet_s,10514256,37.186
-pwcnet,8243008,28.271
-pwcdcnet,9374274,33.085
-raft,5257536,133.857
-raft_small,990162,45.269
-scopeflow,6362092,174.195
-scv4,5236240,933.896
-scv8,5347264,341.227
-starflow,4772256,139.875
-vcn,10310781,149.529
-vcn_small,8370804,54.778
+dicl,11226036,81.499
+fastflownet,1366114,20.416
+flownet2,162518834,102.711
+flownetc,39175298,46.593
+flownetcs,77870620,58.934
+flownetcss,116565942,71.323
+flownets,38676506,10.748
+flownetsd,45371666,16.331
+gma,5879873,165.717
+hd3,39561975,66.659
+hd3_ctxt,39942647,67.767
+irr_pwc,6362092,176.973
+irr_pwcnet,8639230,37.118
+irr_pwcnet_irr,3354146,40.394
+lcv_raft,5323328,130.238
+lcv_raft_small,1006674,49.608
+liteflownet,5379613,71.395
+liteflownet2,6429120,33.116
+liteflownet2_pseudoreg,6492907,37.444
+liteflownet3,7524188,55.895
+liteflownet3_pseudoreg,7587975,60.264
+liteflownet3s,8005810,57.595
+liteflownet3s_pseudoreg,8069597,62.03
+maskflownet,20655716,76.932
+maskflownet_s,10514256,42.617
+pwcnet,8243008,33.002
+pwcdcnet,9374274,38.314
+raft,5257536,129.953
+raft_small,990162,49.286
+scopeflow,6362092,182.887
+starflow,4772256,148.879
+vcn,10310781,156.231
+vcn_small,8370804,60.097