├── .gitignore ├── LICENSE ├── README.md ├── __init__.py ├── colab_req.txt ├── convert.py ├── images └── demo │ ├── DNS_turbulence_LFN_quiver.png │ ├── DNS_turbulence_flow.flo │ ├── DNS_turbulence_gt_quiver.png │ ├── DNS_turbulence_img1.tif │ ├── DNS_turbulence_img2.tif │ ├── DNS_turbulence_out.flo │ ├── demo01.png │ └── demo02.png ├── inference.py ├── models └── pretrain_torch │ ├── Hui-LiteFlowNet.paramOnly │ ├── PIV-LiteFlowNet-en.paramOnly │ ├── network-default.pytorch │ ├── network-kitti.pytorch │ └── network-sintel.pytorch ├── requirements.txt ├── run.bash ├── run.py ├── src ├── __init__.py ├── correlation.py ├── datasets.py ├── flow_transforms.py ├── johswald │ ├── colorTest.py │ ├── computeColor.py │ └── readFlowFile.py ├── loss.py ├── models.py ├── particle_image_generator.py ├── postpro.py ├── utils.py ├── utils_color.py ├── utils_data.py ├── utils_plot.py └── write_data.py ├── stereo ├── __init__.py ├── dewarp.py ├── matching.py └── vel3d.py ├── stereo_cal.py ├── stereo_run.py └── trainer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | # blank 2 | -------------------------------------------------------------------------------- /colab_req.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/colab_req.txt -------------------------------------------------------------------------------- /convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/convert.py -------------------------------------------------------------------------------- /images/demo/DNS_turbulence_LFN_quiver.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/images/demo/DNS_turbulence_LFN_quiver.png -------------------------------------------------------------------------------- /images/demo/DNS_turbulence_flow.flo: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/images/demo/DNS_turbulence_flow.flo -------------------------------------------------------------------------------- /images/demo/DNS_turbulence_gt_quiver.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/images/demo/DNS_turbulence_gt_quiver.png -------------------------------------------------------------------------------- /images/demo/DNS_turbulence_img1.tif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/images/demo/DNS_turbulence_img1.tif -------------------------------------------------------------------------------- /images/demo/DNS_turbulence_img2.tif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/images/demo/DNS_turbulence_img2.tif -------------------------------------------------------------------------------- /images/demo/DNS_turbulence_out.flo: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/images/demo/DNS_turbulence_out.flo -------------------------------------------------------------------------------- /images/demo/demo01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/images/demo/demo01.png -------------------------------------------------------------------------------- /images/demo/demo02.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/images/demo/demo02.png -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/inference.py -------------------------------------------------------------------------------- /models/pretrain_torch/Hui-LiteFlowNet.paramOnly: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/models/pretrain_torch/Hui-LiteFlowNet.paramOnly -------------------------------------------------------------------------------- /models/pretrain_torch/PIV-LiteFlowNet-en.paramOnly: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/models/pretrain_torch/PIV-LiteFlowNet-en.paramOnly -------------------------------------------------------------------------------- /models/pretrain_torch/network-default.pytorch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/models/pretrain_torch/network-default.pytorch -------------------------------------------------------------------------------- /models/pretrain_torch/network-kitti.pytorch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/models/pretrain_torch/network-kitti.pytorch -------------------------------------------------------------------------------- /models/pretrain_torch/network-sintel.pytorch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/models/pretrain_torch/network-sintel.pytorch -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.bash: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/run.bash -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/run.py -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/src/__init__.py -------------------------------------------------------------------------------- /src/correlation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/src/correlation.py -------------------------------------------------------------------------------- /src/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/src/datasets.py -------------------------------------------------------------------------------- /src/flow_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/src/flow_transforms.py -------------------------------------------------------------------------------- /src/johswald/colorTest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/src/johswald/colorTest.py -------------------------------------------------------------------------------- /src/johswald/computeColor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/src/johswald/computeColor.py -------------------------------------------------------------------------------- /src/johswald/readFlowFile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/src/johswald/readFlowFile.py -------------------------------------------------------------------------------- /src/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/src/loss.py -------------------------------------------------------------------------------- /src/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/src/models.py -------------------------------------------------------------------------------- /src/particle_image_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/src/particle_image_generator.py -------------------------------------------------------------------------------- /src/postpro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/src/postpro.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/src/utils.py -------------------------------------------------------------------------------- /src/utils_color.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/src/utils_color.py -------------------------------------------------------------------------------- /src/utils_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/src/utils_data.py -------------------------------------------------------------------------------- /src/utils_plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/src/utils_plot.py -------------------------------------------------------------------------------- /src/write_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/src/write_data.py -------------------------------------------------------------------------------- /stereo/__init__.py: -------------------------------------------------------------------------------- 1 | # blank 2 | -------------------------------------------------------------------------------- /stereo/dewarp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/stereo/dewarp.py -------------------------------------------------------------------------------- /stereo/matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/stereo/matching.py -------------------------------------------------------------------------------- /stereo/vel3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/stereo/vel3d.py -------------------------------------------------------------------------------- /stereo_cal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/stereo_cal.py -------------------------------------------------------------------------------- /stereo_run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/stereo_run.py -------------------------------------------------------------------------------- /trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/abrosua/piv_liteflownet-pytorch/HEAD/trainer.py --------------------------------------------------------------------------------