├── .gitignore ├── HandCraftedModules.py ├── HardNet++.pth ├── HardNet.py ├── HardTFeat.pth ├── LAF.py ├── LICENSE ├── Losses.py ├── OnePassSIR.py ├── README.md ├── ReprojectionStuff.py ├── SparseImgRepresenter.py ├── Utils.py ├── architectures.py ├── augmentation.py ├── convertJIT ├── AffNetJIT.pt ├── OriNetJIT.pt └── convert_OriNet_and_AffNet_to_JIT.ipynb ├── dataset.py ├── examples ├── SIFT-AffNet-HardNet-kornia-matching.ipynb ├── direct_shape_optimization │ ├── Align_ideal_int_joint_Figure2_top.ipynb │ ├── Align_ideal_sep_Figure2_bottom.ipynb │ ├── Align_non_ideal_separate_Figure3.ipynb │ ├── HandCraftedModules.py │ ├── HardNet.py │ ├── LAF.py │ ├── Losses.py │ ├── ReprojectionStuff.py │ ├── SparseImgRepresenter.py │ ├── Utils.py │ ├── all_results_ideal.pickle │ ├── all_results_ideal_joint2.pickle │ ├── all_results_new_ideal_sep.pickle │ ├── architectures.py │ ├── figs_desc_comparison │ │ ├── non_ideal_all_MSE_survived_HardNegC.eps │ │ ├── non_ideal_all_MSE_survived_HardNet.eps │ │ ├── non_ideal_all_MSE_survived_PosDist.eps │ │ ├── non_ideal_loss_value_HardNegC.eps │ │ ├── non_ideal_loss_value_HardNet.eps │ │ └── non_ideal_loss_value_PosDist.eps │ ├── figs_loss_comparison │ │ ├── totalHardNet_geoms_ideal_joint_snn1.eps │ │ ├── totalHardNet_geoms_ideal_separate_snn1.eps │ │ ├── totalHardNet_geoms_non_ideal_separate_snn1.eps │ │ ├── totalHardNet_matches_ideal_joint_snn1.eps │ │ ├── totalHardNet_matches_ideal_separate_snn1.eps │ │ ├── totalHardNet_matches_non_ideal_separate_snn1.eps │ │ ├── totalPixels_geoms_ideal_joint_snn1.eps │ │ ├── totalPixels_geoms_ideal_separate_snn1.eps │ │ ├── totalPixels_geoms_non_ideal_separate_snn1.eps │ │ ├── totalPixels_matches_ideal_joint_snn1.eps │ │ ├── totalPixels_matches_ideal_separate_snn1.eps │ │ ├── totalPixels_matches_non_ideal_separate_snn1.eps │ │ ├── totalSIFT_geoms_ideal_joint_snn1.eps │ │ ├── totalSIFT_geoms_ideal_separate_snn1.eps │ │ ├── totalSIFT_geoms_non_ideal_separate_snn1.eps │ │ ├── totalSIFT_matches_ideal_joint_snn1.eps │ │ ├── totalSIFT_matches_ideal_separate_snn1.eps │ │ ├── totalSIFT_matches_non_ideal_separate_snn1.eps │ │ ├── totalTFeat_geoms_ideal_joint_snn1.eps │ │ ├── totalTFeat_geoms_ideal_separate_snn1.eps │ │ ├── totalTFeat_geoms_non_ideal_separate_snn1.eps │ │ ├── totalTFeat_matches_ideal_joint_snn1.eps │ │ ├── totalTFeat_matches_ideal_separate_snn1.eps │ │ └── totalTFeat_matches_non_ideal_separate_snn1.eps │ ├── hesaffBaum.py │ ├── non_ideal_separate.pickle │ ├── optimization_script.py │ └── pytorch_sift.py ├── hesaffnet │ ├── HandCraftedModules.py │ ├── HardNet.py │ ├── LAF.py │ ├── Losses.py │ ├── NMS.py │ ├── OnePassSIR.py │ ├── ReprojectionStuff.py │ ├── SparseImgRepresenter.py │ ├── Utils.py │ ├── WBS demo.ipynb │ ├── architectures.py │ ├── extract_geomOriTh.py │ ├── extract_geom_and_desc_upisup.py │ ├── extract_geom_and_desc_upisupTh.py │ ├── hesaffBaum.py │ ├── hesaffnet.py │ └── img │ │ ├── cat.png │ │ ├── fox1.png │ │ ├── fox1to2.txt │ │ ├── fox2.png │ │ ├── kpi1.png │ │ └── kpi2.png ├── just_shape │ ├── LAF.py │ ├── Utils.py │ ├── architectures.py │ ├── detect_affine_shape.py │ └── img │ │ └── face.png └── toy_example_figure1 │ ├── Figure1-video.ipynb │ ├── Figure1.ipynb │ ├── Losses.py │ ├── Utils.py │ └── points_from_paper.npy ├── gen_ds.py ├── imgs ├── graf16HesAff.jpg └── graf16HesAffNet.jpg ├── pretrained ├── AffNet.pth ├── AffNetFast.caffemodel ├── AffNetFast.prototxt └── OriNet.pth ├── pytorch_sift.py ├── run_me.sh ├── test-graf ├── H1to2p ├── H1to3p ├── H1to4p ├── H1to5p ├── H1to6p ├── img1.png ├── img2.png ├── img3.png ├── img4.png ├── img5.png └── img6.png ├── train_AffNet_test_on_graffity.py └── train_OriNet_test_on_graffity.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/.gitignore -------------------------------------------------------------------------------- /HandCraftedModules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/HandCraftedModules.py -------------------------------------------------------------------------------- /HardNet++.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/HardNet++.pth -------------------------------------------------------------------------------- /HardNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/HardNet.py -------------------------------------------------------------------------------- /HardTFeat.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/HardTFeat.pth -------------------------------------------------------------------------------- /LAF.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/LAF.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/LICENSE -------------------------------------------------------------------------------- /Losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/Losses.py -------------------------------------------------------------------------------- /OnePassSIR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/OnePassSIR.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/README.md -------------------------------------------------------------------------------- /ReprojectionStuff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/ReprojectionStuff.py -------------------------------------------------------------------------------- /SparseImgRepresenter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/SparseImgRepresenter.py -------------------------------------------------------------------------------- /Utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/Utils.py -------------------------------------------------------------------------------- /architectures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/architectures.py -------------------------------------------------------------------------------- /augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/augmentation.py -------------------------------------------------------------------------------- /convertJIT/AffNetJIT.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/convertJIT/AffNetJIT.pt -------------------------------------------------------------------------------- /convertJIT/OriNetJIT.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/convertJIT/OriNetJIT.pt -------------------------------------------------------------------------------- /convertJIT/convert_OriNet_and_AffNet_to_JIT.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/convertJIT/convert_OriNet_and_AffNet_to_JIT.ipynb -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/dataset.py -------------------------------------------------------------------------------- /examples/SIFT-AffNet-HardNet-kornia-matching.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/SIFT-AffNet-HardNet-kornia-matching.ipynb -------------------------------------------------------------------------------- /examples/direct_shape_optimization/Align_ideal_int_joint_Figure2_top.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/Align_ideal_int_joint_Figure2_top.ipynb -------------------------------------------------------------------------------- /examples/direct_shape_optimization/Align_ideal_sep_Figure2_bottom.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/Align_ideal_sep_Figure2_bottom.ipynb -------------------------------------------------------------------------------- /examples/direct_shape_optimization/Align_non_ideal_separate_Figure3.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/Align_non_ideal_separate_Figure3.ipynb -------------------------------------------------------------------------------- /examples/direct_shape_optimization/HandCraftedModules.py: -------------------------------------------------------------------------------- 1 | ../../HandCraftedModules.py -------------------------------------------------------------------------------- /examples/direct_shape_optimization/HardNet.py: -------------------------------------------------------------------------------- 1 | ../../HardNet.py -------------------------------------------------------------------------------- /examples/direct_shape_optimization/LAF.py: -------------------------------------------------------------------------------- 1 | ../../LAF.py -------------------------------------------------------------------------------- /examples/direct_shape_optimization/Losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/Losses.py -------------------------------------------------------------------------------- /examples/direct_shape_optimization/ReprojectionStuff.py: -------------------------------------------------------------------------------- 1 | ../../ReprojectionStuff.py -------------------------------------------------------------------------------- /examples/direct_shape_optimization/SparseImgRepresenter.py: -------------------------------------------------------------------------------- 1 | ../../SparseImgRepresenter.py -------------------------------------------------------------------------------- /examples/direct_shape_optimization/Utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/Utils.py -------------------------------------------------------------------------------- /examples/direct_shape_optimization/all_results_ideal.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/all_results_ideal.pickle -------------------------------------------------------------------------------- /examples/direct_shape_optimization/all_results_ideal_joint2.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/all_results_ideal_joint2.pickle -------------------------------------------------------------------------------- /examples/direct_shape_optimization/all_results_new_ideal_sep.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/all_results_new_ideal_sep.pickle -------------------------------------------------------------------------------- /examples/direct_shape_optimization/architectures.py: -------------------------------------------------------------------------------- 1 | ../../architectures.py -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_desc_comparison/non_ideal_all_MSE_survived_HardNegC.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_desc_comparison/non_ideal_all_MSE_survived_HardNegC.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_desc_comparison/non_ideal_all_MSE_survived_HardNet.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_desc_comparison/non_ideal_all_MSE_survived_HardNet.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_desc_comparison/non_ideal_all_MSE_survived_PosDist.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_desc_comparison/non_ideal_all_MSE_survived_PosDist.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_desc_comparison/non_ideal_loss_value_HardNegC.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_desc_comparison/non_ideal_loss_value_HardNegC.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_desc_comparison/non_ideal_loss_value_HardNet.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_desc_comparison/non_ideal_loss_value_HardNet.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_desc_comparison/non_ideal_loss_value_PosDist.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_desc_comparison/non_ideal_loss_value_PosDist.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_loss_comparison/totalHardNet_geoms_ideal_joint_snn1.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_loss_comparison/totalHardNet_geoms_ideal_joint_snn1.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_loss_comparison/totalHardNet_geoms_ideal_separate_snn1.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_loss_comparison/totalHardNet_geoms_ideal_separate_snn1.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_loss_comparison/totalHardNet_geoms_non_ideal_separate_snn1.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_loss_comparison/totalHardNet_geoms_non_ideal_separate_snn1.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_loss_comparison/totalHardNet_matches_ideal_joint_snn1.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_loss_comparison/totalHardNet_matches_ideal_joint_snn1.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_loss_comparison/totalHardNet_matches_ideal_separate_snn1.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_loss_comparison/totalHardNet_matches_ideal_separate_snn1.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_loss_comparison/totalHardNet_matches_non_ideal_separate_snn1.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_loss_comparison/totalHardNet_matches_non_ideal_separate_snn1.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_loss_comparison/totalPixels_geoms_ideal_joint_snn1.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_loss_comparison/totalPixels_geoms_ideal_joint_snn1.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_loss_comparison/totalPixels_geoms_ideal_separate_snn1.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_loss_comparison/totalPixels_geoms_ideal_separate_snn1.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_loss_comparison/totalPixels_geoms_non_ideal_separate_snn1.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_loss_comparison/totalPixels_geoms_non_ideal_separate_snn1.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_loss_comparison/totalPixels_matches_ideal_joint_snn1.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_loss_comparison/totalPixels_matches_ideal_joint_snn1.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_loss_comparison/totalPixels_matches_ideal_separate_snn1.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_loss_comparison/totalPixels_matches_ideal_separate_snn1.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_loss_comparison/totalPixels_matches_non_ideal_separate_snn1.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_loss_comparison/totalPixels_matches_non_ideal_separate_snn1.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_loss_comparison/totalSIFT_geoms_ideal_joint_snn1.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_loss_comparison/totalSIFT_geoms_ideal_joint_snn1.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_loss_comparison/totalSIFT_geoms_ideal_separate_snn1.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_loss_comparison/totalSIFT_geoms_ideal_separate_snn1.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_loss_comparison/totalSIFT_geoms_non_ideal_separate_snn1.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_loss_comparison/totalSIFT_geoms_non_ideal_separate_snn1.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_loss_comparison/totalSIFT_matches_ideal_joint_snn1.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_loss_comparison/totalSIFT_matches_ideal_joint_snn1.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_loss_comparison/totalSIFT_matches_ideal_separate_snn1.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_loss_comparison/totalSIFT_matches_ideal_separate_snn1.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_loss_comparison/totalSIFT_matches_non_ideal_separate_snn1.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_loss_comparison/totalSIFT_matches_non_ideal_separate_snn1.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_loss_comparison/totalTFeat_geoms_ideal_joint_snn1.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_loss_comparison/totalTFeat_geoms_ideal_joint_snn1.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_loss_comparison/totalTFeat_geoms_ideal_separate_snn1.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_loss_comparison/totalTFeat_geoms_ideal_separate_snn1.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_loss_comparison/totalTFeat_geoms_non_ideal_separate_snn1.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_loss_comparison/totalTFeat_geoms_non_ideal_separate_snn1.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_loss_comparison/totalTFeat_matches_ideal_joint_snn1.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_loss_comparison/totalTFeat_matches_ideal_joint_snn1.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_loss_comparison/totalTFeat_matches_ideal_separate_snn1.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_loss_comparison/totalTFeat_matches_ideal_separate_snn1.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/figs_loss_comparison/totalTFeat_matches_non_ideal_separate_snn1.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/figs_loss_comparison/totalTFeat_matches_non_ideal_separate_snn1.eps -------------------------------------------------------------------------------- /examples/direct_shape_optimization/hesaffBaum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/hesaffBaum.py -------------------------------------------------------------------------------- /examples/direct_shape_optimization/non_ideal_separate.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/non_ideal_separate.pickle -------------------------------------------------------------------------------- /examples/direct_shape_optimization/optimization_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/optimization_script.py -------------------------------------------------------------------------------- /examples/direct_shape_optimization/pytorch_sift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/direct_shape_optimization/pytorch_sift.py -------------------------------------------------------------------------------- /examples/hesaffnet/HandCraftedModules.py: -------------------------------------------------------------------------------- 1 | ../../HandCraftedModules.py -------------------------------------------------------------------------------- /examples/hesaffnet/HardNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/hesaffnet/HardNet.py -------------------------------------------------------------------------------- /examples/hesaffnet/LAF.py: -------------------------------------------------------------------------------- 1 | ../../LAF.py -------------------------------------------------------------------------------- /examples/hesaffnet/Losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/hesaffnet/Losses.py -------------------------------------------------------------------------------- /examples/hesaffnet/NMS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/hesaffnet/NMS.py -------------------------------------------------------------------------------- /examples/hesaffnet/OnePassSIR.py: -------------------------------------------------------------------------------- 1 | ../../OnePassSIR.py -------------------------------------------------------------------------------- /examples/hesaffnet/ReprojectionStuff.py: -------------------------------------------------------------------------------- 1 | ../../ReprojectionStuff.py -------------------------------------------------------------------------------- /examples/hesaffnet/SparseImgRepresenter.py: -------------------------------------------------------------------------------- 1 | ../../SparseImgRepresenter.py -------------------------------------------------------------------------------- /examples/hesaffnet/Utils.py: -------------------------------------------------------------------------------- 1 | ../../Utils.py -------------------------------------------------------------------------------- /examples/hesaffnet/WBS demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/hesaffnet/WBS demo.ipynb -------------------------------------------------------------------------------- /examples/hesaffnet/architectures.py: -------------------------------------------------------------------------------- 1 | ../../architectures.py -------------------------------------------------------------------------------- /examples/hesaffnet/extract_geomOriTh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/hesaffnet/extract_geomOriTh.py -------------------------------------------------------------------------------- /examples/hesaffnet/extract_geom_and_desc_upisup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/hesaffnet/extract_geom_and_desc_upisup.py -------------------------------------------------------------------------------- /examples/hesaffnet/extract_geom_and_desc_upisupTh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/hesaffnet/extract_geom_and_desc_upisupTh.py -------------------------------------------------------------------------------- /examples/hesaffnet/hesaffBaum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/hesaffnet/hesaffBaum.py -------------------------------------------------------------------------------- /examples/hesaffnet/hesaffnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/hesaffnet/hesaffnet.py -------------------------------------------------------------------------------- /examples/hesaffnet/img/cat.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/hesaffnet/img/cat.png -------------------------------------------------------------------------------- /examples/hesaffnet/img/fox1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/hesaffnet/img/fox1.png -------------------------------------------------------------------------------- /examples/hesaffnet/img/fox1to2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/hesaffnet/img/fox1to2.txt -------------------------------------------------------------------------------- /examples/hesaffnet/img/fox2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/hesaffnet/img/fox2.png -------------------------------------------------------------------------------- /examples/hesaffnet/img/kpi1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/hesaffnet/img/kpi1.png -------------------------------------------------------------------------------- /examples/hesaffnet/img/kpi2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/hesaffnet/img/kpi2.png -------------------------------------------------------------------------------- /examples/just_shape/LAF.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/just_shape/LAF.py -------------------------------------------------------------------------------- /examples/just_shape/Utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/just_shape/Utils.py -------------------------------------------------------------------------------- /examples/just_shape/architectures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/just_shape/architectures.py -------------------------------------------------------------------------------- /examples/just_shape/detect_affine_shape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/just_shape/detect_affine_shape.py -------------------------------------------------------------------------------- /examples/just_shape/img/face.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/just_shape/img/face.png -------------------------------------------------------------------------------- /examples/toy_example_figure1/Figure1-video.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/toy_example_figure1/Figure1-video.ipynb -------------------------------------------------------------------------------- /examples/toy_example_figure1/Figure1.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/toy_example_figure1/Figure1.ipynb -------------------------------------------------------------------------------- /examples/toy_example_figure1/Losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/toy_example_figure1/Losses.py -------------------------------------------------------------------------------- /examples/toy_example_figure1/Utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/toy_example_figure1/Utils.py -------------------------------------------------------------------------------- /examples/toy_example_figure1/points_from_paper.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/examples/toy_example_figure1/points_from_paper.npy -------------------------------------------------------------------------------- /gen_ds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/gen_ds.py -------------------------------------------------------------------------------- /imgs/graf16HesAff.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/imgs/graf16HesAff.jpg -------------------------------------------------------------------------------- /imgs/graf16HesAffNet.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/imgs/graf16HesAffNet.jpg -------------------------------------------------------------------------------- /pretrained/AffNet.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/pretrained/AffNet.pth -------------------------------------------------------------------------------- /pretrained/AffNetFast.caffemodel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/pretrained/AffNetFast.caffemodel -------------------------------------------------------------------------------- /pretrained/AffNetFast.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/pretrained/AffNetFast.prototxt -------------------------------------------------------------------------------- /pretrained/OriNet.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/pretrained/OriNet.pth -------------------------------------------------------------------------------- /pytorch_sift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/pytorch_sift.py -------------------------------------------------------------------------------- /run_me.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/run_me.sh -------------------------------------------------------------------------------- /test-graf/H1to2p: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/test-graf/H1to2p -------------------------------------------------------------------------------- /test-graf/H1to3p: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/test-graf/H1to3p -------------------------------------------------------------------------------- /test-graf/H1to4p: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/test-graf/H1to4p -------------------------------------------------------------------------------- /test-graf/H1to5p: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/test-graf/H1to5p -------------------------------------------------------------------------------- /test-graf/H1to6p: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/test-graf/H1to6p -------------------------------------------------------------------------------- /test-graf/img1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/test-graf/img1.png -------------------------------------------------------------------------------- /test-graf/img2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/test-graf/img2.png -------------------------------------------------------------------------------- /test-graf/img3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/test-graf/img3.png -------------------------------------------------------------------------------- /test-graf/img4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/test-graf/img4.png -------------------------------------------------------------------------------- /test-graf/img5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/test-graf/img5.png -------------------------------------------------------------------------------- /test-graf/img6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/test-graf/img6.png -------------------------------------------------------------------------------- /train_AffNet_test_on_graffity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/train_AffNet_test_on_graffity.py -------------------------------------------------------------------------------- /train_OriNet_test_on_graffity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ducha-aiki/affnet/HEAD/train_OriNet_test_on_graffity.py --------------------------------------------------------------------------------