├── .gitignore ├── LICENSE ├── README.md ├── data ├── caltech_dataset.py ├── download_datasets.py ├── pascal_parts_dataset.py ├── pf_dataset.py ├── synth_dataset.py ├── tss_dataset.py └── weak_dataset.py ├── demo.py ├── demo_notebook.ipynb ├── eval.py ├── geotnf ├── flow.py ├── point_tnf.py └── transformation.py ├── image └── normalization.py ├── model ├── cnn_geometric_model.py └── loss.py ├── options └── options.py ├── scripts ├── train_strong_random_affine_pascal.sh ├── train_strong_random_tps_pascal.sh └── train_weak_pf_pascal.sh ├── train_strong.py ├── train_weak.py ├── trained_models └── .gitignore ├── training_data ├── pascal-random │ ├── test.csv │ └── train.csv └── pf-pascal-flip │ ├── test_pairs.csv │ ├── train_pairs.csv │ └── train_single_image.csv └── util ├── dataloader.py ├── eval_util.py ├── py_util.py ├── torch_util.py └── train_test_fn.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/README.md -------------------------------------------------------------------------------- /data/caltech_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/data/caltech_dataset.py -------------------------------------------------------------------------------- /data/download_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/data/download_datasets.py -------------------------------------------------------------------------------- /data/pascal_parts_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/data/pascal_parts_dataset.py -------------------------------------------------------------------------------- /data/pf_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/data/pf_dataset.py -------------------------------------------------------------------------------- /data/synth_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/data/synth_dataset.py -------------------------------------------------------------------------------- /data/tss_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/data/tss_dataset.py -------------------------------------------------------------------------------- /data/weak_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/data/weak_dataset.py -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/demo.py -------------------------------------------------------------------------------- /demo_notebook.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/demo_notebook.ipynb -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/eval.py -------------------------------------------------------------------------------- /geotnf/flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/geotnf/flow.py -------------------------------------------------------------------------------- /geotnf/point_tnf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/geotnf/point_tnf.py -------------------------------------------------------------------------------- /geotnf/transformation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/geotnf/transformation.py -------------------------------------------------------------------------------- /image/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/image/normalization.py -------------------------------------------------------------------------------- /model/cnn_geometric_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/model/cnn_geometric_model.py -------------------------------------------------------------------------------- /model/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/model/loss.py -------------------------------------------------------------------------------- /options/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/options/options.py -------------------------------------------------------------------------------- /scripts/train_strong_random_affine_pascal.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/scripts/train_strong_random_affine_pascal.sh -------------------------------------------------------------------------------- /scripts/train_strong_random_tps_pascal.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/scripts/train_strong_random_tps_pascal.sh -------------------------------------------------------------------------------- /scripts/train_weak_pf_pascal.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/scripts/train_weak_pf_pascal.sh -------------------------------------------------------------------------------- /train_strong.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/train_strong.py -------------------------------------------------------------------------------- /train_weak.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/train_weak.py -------------------------------------------------------------------------------- /trained_models/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/trained_models/.gitignore -------------------------------------------------------------------------------- /training_data/pascal-random/test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/training_data/pascal-random/test.csv -------------------------------------------------------------------------------- /training_data/pascal-random/train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/training_data/pascal-random/train.csv -------------------------------------------------------------------------------- /training_data/pf-pascal-flip/test_pairs.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/training_data/pf-pascal-flip/test_pairs.csv -------------------------------------------------------------------------------- /training_data/pf-pascal-flip/train_pairs.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/training_data/pf-pascal-flip/train_pairs.csv -------------------------------------------------------------------------------- /training_data/pf-pascal-flip/train_single_image.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/training_data/pf-pascal-flip/train_single_image.csv -------------------------------------------------------------------------------- /util/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/util/dataloader.py -------------------------------------------------------------------------------- /util/eval_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/util/eval_util.py -------------------------------------------------------------------------------- /util/py_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/util/py_util.py -------------------------------------------------------------------------------- /util/torch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/util/torch_util.py -------------------------------------------------------------------------------- /util/train_test_fn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ignacio-rocco/weakalign/HEAD/util/train_test_fn.py --------------------------------------------------------------------------------