LICENSE
README.md
pyproject.toml
src/flow_gym_suite.egg-info/PKG-INFO
src/flow_gym_suite.egg-info/SOURCES.txt
src/flow_gym_suite.egg-info/dependency_links.txt
src/flow_gym_suite.egg-info/requires.txt
src/flow_gym_suite.egg-info/top_level.txt
src/flowgym/__init__.py
src/flowgym/make.py
src/flowgym/types.py
src/flowgym/utils.py
src/flowgym/common/__init__.py
src/flowgym/common/evaluation.py
src/flowgym/common/filters.py
src/flowgym/common/median.py
src/flowgym/common/preprocess.py
src/flowgym/common/base/__init__.py
src/flowgym/common/base/estimator.py
src/flowgym/common/base/trainable_states.py
src/flowgym/density/__init__.py
src/flowgym/density/nn.py
src/flowgym/density/simple.py
src/flowgym/environment/__init__.py
src/flowgym/environment/fluid_env.py
src/flowgym/flow/__init__.py
src/flowgym/flow/base.py
src/flowgym/flow/deepflow.py
src/flowgym/flow/dummy.py
src/flowgym/flow/farneback.py
src/flowgym/flow/hornschunck.py
src/flowgym/flow/process.py
src/flowgym/flow/utils.py
src/flowgym/flow/consensus/__init__.py
src/flowgym/flow/consensus/admm.py
src/flowgym/flow/consensus/consensus.py
src/flowgym/flow/consensus/consensus_algorithms.py
src/flowgym/flow/consensus/objectives.py
src/flowgym/flow/consensus/regularizers.py
src/flowgym/flow/consensus/solvers.py
src/flowgym/flow/dis/__init__.py
src/flowgym/flow/dis/dis.py
src/flowgym/flow/dis/dis_jax.py
src/flowgym/flow/dis/process.py
src/flowgym/flow/open_piv/__init__.py
src/flowgym/flow/open_piv/openpiv.py
src/flowgym/flow/open_piv/openpiv_jax.py
src/flowgym/flow/open_piv/process.py
src/flowgym/flow/postprocess/__init__.py
src/flowgym/flow/postprocess/data_interpolation.py
src/flowgym/flow/postprocess/data_smoothing.py
src/flowgym/flow/postprocess/data_validation.py
src/flowgym/flow/raft/__init__.py
src/flowgym/flow/raft/process.py
src/flowgym/flow/raft/raft_jax.py
src/flowgym/flow/raft/raft_piv_pytorch.py
src/flowgym/nn/__init__.py
src/flowgym/nn/blocks.py
src/flowgym/nn/cnn.py
src/flowgym/nn/raft_model.py
src/flowgym/nn/raft_torch_nn/__init__.py
src/flowgym/nn/raft_torch_nn/corr_block.py
src/flowgym/nn/raft_torch_nn/flowNetsRAFT.py
src/flowgym/nn/raft_torch_nn/submodules_RAFT_GRU.py
src/flowgym/nn/raft_torch_nn/submodules_RAFT_extractor.py
src/flowgym/training/__init__.py
src/flowgym/training/optimizer.py
src/flowgym/training/schedules.py
src/flowgym/training/utils.py
tests/test_data_interpolation.py
tests/test_data_smoothing.py
tests/test_data_validation.py
tests/test_density_estimate_simple.py
tests/test_dis_jax.py
tests/test_estimation_stats.py
tests/test_filters.py
tests/test_flow_estimate_base.py
tests/test_flow_estimate_dis.py
tests/test_median.py
tests/test_openpiv_jax.py
tests/test_postprocess_config.py
tests/test_preprocess.py
tests/test_preprocess_config.py
tests/test_trainable_state.py
tests/test_training_utils.py