.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/impute_dataset.py
DISK/main_fillmissing.py
DISK/test_fillmissing.py
DISK/conf/__init__.py
DISK/conf/conf_create_dataset.yaml
DISK/conf/conf_impute.yaml
DISK/conf/conf_missing.yaml
DISK/conf/conf_proba_missing_files.yaml
DISK/conf/conf_test.yaml
DISK/conf/.ipynb_checkpoints/conf_create_dataset-checkpoint.yaml
DISK/conf/.ipynb_checkpoints/conf_impute-checkpoint.yaml
DISK/conf/.ipynb_checkpoints/conf_test-checkpoint.yaml
DISK/conf/network/ae_tcn.yaml
DISK/conf/network/gru.yaml
DISK/conf/network/st_gcn.yaml
DISK/conf/network/sts_gcn.yaml
DISK/conf/network/transformer.yaml
DISK/launchers/DISK_launcher.py
DISK/launchers/__init__.py
DISK/launchers/check_gpu.py
DISK/launchers/check_install.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/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/DISK_tutorial_2024-05.ipynb
notebooks/conf_impute.yaml
notebooks/conf_missing.yaml
notebooks/conf_test.yaml
notebooks/tutorial_mocap.md
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
tests/import_test.py
tests/test_create_dataset.sh
tests/test_dlc_csv.sh
tests/test_dlc_h5.sh
tests/test_multianimal_dlc_csv.sh
tests/test_multianimal_dlc_csv_sequential.sh
tests/test_multianimal_dlc_h5.sh
tests/test_multianimal_dlc_h5_sequential.sh
tests/test_npy_mocap.sh
tests/test_npy_mocap_sequential.sh
tests/test_npy_olivier_sequential.sh
tests/test_simple_csv.sh
tests/test_tutorial_drive.sh
tests/test_tutorial_zenodo.sh
tests/test_windows.sh
tests/test_files/21_11_8_one_mouse.top.irDLC_resnet50_moseq_exampleAug21shuffle1_500000.h5
tests/test_files/21_12_2_def6a_1.top.irDLC_resnet50_moseq_exampleAug21shuffle1_500000.h5
tests/test_files/22_04_26_cage4_0_2.top.irDLC_resnet50_moseq_exampleAug21shuffle1_500000.h5
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