├── .gitignore ├── LICENSE.DATA.txt ├── LICENSE.SOURCE.txt ├── README.md ├── __init__.py ├── assets └── teaser.gif ├── dataset ├── __init__.py ├── augmentation.py ├── kandao.py ├── kandao_test.py ├── kandao_train.py └── transform.py ├── download.sh ├── engines ├── __init__.py ├── metaclass.py ├── tester.py ├── testerDDP.py ├── trainer.py └── trainerDDP.py ├── env.sh ├── librarians ├── __init__.py ├── librarian.py ├── plot.py ├── record.py └── sampler.py ├── loss ├── __init__.py ├── control.py ├── modules.py ├── photometric.py ├── ssim.py └── vgg │ ├── __init__.py │ └── vgg16_inter.py ├── networks ├── __init__.py ├── base.py ├── homography │ ├── __init__.py │ ├── base.py │ ├── builder.py │ └── large.py ├── models │ ├── DoubleColorCorNet.py │ ├── HomoDispNet.py │ ├── PostColorCorNet.py │ ├── PreColorCorNet.py │ ├── __init__.py │ └── base.py └── unet │ ├── __init__.py │ ├── base.py │ ├── builder.py │ └── large.py ├── normalization ├── __init__.py ├── ground.py └── manipulator.py ├── options.py ├── scripts ├── test-L1.sh ├── test-final.sh ├── test-homography.sh ├── test-post.sh ├── test-pre.sh ├── test-spl.sh ├── train-L1.sh ├── train-final.sh ├── train-homography.sh ├── train-post.sh ├── train-pre.sh └── train-spl.sh ├── test.py ├── train.py └── utils ├── __init__.py ├── cuda.py ├── debug.py ├── image.py ├── inspection.py ├── masking.py └── time.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.DATA.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/LICENSE.DATA.txt -------------------------------------------------------------------------------- /LICENSE.SOURCE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/LICENSE.SOURCE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/__init__.py -------------------------------------------------------------------------------- /assets/teaser.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/assets/teaser.gif -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/dataset/__init__.py -------------------------------------------------------------------------------- /dataset/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/dataset/augmentation.py -------------------------------------------------------------------------------- /dataset/kandao.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/dataset/kandao.py -------------------------------------------------------------------------------- /dataset/kandao_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/dataset/kandao_test.py -------------------------------------------------------------------------------- /dataset/kandao_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/dataset/kandao_train.py -------------------------------------------------------------------------------- /dataset/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/dataset/transform.py -------------------------------------------------------------------------------- /download.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/download.sh -------------------------------------------------------------------------------- /engines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/engines/__init__.py -------------------------------------------------------------------------------- /engines/metaclass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/engines/metaclass.py -------------------------------------------------------------------------------- /engines/tester.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/engines/tester.py -------------------------------------------------------------------------------- /engines/testerDDP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/engines/testerDDP.py -------------------------------------------------------------------------------- /engines/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/engines/trainer.py -------------------------------------------------------------------------------- /engines/trainerDDP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/engines/trainerDDP.py -------------------------------------------------------------------------------- /env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/env.sh -------------------------------------------------------------------------------- /librarians/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/librarians/__init__.py -------------------------------------------------------------------------------- /librarians/librarian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/librarians/librarian.py -------------------------------------------------------------------------------- /librarians/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/librarians/plot.py -------------------------------------------------------------------------------- /librarians/record.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/librarians/record.py -------------------------------------------------------------------------------- /librarians/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/librarians/sampler.py -------------------------------------------------------------------------------- /loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/loss/__init__.py -------------------------------------------------------------------------------- /loss/control.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/loss/control.py -------------------------------------------------------------------------------- /loss/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/loss/modules.py -------------------------------------------------------------------------------- /loss/photometric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/loss/photometric.py -------------------------------------------------------------------------------- /loss/ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/loss/ssim.py -------------------------------------------------------------------------------- /loss/vgg/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /loss/vgg/vgg16_inter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/loss/vgg/vgg16_inter.py -------------------------------------------------------------------------------- /networks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/networks/__init__.py -------------------------------------------------------------------------------- /networks/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/networks/base.py -------------------------------------------------------------------------------- /networks/homography/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/networks/homography/__init__.py -------------------------------------------------------------------------------- /networks/homography/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/networks/homography/base.py -------------------------------------------------------------------------------- /networks/homography/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/networks/homography/builder.py -------------------------------------------------------------------------------- /networks/homography/large.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/networks/homography/large.py -------------------------------------------------------------------------------- /networks/models/DoubleColorCorNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/networks/models/DoubleColorCorNet.py -------------------------------------------------------------------------------- /networks/models/HomoDispNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/networks/models/HomoDispNet.py -------------------------------------------------------------------------------- /networks/models/PostColorCorNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/networks/models/PostColorCorNet.py -------------------------------------------------------------------------------- /networks/models/PreColorCorNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/networks/models/PreColorCorNet.py -------------------------------------------------------------------------------- /networks/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/networks/models/__init__.py -------------------------------------------------------------------------------- /networks/models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/networks/models/base.py -------------------------------------------------------------------------------- /networks/unet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/networks/unet/__init__.py -------------------------------------------------------------------------------- /networks/unet/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/networks/unet/base.py -------------------------------------------------------------------------------- /networks/unet/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/networks/unet/builder.py -------------------------------------------------------------------------------- /networks/unet/large.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/networks/unet/large.py -------------------------------------------------------------------------------- /normalization/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/normalization/__init__.py -------------------------------------------------------------------------------- /normalization/ground.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/normalization/ground.py -------------------------------------------------------------------------------- /normalization/manipulator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/normalization/manipulator.py -------------------------------------------------------------------------------- /options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/options.py -------------------------------------------------------------------------------- /scripts/test-L1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/scripts/test-L1.sh -------------------------------------------------------------------------------- /scripts/test-final.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/scripts/test-final.sh -------------------------------------------------------------------------------- /scripts/test-homography.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/scripts/test-homography.sh -------------------------------------------------------------------------------- /scripts/test-post.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/scripts/test-post.sh -------------------------------------------------------------------------------- /scripts/test-pre.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/scripts/test-pre.sh -------------------------------------------------------------------------------- /scripts/test-spl.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/scripts/test-spl.sh -------------------------------------------------------------------------------- /scripts/train-L1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/scripts/train-L1.sh -------------------------------------------------------------------------------- /scripts/train-final.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/scripts/train-final.sh -------------------------------------------------------------------------------- /scripts/train-homography.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/scripts/train-homography.sh -------------------------------------------------------------------------------- /scripts/train-post.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/scripts/train-post.sh -------------------------------------------------------------------------------- /scripts/train-pre.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/scripts/train-pre.sh -------------------------------------------------------------------------------- /scripts/train-spl.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/scripts/train-spl.sh -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/cuda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/utils/cuda.py -------------------------------------------------------------------------------- /utils/debug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/utils/debug.py -------------------------------------------------------------------------------- /utils/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/utils/image.py -------------------------------------------------------------------------------- /utils/inspection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/utils/inspection.py -------------------------------------------------------------------------------- /utils/masking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/utils/masking.py -------------------------------------------------------------------------------- /utils/time.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EadCat/WeaklySupervisedStitchingNetwork/HEAD/utils/time.py --------------------------------------------------------------------------------