├── .gitignore ├── .gitmodules ├── README.md ├── data ├── FiveK │ ├── FiveK.json │ ├── FiveK.py │ ├── README.md │ └── annotations │ │ ├── test_sess_1.json │ │ ├── train_sess_1.json │ │ └── val_sess_1.json ├── GIER │ ├── GIER.json │ ├── GIER.py │ ├── README.md │ ├── define_split.py │ ├── splits │ │ ├── Ids_L1Thr_0.06.json │ │ ├── test_Ids_L1Thr_0.06_sess_3.json │ │ ├── test_global_sess_3.json │ │ ├── test_sess_3.json │ │ ├── test_shapeAlignNonCrop_sess_3.json │ │ ├── test_shapeAlign_sess_3.json │ │ ├── train_Ids_L1Thr_0.06_sess_3.json │ │ ├── train_global_sess_3.json │ │ ├── train_sess_3.json │ │ ├── train_shapeAlignNonCrop_sess_3.json │ │ ├── train_shapeAlign_sess_3.json │ │ ├── val_Ids_L1Thr_0.06_sess_3.json │ │ ├── val_global_sess_3.json │ │ ├── val_sess_3.json │ │ ├── val_shapeAlignNonCrop_sess_3.json │ │ └── val_shapeAlign_sess_3.json │ └── url_dict.json └── language │ ├── FiveK_operator_vocabs_sess_1.json │ ├── FiveK_vocabs_glove_feat_1.h5 │ ├── FiveK_vocabs_sess_1.json │ ├── GIER_operator_vocabs_sess_3.json │ ├── GIER_vocabs_glove_feat_3.h5 │ └── GIER_vocabs_sess_3.json ├── datasets ├── FiveKdataset.py ├── GIERdataset.py └── __init__.py ├── demo ├── run_demo_FiveK.sh └── seq2seqL1.py ├── executors └── executor.py ├── experiments ├── t2onet+D-L1 │ ├── test_seq2seqGAN.py │ └── train_seq2seqGAN.py ├── t2onet-L1 │ ├── test_actor_fs.py │ └── train_actor_fs.py └── t2onet │ ├── test_GIER_seq2seqL1.py │ ├── test_seq2seqL1.py │ ├── train_GIER_seq2seqL1.py │ └── train_seq2seqL1.py ├── models ├── action_decoder.py ├── actor.py ├── actor_resnet.py ├── attention.py ├── lang_encoder.py ├── operators.py └── seq2seqGAN │ ├── networks.py │ ├── seq2seqAdaptGAN.py │ ├── seq2seqGAN.py │ └── seq2seqGANDisc.py ├── options ├── HaiWang_base_options.py ├── HaiWang_train_options.py ├── bilinearGAN_base_options.py ├── bilinearGAN_train_options.py ├── fiveK_base_options.py ├── fiveK_train_options.py ├── seq2seqGAN_base_options.py └── seq2seqGAN_train_options.py ├── preprocess ├── gen_greedy_seqs_FiveK.py └── gen_greedy_seqs_GIER.py ├── requirements.txt └── utils ├── FID ├── fid_score.py └── inception.py ├── beam_search.py ├── beam_search_eps_greedy.py ├── beam_search_fixed_order.py ├── eval.py ├── html.py ├── operator_utils.py ├── ssim └── __init__.py ├── text_utils.py ├── visual_utils.py └── visualize.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/.gitmodules -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/README.md -------------------------------------------------------------------------------- /data/FiveK/FiveK.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/FiveK/FiveK.json -------------------------------------------------------------------------------- /data/FiveK/FiveK.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/FiveK/FiveK.py -------------------------------------------------------------------------------- /data/FiveK/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/FiveK/README.md -------------------------------------------------------------------------------- /data/FiveK/annotations/test_sess_1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/FiveK/annotations/test_sess_1.json -------------------------------------------------------------------------------- /data/FiveK/annotations/train_sess_1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/FiveK/annotations/train_sess_1.json -------------------------------------------------------------------------------- /data/FiveK/annotations/val_sess_1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/FiveK/annotations/val_sess_1.json -------------------------------------------------------------------------------- /data/GIER/GIER.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/GIER/GIER.json -------------------------------------------------------------------------------- /data/GIER/GIER.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/GIER/GIER.py -------------------------------------------------------------------------------- /data/GIER/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/GIER/README.md -------------------------------------------------------------------------------- /data/GIER/define_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/GIER/define_split.py -------------------------------------------------------------------------------- /data/GIER/splits/Ids_L1Thr_0.06.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/GIER/splits/Ids_L1Thr_0.06.json -------------------------------------------------------------------------------- /data/GIER/splits/test_Ids_L1Thr_0.06_sess_3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/GIER/splits/test_Ids_L1Thr_0.06_sess_3.json -------------------------------------------------------------------------------- /data/GIER/splits/test_global_sess_3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/GIER/splits/test_global_sess_3.json -------------------------------------------------------------------------------- /data/GIER/splits/test_sess_3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/GIER/splits/test_sess_3.json -------------------------------------------------------------------------------- /data/GIER/splits/test_shapeAlignNonCrop_sess_3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/GIER/splits/test_shapeAlignNonCrop_sess_3.json -------------------------------------------------------------------------------- /data/GIER/splits/test_shapeAlign_sess_3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/GIER/splits/test_shapeAlign_sess_3.json -------------------------------------------------------------------------------- /data/GIER/splits/train_Ids_L1Thr_0.06_sess_3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/GIER/splits/train_Ids_L1Thr_0.06_sess_3.json -------------------------------------------------------------------------------- /data/GIER/splits/train_global_sess_3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/GIER/splits/train_global_sess_3.json -------------------------------------------------------------------------------- /data/GIER/splits/train_sess_3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/GIER/splits/train_sess_3.json -------------------------------------------------------------------------------- /data/GIER/splits/train_shapeAlignNonCrop_sess_3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/GIER/splits/train_shapeAlignNonCrop_sess_3.json -------------------------------------------------------------------------------- /data/GIER/splits/train_shapeAlign_sess_3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/GIER/splits/train_shapeAlign_sess_3.json -------------------------------------------------------------------------------- /data/GIER/splits/val_Ids_L1Thr_0.06_sess_3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/GIER/splits/val_Ids_L1Thr_0.06_sess_3.json -------------------------------------------------------------------------------- /data/GIER/splits/val_global_sess_3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/GIER/splits/val_global_sess_3.json -------------------------------------------------------------------------------- /data/GIER/splits/val_sess_3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/GIER/splits/val_sess_3.json -------------------------------------------------------------------------------- /data/GIER/splits/val_shapeAlignNonCrop_sess_3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/GIER/splits/val_shapeAlignNonCrop_sess_3.json -------------------------------------------------------------------------------- /data/GIER/splits/val_shapeAlign_sess_3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/GIER/splits/val_shapeAlign_sess_3.json -------------------------------------------------------------------------------- /data/GIER/url_dict.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/GIER/url_dict.json -------------------------------------------------------------------------------- /data/language/FiveK_operator_vocabs_sess_1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/language/FiveK_operator_vocabs_sess_1.json -------------------------------------------------------------------------------- /data/language/FiveK_vocabs_glove_feat_1.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/language/FiveK_vocabs_glove_feat_1.h5 -------------------------------------------------------------------------------- /data/language/FiveK_vocabs_sess_1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/language/FiveK_vocabs_sess_1.json -------------------------------------------------------------------------------- /data/language/GIER_operator_vocabs_sess_3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/language/GIER_operator_vocabs_sess_3.json -------------------------------------------------------------------------------- /data/language/GIER_vocabs_glove_feat_3.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/language/GIER_vocabs_glove_feat_3.h5 -------------------------------------------------------------------------------- /data/language/GIER_vocabs_sess_3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/data/language/GIER_vocabs_sess_3.json -------------------------------------------------------------------------------- /datasets/FiveKdataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/datasets/FiveKdataset.py -------------------------------------------------------------------------------- /datasets/GIERdataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/datasets/GIERdataset.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /demo/run_demo_FiveK.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/demo/run_demo_FiveK.sh -------------------------------------------------------------------------------- /demo/seq2seqL1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/demo/seq2seqL1.py -------------------------------------------------------------------------------- /executors/executor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/executors/executor.py -------------------------------------------------------------------------------- /experiments/t2onet+D-L1/test_seq2seqGAN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/experiments/t2onet+D-L1/test_seq2seqGAN.py -------------------------------------------------------------------------------- /experiments/t2onet+D-L1/train_seq2seqGAN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/experiments/t2onet+D-L1/train_seq2seqGAN.py -------------------------------------------------------------------------------- /experiments/t2onet-L1/test_actor_fs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/experiments/t2onet-L1/test_actor_fs.py -------------------------------------------------------------------------------- /experiments/t2onet-L1/train_actor_fs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/experiments/t2onet-L1/train_actor_fs.py -------------------------------------------------------------------------------- /experiments/t2onet/test_GIER_seq2seqL1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/experiments/t2onet/test_GIER_seq2seqL1.py -------------------------------------------------------------------------------- /experiments/t2onet/test_seq2seqL1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/experiments/t2onet/test_seq2seqL1.py -------------------------------------------------------------------------------- /experiments/t2onet/train_GIER_seq2seqL1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/experiments/t2onet/train_GIER_seq2seqL1.py -------------------------------------------------------------------------------- /experiments/t2onet/train_seq2seqL1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/experiments/t2onet/train_seq2seqL1.py -------------------------------------------------------------------------------- /models/action_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/models/action_decoder.py -------------------------------------------------------------------------------- /models/actor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/models/actor.py -------------------------------------------------------------------------------- /models/actor_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/models/actor_resnet.py -------------------------------------------------------------------------------- /models/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/models/attention.py -------------------------------------------------------------------------------- /models/lang_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/models/lang_encoder.py -------------------------------------------------------------------------------- /models/operators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/models/operators.py -------------------------------------------------------------------------------- /models/seq2seqGAN/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/models/seq2seqGAN/networks.py -------------------------------------------------------------------------------- /models/seq2seqGAN/seq2seqAdaptGAN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/models/seq2seqGAN/seq2seqAdaptGAN.py -------------------------------------------------------------------------------- /models/seq2seqGAN/seq2seqGAN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/models/seq2seqGAN/seq2seqGAN.py -------------------------------------------------------------------------------- /models/seq2seqGAN/seq2seqGANDisc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/models/seq2seqGAN/seq2seqGANDisc.py -------------------------------------------------------------------------------- /options/HaiWang_base_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/options/HaiWang_base_options.py -------------------------------------------------------------------------------- /options/HaiWang_train_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/options/HaiWang_train_options.py -------------------------------------------------------------------------------- /options/bilinearGAN_base_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/options/bilinearGAN_base_options.py -------------------------------------------------------------------------------- /options/bilinearGAN_train_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/options/bilinearGAN_train_options.py -------------------------------------------------------------------------------- /options/fiveK_base_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/options/fiveK_base_options.py -------------------------------------------------------------------------------- /options/fiveK_train_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/options/fiveK_train_options.py -------------------------------------------------------------------------------- /options/seq2seqGAN_base_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/options/seq2seqGAN_base_options.py -------------------------------------------------------------------------------- /options/seq2seqGAN_train_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/options/seq2seqGAN_train_options.py -------------------------------------------------------------------------------- /preprocess/gen_greedy_seqs_FiveK.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/preprocess/gen_greedy_seqs_FiveK.py -------------------------------------------------------------------------------- /preprocess/gen_greedy_seqs_GIER.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/preprocess/gen_greedy_seqs_GIER.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/requirements.txt -------------------------------------------------------------------------------- /utils/FID/fid_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/utils/FID/fid_score.py -------------------------------------------------------------------------------- /utils/FID/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/utils/FID/inception.py -------------------------------------------------------------------------------- /utils/beam_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/utils/beam_search.py -------------------------------------------------------------------------------- /utils/beam_search_eps_greedy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/utils/beam_search_eps_greedy.py -------------------------------------------------------------------------------- /utils/beam_search_fixed_order.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/utils/beam_search_fixed_order.py -------------------------------------------------------------------------------- /utils/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/utils/eval.py -------------------------------------------------------------------------------- /utils/html.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/utils/html.py -------------------------------------------------------------------------------- /utils/operator_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/utils/operator_utils.py -------------------------------------------------------------------------------- /utils/ssim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/utils/ssim/__init__.py -------------------------------------------------------------------------------- /utils/text_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/utils/text_utils.py -------------------------------------------------------------------------------- /utils/visual_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/utils/visual_utils.py -------------------------------------------------------------------------------- /utils/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshi31/T2ONet/HEAD/utils/visualize.py --------------------------------------------------------------------------------