.gitignore
FAQ.md
LICENSE.txt
METADATA.in
README.md
pyproject.toml
DISK/__init__.py
DISK/_version.py
DISK/create_dataset.py
DISK/create_proba_missing_files.py
DISK/embedding_umap.py
DISK/evaluate_fillmissing.py
DISK/impute_dataset.py
DISK/main_fillmissing.py
DISK/conf/__init__.py
DISK/conf/config_create_project.yaml
DISK/conf/config_evaluate.yaml
DISK/conf/config_impute.yaml
DISK/conf/config_prepare_data.yaml
DISK/conf/config_train.yaml
DISK/conf/network/gru.yaml
DISK/conf/network/st_gcn.yaml
DISK/conf/network/sts_gcn.yaml
DISK/conf/network/tcn.yaml
DISK/conf/network/transformer.yaml
DISK/launchers/__init__.py
DISK/launchers/add_skeleton.py
DISK/launchers/bla_config.py
DISK/launchers/check_gpu.py
DISK/launchers/check_install.py
DISK/launchers/create_project.py
DISK/launchers/evaluate_compare.py
DISK/launchers/impute.py
DISK/launchers/prepare_data.py
DISK/launchers/restore_default_config.py
DISK/launchers/train_evaluate.py
DISK/models/GRU_models.py
DISK/models/TCN.py
DISK/models/__init__.py
DISK/models/distribution_head.py
DISK/models/graph.py
DISK/models/st_gcn.py
DISK/models/stgcn_AE.py
DISK/models/sts_gcn.py
DISK/models/transformer.py
DISK/models/transformer_utils.py
DISK/utils/__init__.py
DISK/utils/config_decorator.py
DISK/utils/coordinates_utils.py
DISK/utils/dataset_utils.py
DISK/utils/logger_setup.py
DISK/utils/plot_missing_dataset.py
DISK/utils/train_fillmissing.py
DISK/utils/transformer_lr_scheduler.py
DISK/utils/transforms.py
DISK/utils/utils.py
DISK_impute.egg-info/PKG-INFO
DISK_impute.egg-info/SOURCES.txt
DISK_impute.egg-info/dependency_links.txt
DISK_impute.egg-info/entry_points.txt
DISK_impute.egg-info/requires.txt
DISK_impute.egg-info/top_level.txt
images/barplot_newmissing_compare_networks.png
images/heatmaps_FL2_pad_tests.png
images/imputation_method_summary_wskeleton.png
images/loss_dark_GRU_FL2_good.svg
images/loss_dark_GRU_Mice_bad.svg
images/loss_dark_transformer_NLL_FL2_good.svg
images/missing_pattern_8keypoints_INH1A_S10_M10_MC8_FL2_18_04_2019_proc_nij_8_08_19_C.png
images/reconstructions_whitebg-1.png
images/test_barplot_wo_RMSE.png
notebooks/images/csv_input_format.png
notebooks/images/dunn_fig2.webp
notebooks/images/fig1_human_mocap_presentation.png
notebooks/images/human_comparison_length_hole_all_vs_RMSE_repeat-0.png
notebooks/images/human_reconstruction_xyz_0_repeat-0.png
notebooks/images/install_google_colab.png
notebooks/images/rat7M_DISK_NLL_loss.png
notebooks/images/rat7M_comparison_length_hole_all_vs_RMSE_repeat-0.png
notebooks/images/rat7M_corrplot-model-RMSE-DISK_dataset_length60_stride60_repeat-0.png
notebooks/images/rat7M_count_vs_keypoint_set_keypoints.png
notebooks/images/rat7M_hist_length_original_vs_fake_set_keypoints.png
notebooks/images/rat7M_reconstruction_xyz_22_repeat-0.png
notebooks/images/rat7M_reconstruction_xyz_22_repeat-0_crop.png
notebooks/images/rat7M_thresholding_curve_RMSE_repeat-0.png
tests/test_tutorial_drive.sh
tests/test_tutorial_zenodo.sh
tests/test_windows.sh
tests/approval_tests/addmissing_approval_test.py
tests/cli_tests/test_create_project_cli.py
tests/cli_tests/test_evaluate_compare_cli.py
tests/cli_tests/test_impute_cli.py
tests/cli_tests/test_prepare_data_cli.py
tests/cli_tests/test_train_evaluate_cli.py
tests/pipeline_tests/shared_assertions.py
tests/pipeline_tests/test_create_dataset.py
tests/pipeline_tests/test_create_dataset.sh
tests/pipeline_tests/test_dlc_csv.py
tests/pipeline_tests/test_dlc_csv.sh
tests/pipeline_tests/test_dlc_h5.py
tests/pipeline_tests/test_dlc_h5.sh
tests/pipeline_tests/test_multianimal_dlc_csv.py
tests/pipeline_tests/test_multianimal_dlc_csv.sh
tests/pipeline_tests/test_multianimal_dlc_csv_sequential.py
tests/pipeline_tests/test_multianimal_dlc_csv_sequential.sh
tests/pipeline_tests/test_multianimal_dlc_h5.py
tests/pipeline_tests/test_multianimal_dlc_h5.sh
tests/pipeline_tests/test_multianimal_dlc_h5_sequential.sh
tests/pipeline_tests/test_npy_mocap.py
tests/pipeline_tests/test_npy_mocap.sh
tests/pipeline_tests/test_npy_mocap_sequential.sh
tests/pipeline_tests/test_npy_olivier_sequential.py
tests/pipeline_tests/test_npy_olivier_sequential.sh
tests/pipeline_tests/test_simple_csv.sh
tests/pipeline_tests/DISK_multianimal_dlc_csv/config_project.yaml
tests/pipeline_tests/DISK_multianimal_dlc_csv/DISK_data/dataset_60Hz_length30_stride30/config/config_prepare_data.yaml
tests/pipeline_tests/DISK_multianimal_dlc_csv/example_configs/config_create_project.yaml
tests/pipeline_tests/DISK_multianimal_dlc_csv/example_configs/config_evaluate.yaml
tests/pipeline_tests/DISK_multianimal_dlc_csv/example_configs/config_impute.yaml
tests/pipeline_tests/DISK_multianimal_dlc_csv/example_configs/config_prepare_data.yaml
tests/pipeline_tests/DISK_multianimal_dlc_csv/example_configs/config_train.yaml
tests/test_files/fish_fighting_interpolated_head.csv
tests/test_files/fish_fighting_interpolated_head_2D.csv
tests/test_files/freewalking_20220811_WTTB_fly1_1DLC_resnet50_FreeWalkingMp4Jul30shuffle1_190000.csv
tests/test_files/dlc_h5/21_11_8_one_mouse.top.irDLC_resnet50_moseq_exampleAug21shuffle1_500000.h5
tests/test_files/dlc_h5/21_12_2_def6a_1.top.irDLC_resnet50_moseq_exampleAug21shuffle1_500000.h5
tests/test_files/dlc_h5/22_04_26_cage4_0_2.top.irDLC_resnet50_moseq_exampleAug21shuffle1_500000.h5