├── .gitignore ├── LICENCE ├── README.md ├── configs └── EndoSTTN_dataset.json ├── core ├── .dataset.py.swp ├── dataset.py ├── dist.py ├── loss.py ├── spectral_norm.py ├── trainer.py └── utils.py ├── dataset_prep ├── README.md ├── SpecularDetectionMeslouhi2011.m ├── createSpecularMasks288.m ├── folder_dicts.py ├── rename.py ├── resizeFrames288.py ├── video2frames.py └── zipper.sh ├── datasets └── .gitkeep ├── docs ├── Flowchart.png ├── FlowchartArchi.png ├── Motivation2023.png ├── Table1Python385.png └── Table2Python385.png ├── model ├── sttn.py └── vis.py ├── quantifyResults.ipynb ├── requirements.txt ├── requirements36.txt ├── test-singleframe.py ├── test.py ├── train.py └── visualization.ipynb /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENCE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/LICENCE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/README.md -------------------------------------------------------------------------------- /configs/EndoSTTN_dataset.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/configs/EndoSTTN_dataset.json -------------------------------------------------------------------------------- /core/.dataset.py.swp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/core/.dataset.py.swp -------------------------------------------------------------------------------- /core/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/core/dataset.py -------------------------------------------------------------------------------- /core/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/core/dist.py -------------------------------------------------------------------------------- /core/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/core/loss.py -------------------------------------------------------------------------------- /core/spectral_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/core/spectral_norm.py -------------------------------------------------------------------------------- /core/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/core/trainer.py -------------------------------------------------------------------------------- /core/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/core/utils.py -------------------------------------------------------------------------------- /dataset_prep/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/dataset_prep/README.md -------------------------------------------------------------------------------- /dataset_prep/SpecularDetectionMeslouhi2011.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/dataset_prep/SpecularDetectionMeslouhi2011.m -------------------------------------------------------------------------------- /dataset_prep/createSpecularMasks288.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/dataset_prep/createSpecularMasks288.m -------------------------------------------------------------------------------- /dataset_prep/folder_dicts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/dataset_prep/folder_dicts.py -------------------------------------------------------------------------------- /dataset_prep/rename.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/dataset_prep/rename.py -------------------------------------------------------------------------------- /dataset_prep/resizeFrames288.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/dataset_prep/resizeFrames288.py -------------------------------------------------------------------------------- /dataset_prep/video2frames.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/dataset_prep/video2frames.py -------------------------------------------------------------------------------- /dataset_prep/zipper.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/dataset_prep/zipper.sh -------------------------------------------------------------------------------- /datasets/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /docs/Flowchart.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/docs/Flowchart.png -------------------------------------------------------------------------------- /docs/FlowchartArchi.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/docs/FlowchartArchi.png -------------------------------------------------------------------------------- /docs/Motivation2023.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/docs/Motivation2023.png -------------------------------------------------------------------------------- /docs/Table1Python385.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/docs/Table1Python385.png -------------------------------------------------------------------------------- /docs/Table2Python385.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/docs/Table2Python385.png -------------------------------------------------------------------------------- /model/sttn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/model/sttn.py -------------------------------------------------------------------------------- /model/vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/model/vis.py -------------------------------------------------------------------------------- /quantifyResults.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/quantifyResults.ipynb -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/requirements.txt -------------------------------------------------------------------------------- /requirements36.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/requirements36.txt -------------------------------------------------------------------------------- /test-singleframe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/test-singleframe.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/train.py -------------------------------------------------------------------------------- /visualization.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endomapper/Endo-STTN/HEAD/visualization.ipynb --------------------------------------------------------------------------------