├── .gitignore ├── README.md ├── __pycache__ ├── inference.cpython-310.pyc ├── inference.cpython-39.pyc ├── loss.cpython-310.pyc └── loss.cpython-39.pyc ├── checkpoints └── transcg_checkpoint.tar ├── configs ├── default.yaml ├── inference.yaml ├── train_cgsyn+ood_val_cgreal.yaml ├── train_cgsyn+ood_val_transcg.yaml └── train_transcg_val_transcg.yaml ├── copycheckpoint.py ├── datasets ├── __pycache__ │ ├── cleargrasp.cpython-310.pyc │ ├── cleargrasp.cpython-39.pyc │ ├── mixed_dataset.cpython-310.pyc │ ├── mixed_dataset.cpython-39.pyc │ ├── omniverse_object.cpython-310.pyc │ ├── omniverse_object.cpython-39.pyc │ ├── transcg.cpython-310.pyc │ ├── transcg.cpython-39.pyc │ ├── transparent_object.cpython-310.pyc │ └── transparent_object.cpython-39.pyc ├── cleargrasp.py ├── mixed_dataset.py ├── omniverse_object.py ├── transcg.py └── transparent_object.py ├── inference.py ├── models ├── Tode.py ├── __init__.py ├── __pycache__ │ ├── DFNet.cpython-310.pyc │ ├── DFNet.cpython-39.pyc │ ├── Tode.cpython-310.pyc │ ├── __init__.cpython-310.pyc │ ├── __init__.cpython-39.pyc │ ├── canet.cpython-310.pyc │ ├── canet.cpython-39.pyc │ ├── dense.cpython-310.pyc │ ├── dense.cpython-39.pyc │ ├── duc.cpython-310.pyc │ ├── duc.cpython-39.pyc │ ├── layers.cpython-310.pyc │ ├── layers.cpython-39.pyc │ ├── lg_transformer.cpython-310.pyc │ ├── lgtnet.cpython-310.pyc │ ├── miniViT.cpython-310.pyc │ ├── miniViT.cpython-39.pyc │ ├── swin_transformer.cpython-310.pyc │ ├── swin_transformer.cpython-39.pyc │ ├── trannet.cpython-310.pyc │ ├── trannet.cpython-39.pyc │ ├── trannet2.cpython-39.pyc │ ├── trannet3.cpython-39.pyc │ ├── trannet5.cpython-310.pyc │ ├── trannet5.cpython-39.pyc │ ├── trannet_pyramid.cpython-310.pyc │ ├── unet_adaptive_bins.cpython-39.pyc │ ├── weight_init.cpython-310.pyc │ └── weight_init.cpython-39.pyc └── swin_transformer.py ├── requirements.txt ├── sample_inference.py ├── test.py ├── train.py └── utils ├── __init__.py ├── __pycache__ ├── __init__.cpython-310.pyc ├── __init__.cpython-39.pyc ├── builder.cpython-310.pyc ├── builder.cpython-39.pyc ├── constants.cpython-310.pyc ├── constants.cpython-39.pyc ├── criterion.cpython-310.pyc ├── criterion.cpython-39.pyc ├── data_preparation.cpython-310.pyc ├── data_preparation.cpython-39.pyc ├── functions.cpython-310.pyc ├── functions.cpython-39.pyc ├── logger.cpython-310.pyc ├── logger.cpython-39.pyc ├── metrics.cpython-310.pyc └── metrics.cpython-39.pyc ├── builder.py ├── constants.py ├── criterion.py ├── data_preparation.py ├── functions.py ├── logger.py └── metrics.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/README.md -------------------------------------------------------------------------------- /__pycache__/inference.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/__pycache__/inference.cpython-310.pyc -------------------------------------------------------------------------------- /__pycache__/inference.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/__pycache__/inference.cpython-39.pyc -------------------------------------------------------------------------------- /__pycache__/loss.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/__pycache__/loss.cpython-310.pyc -------------------------------------------------------------------------------- /__pycache__/loss.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/__pycache__/loss.cpython-39.pyc -------------------------------------------------------------------------------- /checkpoints/transcg_checkpoint.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/checkpoints/transcg_checkpoint.tar -------------------------------------------------------------------------------- /configs/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/configs/default.yaml -------------------------------------------------------------------------------- /configs/inference.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/configs/inference.yaml -------------------------------------------------------------------------------- /configs/train_cgsyn+ood_val_cgreal.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/configs/train_cgsyn+ood_val_cgreal.yaml -------------------------------------------------------------------------------- /configs/train_cgsyn+ood_val_transcg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/configs/train_cgsyn+ood_val_transcg.yaml -------------------------------------------------------------------------------- /configs/train_transcg_val_transcg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/configs/train_transcg_val_transcg.yaml -------------------------------------------------------------------------------- /copycheckpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/copycheckpoint.py -------------------------------------------------------------------------------- /datasets/__pycache__/cleargrasp.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/datasets/__pycache__/cleargrasp.cpython-310.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/cleargrasp.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/datasets/__pycache__/cleargrasp.cpython-39.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/mixed_dataset.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/datasets/__pycache__/mixed_dataset.cpython-310.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/mixed_dataset.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/datasets/__pycache__/mixed_dataset.cpython-39.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/omniverse_object.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/datasets/__pycache__/omniverse_object.cpython-310.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/omniverse_object.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/datasets/__pycache__/omniverse_object.cpython-39.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/transcg.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/datasets/__pycache__/transcg.cpython-310.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/transcg.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/datasets/__pycache__/transcg.cpython-39.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/transparent_object.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/datasets/__pycache__/transparent_object.cpython-310.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/transparent_object.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/datasets/__pycache__/transparent_object.cpython-39.pyc -------------------------------------------------------------------------------- /datasets/cleargrasp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/datasets/cleargrasp.py -------------------------------------------------------------------------------- /datasets/mixed_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/datasets/mixed_dataset.py -------------------------------------------------------------------------------- /datasets/omniverse_object.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/datasets/omniverse_object.py -------------------------------------------------------------------------------- /datasets/transcg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/datasets/transcg.py -------------------------------------------------------------------------------- /datasets/transparent_object.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/datasets/transparent_object.py -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/inference.py -------------------------------------------------------------------------------- /models/Tode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/Tode.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/__pycache__/DFNet.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/DFNet.cpython-310.pyc -------------------------------------------------------------------------------- /models/__pycache__/DFNet.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/DFNet.cpython-39.pyc -------------------------------------------------------------------------------- /models/__pycache__/Tode.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/Tode.cpython-310.pyc -------------------------------------------------------------------------------- /models/__pycache__/__init__.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/__init__.cpython-310.pyc -------------------------------------------------------------------------------- /models/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /models/__pycache__/canet.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/canet.cpython-310.pyc -------------------------------------------------------------------------------- /models/__pycache__/canet.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/canet.cpython-39.pyc -------------------------------------------------------------------------------- /models/__pycache__/dense.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/dense.cpython-310.pyc -------------------------------------------------------------------------------- /models/__pycache__/dense.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/dense.cpython-39.pyc -------------------------------------------------------------------------------- /models/__pycache__/duc.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/duc.cpython-310.pyc -------------------------------------------------------------------------------- /models/__pycache__/duc.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/duc.cpython-39.pyc -------------------------------------------------------------------------------- /models/__pycache__/layers.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/layers.cpython-310.pyc -------------------------------------------------------------------------------- /models/__pycache__/layers.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/layers.cpython-39.pyc -------------------------------------------------------------------------------- /models/__pycache__/lg_transformer.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/lg_transformer.cpython-310.pyc -------------------------------------------------------------------------------- /models/__pycache__/lgtnet.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/lgtnet.cpython-310.pyc -------------------------------------------------------------------------------- /models/__pycache__/miniViT.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/miniViT.cpython-310.pyc -------------------------------------------------------------------------------- /models/__pycache__/miniViT.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/miniViT.cpython-39.pyc -------------------------------------------------------------------------------- /models/__pycache__/swin_transformer.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/swin_transformer.cpython-310.pyc -------------------------------------------------------------------------------- /models/__pycache__/swin_transformer.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/swin_transformer.cpython-39.pyc -------------------------------------------------------------------------------- /models/__pycache__/trannet.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/trannet.cpython-310.pyc -------------------------------------------------------------------------------- /models/__pycache__/trannet.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/trannet.cpython-39.pyc -------------------------------------------------------------------------------- /models/__pycache__/trannet2.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/trannet2.cpython-39.pyc -------------------------------------------------------------------------------- /models/__pycache__/trannet3.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/trannet3.cpython-39.pyc -------------------------------------------------------------------------------- /models/__pycache__/trannet5.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/trannet5.cpython-310.pyc -------------------------------------------------------------------------------- /models/__pycache__/trannet5.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/trannet5.cpython-39.pyc -------------------------------------------------------------------------------- /models/__pycache__/trannet_pyramid.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/trannet_pyramid.cpython-310.pyc -------------------------------------------------------------------------------- /models/__pycache__/unet_adaptive_bins.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/unet_adaptive_bins.cpython-39.pyc -------------------------------------------------------------------------------- /models/__pycache__/weight_init.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/weight_init.cpython-310.pyc -------------------------------------------------------------------------------- /models/__pycache__/weight_init.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/__pycache__/weight_init.cpython-39.pyc -------------------------------------------------------------------------------- /models/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/models/swin_transformer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/requirements.txt -------------------------------------------------------------------------------- /sample_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/sample_inference.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/__pycache__/__init__.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/utils/__pycache__/__init__.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/utils/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /utils/__pycache__/builder.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/utils/__pycache__/builder.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/builder.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/utils/__pycache__/builder.cpython-39.pyc -------------------------------------------------------------------------------- /utils/__pycache__/constants.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/utils/__pycache__/constants.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/constants.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/utils/__pycache__/constants.cpython-39.pyc -------------------------------------------------------------------------------- /utils/__pycache__/criterion.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/utils/__pycache__/criterion.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/criterion.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/utils/__pycache__/criterion.cpython-39.pyc -------------------------------------------------------------------------------- /utils/__pycache__/data_preparation.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/utils/__pycache__/data_preparation.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/data_preparation.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/utils/__pycache__/data_preparation.cpython-39.pyc -------------------------------------------------------------------------------- /utils/__pycache__/functions.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/utils/__pycache__/functions.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/functions.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/utils/__pycache__/functions.cpython-39.pyc -------------------------------------------------------------------------------- /utils/__pycache__/logger.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/utils/__pycache__/logger.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/logger.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/utils/__pycache__/logger.cpython-39.pyc -------------------------------------------------------------------------------- /utils/__pycache__/metrics.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/utils/__pycache__/metrics.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/metrics.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/utils/__pycache__/metrics.cpython-39.pyc -------------------------------------------------------------------------------- /utils/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/utils/builder.py -------------------------------------------------------------------------------- /utils/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/utils/constants.py -------------------------------------------------------------------------------- /utils/criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/utils/criterion.py -------------------------------------------------------------------------------- /utils/data_preparation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/utils/data_preparation.py -------------------------------------------------------------------------------- /utils/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/utils/functions.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuchendoudou/TODE/HEAD/utils/metrics.py --------------------------------------------------------------------------------