├── .gitignore ├── LICENSE ├── README.md ├── checkpoints.lua ├── criterion ├── EPECriterion.lua ├── TOMCriterionFlow.lua └── TOMCriterionUnsup.lua ├── dataloader.lua ├── datasets ├── TOMDataset.lua └── transforms.lua ├── eval ├── eval_real_data.py ├── eval_utils.lua ├── fastaniso.py ├── psnr.py ├── py_utils.py ├── run_model.lua └── run_synth_data.lua ├── images ├── Results_Colored_Objects │ ├── 8_colored_030_diff.jpg │ ├── 8_colored_030_fcolor.jpg │ ├── 8_colored_030_mask.png │ ├── 8_colored_030_pred.jpg │ ├── 8_colored_030_ref.jpg │ ├── 8_colored_030_rho.png │ └── 8_colored_030_tar.jpg ├── Results_Natural_Illumination │ ├── 23_023_IMG_4999crop_0_ref.jpg │ ├── 23_023_IMG_4999crop_1_fcolor_r.jpg │ ├── 23_023_IMG_4999crop_3_mask_r.jpg │ ├── 23_023_IMG_4999crop_4_rho_r.jpg │ ├── 23_023_IMG_4999crop_5_final_r.jpg │ ├── 23_023_IMG_4999crop_7_tar.jpg │ ├── 23_023_IMG_5052crop_0_ref.jpg │ ├── 23_023_IMG_5052crop_1_fcolor_r.jpg │ ├── 23_023_IMG_5052crop_3_mask_r.jpg │ ├── 23_023_IMG_5052crop_4_rho_r.jpg │ ├── 23_023_IMG_5052crop_5_final_r.jpg │ ├── 23_023_IMG_5052crop_7_tar.jpg │ ├── 29_029_IMG_5058crop_0_ref.jpg │ ├── 29_029_IMG_5058crop_1_fcolor_r.jpg │ ├── 29_029_IMG_5058crop_3_mask_r.jpg │ ├── 29_029_IMG_5058crop_4_rho_r.jpg │ ├── 29_029_IMG_5058crop_5_final_r.jpg │ ├── 29_029_IMG_5058crop_7_tar.jpg │ ├── 57_057_IMG_5084crop_0_ref.jpg │ ├── 57_057_IMG_5084crop_1_fcolor_r.jpg │ ├── 57_057_IMG_5084crop_3_mask_r.jpg │ ├── 57_057_IMG_5084crop_4_rho_r.jpg │ ├── 57_057_IMG_5084crop_5_final_r.jpg │ └── 57_057_IMG_5084crop_7_tar.jpg ├── TOM-Net_with_Trimap_Bg_Samples │ ├── dragon.jpg │ ├── dragon_bg.jpg │ ├── dragon_trimap.png │ ├── horse.jpg │ ├── horse_bg.jpg │ ├── horse_trimap.png │ ├── img_bg_trimap_list.txt │ ├── two_glass.jpg │ ├── two_glass_bg.jpg │ └── two_glass_trimap.png ├── bull.jpg └── cvpr2018_tom-net.jpg ├── main.lua ├── models ├── AffineGridGeneratorUSOF.lua ├── CoarseNet.lua ├── RefineNet.lua ├── init.lua └── model_utils.lua ├── opts.lua ├── refine ├── main_refine.lua └── opts_refine.lua ├── scripts ├── download_pretrained_model.sh ├── download_pretrained_models_IJCV.sh ├── download_testing_dataset.sh ├── download_training_dataset.sh └── download_validation_dataset.sh ├── train.lua └── utils ├── dict_utils.lua ├── flow_utils.lua ├── io_utils.lua └── str_utils.lua /.gitignore: -------------------------------------------------------------------------------- 1 | *.pyc 2 | *.t7 3 | tmp/ 4 | data 5 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/README.md -------------------------------------------------------------------------------- /checkpoints.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/checkpoints.lua -------------------------------------------------------------------------------- /criterion/EPECriterion.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/criterion/EPECriterion.lua -------------------------------------------------------------------------------- /criterion/TOMCriterionFlow.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/criterion/TOMCriterionFlow.lua -------------------------------------------------------------------------------- /criterion/TOMCriterionUnsup.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/criterion/TOMCriterionUnsup.lua -------------------------------------------------------------------------------- /dataloader.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/dataloader.lua -------------------------------------------------------------------------------- /datasets/TOMDataset.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/datasets/TOMDataset.lua -------------------------------------------------------------------------------- /datasets/transforms.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/datasets/transforms.lua -------------------------------------------------------------------------------- /eval/eval_real_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/eval/eval_real_data.py -------------------------------------------------------------------------------- /eval/eval_utils.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/eval/eval_utils.lua -------------------------------------------------------------------------------- /eval/fastaniso.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/eval/fastaniso.py -------------------------------------------------------------------------------- /eval/psnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/eval/psnr.py -------------------------------------------------------------------------------- /eval/py_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/eval/py_utils.py -------------------------------------------------------------------------------- /eval/run_model.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/eval/run_model.lua -------------------------------------------------------------------------------- /eval/run_synth_data.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/eval/run_synth_data.lua -------------------------------------------------------------------------------- /images/Results_Colored_Objects/8_colored_030_diff.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Colored_Objects/8_colored_030_diff.jpg -------------------------------------------------------------------------------- /images/Results_Colored_Objects/8_colored_030_fcolor.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Colored_Objects/8_colored_030_fcolor.jpg -------------------------------------------------------------------------------- /images/Results_Colored_Objects/8_colored_030_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Colored_Objects/8_colored_030_mask.png -------------------------------------------------------------------------------- /images/Results_Colored_Objects/8_colored_030_pred.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Colored_Objects/8_colored_030_pred.jpg -------------------------------------------------------------------------------- /images/Results_Colored_Objects/8_colored_030_ref.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Colored_Objects/8_colored_030_ref.jpg -------------------------------------------------------------------------------- /images/Results_Colored_Objects/8_colored_030_rho.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Colored_Objects/8_colored_030_rho.png -------------------------------------------------------------------------------- /images/Results_Colored_Objects/8_colored_030_tar.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Colored_Objects/8_colored_030_tar.jpg -------------------------------------------------------------------------------- /images/Results_Natural_Illumination/23_023_IMG_4999crop_0_ref.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Natural_Illumination/23_023_IMG_4999crop_0_ref.jpg -------------------------------------------------------------------------------- /images/Results_Natural_Illumination/23_023_IMG_4999crop_1_fcolor_r.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Natural_Illumination/23_023_IMG_4999crop_1_fcolor_r.jpg -------------------------------------------------------------------------------- /images/Results_Natural_Illumination/23_023_IMG_4999crop_3_mask_r.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Natural_Illumination/23_023_IMG_4999crop_3_mask_r.jpg -------------------------------------------------------------------------------- /images/Results_Natural_Illumination/23_023_IMG_4999crop_4_rho_r.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Natural_Illumination/23_023_IMG_4999crop_4_rho_r.jpg -------------------------------------------------------------------------------- /images/Results_Natural_Illumination/23_023_IMG_4999crop_5_final_r.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Natural_Illumination/23_023_IMG_4999crop_5_final_r.jpg -------------------------------------------------------------------------------- /images/Results_Natural_Illumination/23_023_IMG_4999crop_7_tar.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Natural_Illumination/23_023_IMG_4999crop_7_tar.jpg -------------------------------------------------------------------------------- /images/Results_Natural_Illumination/23_023_IMG_5052crop_0_ref.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Natural_Illumination/23_023_IMG_5052crop_0_ref.jpg -------------------------------------------------------------------------------- /images/Results_Natural_Illumination/23_023_IMG_5052crop_1_fcolor_r.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Natural_Illumination/23_023_IMG_5052crop_1_fcolor_r.jpg -------------------------------------------------------------------------------- /images/Results_Natural_Illumination/23_023_IMG_5052crop_3_mask_r.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Natural_Illumination/23_023_IMG_5052crop_3_mask_r.jpg -------------------------------------------------------------------------------- /images/Results_Natural_Illumination/23_023_IMG_5052crop_4_rho_r.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Natural_Illumination/23_023_IMG_5052crop_4_rho_r.jpg -------------------------------------------------------------------------------- /images/Results_Natural_Illumination/23_023_IMG_5052crop_5_final_r.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Natural_Illumination/23_023_IMG_5052crop_5_final_r.jpg -------------------------------------------------------------------------------- /images/Results_Natural_Illumination/23_023_IMG_5052crop_7_tar.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Natural_Illumination/23_023_IMG_5052crop_7_tar.jpg -------------------------------------------------------------------------------- /images/Results_Natural_Illumination/29_029_IMG_5058crop_0_ref.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Natural_Illumination/29_029_IMG_5058crop_0_ref.jpg -------------------------------------------------------------------------------- /images/Results_Natural_Illumination/29_029_IMG_5058crop_1_fcolor_r.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Natural_Illumination/29_029_IMG_5058crop_1_fcolor_r.jpg -------------------------------------------------------------------------------- /images/Results_Natural_Illumination/29_029_IMG_5058crop_3_mask_r.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Natural_Illumination/29_029_IMG_5058crop_3_mask_r.jpg -------------------------------------------------------------------------------- /images/Results_Natural_Illumination/29_029_IMG_5058crop_4_rho_r.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Natural_Illumination/29_029_IMG_5058crop_4_rho_r.jpg -------------------------------------------------------------------------------- /images/Results_Natural_Illumination/29_029_IMG_5058crop_5_final_r.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Natural_Illumination/29_029_IMG_5058crop_5_final_r.jpg -------------------------------------------------------------------------------- /images/Results_Natural_Illumination/29_029_IMG_5058crop_7_tar.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Natural_Illumination/29_029_IMG_5058crop_7_tar.jpg -------------------------------------------------------------------------------- /images/Results_Natural_Illumination/57_057_IMG_5084crop_0_ref.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Natural_Illumination/57_057_IMG_5084crop_0_ref.jpg -------------------------------------------------------------------------------- /images/Results_Natural_Illumination/57_057_IMG_5084crop_1_fcolor_r.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Natural_Illumination/57_057_IMG_5084crop_1_fcolor_r.jpg -------------------------------------------------------------------------------- /images/Results_Natural_Illumination/57_057_IMG_5084crop_3_mask_r.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Natural_Illumination/57_057_IMG_5084crop_3_mask_r.jpg -------------------------------------------------------------------------------- /images/Results_Natural_Illumination/57_057_IMG_5084crop_4_rho_r.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Natural_Illumination/57_057_IMG_5084crop_4_rho_r.jpg -------------------------------------------------------------------------------- /images/Results_Natural_Illumination/57_057_IMG_5084crop_5_final_r.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Natural_Illumination/57_057_IMG_5084crop_5_final_r.jpg -------------------------------------------------------------------------------- /images/Results_Natural_Illumination/57_057_IMG_5084crop_7_tar.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/Results_Natural_Illumination/57_057_IMG_5084crop_7_tar.jpg -------------------------------------------------------------------------------- /images/TOM-Net_with_Trimap_Bg_Samples/dragon.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/TOM-Net_with_Trimap_Bg_Samples/dragon.jpg -------------------------------------------------------------------------------- /images/TOM-Net_with_Trimap_Bg_Samples/dragon_bg.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/TOM-Net_with_Trimap_Bg_Samples/dragon_bg.jpg -------------------------------------------------------------------------------- /images/TOM-Net_with_Trimap_Bg_Samples/dragon_trimap.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/TOM-Net_with_Trimap_Bg_Samples/dragon_trimap.png -------------------------------------------------------------------------------- /images/TOM-Net_with_Trimap_Bg_Samples/horse.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/TOM-Net_with_Trimap_Bg_Samples/horse.jpg -------------------------------------------------------------------------------- /images/TOM-Net_with_Trimap_Bg_Samples/horse_bg.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/TOM-Net_with_Trimap_Bg_Samples/horse_bg.jpg -------------------------------------------------------------------------------- /images/TOM-Net_with_Trimap_Bg_Samples/horse_trimap.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/TOM-Net_with_Trimap_Bg_Samples/horse_trimap.png -------------------------------------------------------------------------------- /images/TOM-Net_with_Trimap_Bg_Samples/img_bg_trimap_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/TOM-Net_with_Trimap_Bg_Samples/img_bg_trimap_list.txt -------------------------------------------------------------------------------- /images/TOM-Net_with_Trimap_Bg_Samples/two_glass.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/TOM-Net_with_Trimap_Bg_Samples/two_glass.jpg -------------------------------------------------------------------------------- /images/TOM-Net_with_Trimap_Bg_Samples/two_glass_bg.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/TOM-Net_with_Trimap_Bg_Samples/two_glass_bg.jpg -------------------------------------------------------------------------------- /images/TOM-Net_with_Trimap_Bg_Samples/two_glass_trimap.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/TOM-Net_with_Trimap_Bg_Samples/two_glass_trimap.png -------------------------------------------------------------------------------- /images/bull.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/bull.jpg -------------------------------------------------------------------------------- /images/cvpr2018_tom-net.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/images/cvpr2018_tom-net.jpg -------------------------------------------------------------------------------- /main.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/main.lua -------------------------------------------------------------------------------- /models/AffineGridGeneratorUSOF.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/models/AffineGridGeneratorUSOF.lua -------------------------------------------------------------------------------- /models/CoarseNet.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/models/CoarseNet.lua -------------------------------------------------------------------------------- /models/RefineNet.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/models/RefineNet.lua -------------------------------------------------------------------------------- /models/init.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/models/init.lua -------------------------------------------------------------------------------- /models/model_utils.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/models/model_utils.lua -------------------------------------------------------------------------------- /opts.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/opts.lua -------------------------------------------------------------------------------- /refine/main_refine.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/refine/main_refine.lua -------------------------------------------------------------------------------- /refine/opts_refine.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/refine/opts_refine.lua -------------------------------------------------------------------------------- /scripts/download_pretrained_model.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/scripts/download_pretrained_model.sh -------------------------------------------------------------------------------- /scripts/download_pretrained_models_IJCV.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/scripts/download_pretrained_models_IJCV.sh -------------------------------------------------------------------------------- /scripts/download_testing_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/scripts/download_testing_dataset.sh -------------------------------------------------------------------------------- /scripts/download_training_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/scripts/download_training_dataset.sh -------------------------------------------------------------------------------- /scripts/download_validation_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/scripts/download_validation_dataset.sh -------------------------------------------------------------------------------- /train.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/train.lua -------------------------------------------------------------------------------- /utils/dict_utils.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/utils/dict_utils.lua -------------------------------------------------------------------------------- /utils/flow_utils.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/utils/flow_utils.lua -------------------------------------------------------------------------------- /utils/io_utils.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/utils/io_utils.lua -------------------------------------------------------------------------------- /utils/str_utils.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/guanyingc/TOM-Net/HEAD/utils/str_utils.lua --------------------------------------------------------------------------------