├── .gitignore ├── LICENSE ├── README.md ├── adaptations ├── __init__.py ├── compute_iou.py ├── dataset │ ├── __init__.py │ ├── cityscapes_list │ │ ├── info.json │ │ ├── label.txt │ │ ├── train.txt │ │ └── val.txt │ ├── cs13_dataset_src.py │ ├── cs_dataset_src.py │ ├── densepass_dataset.py │ ├── densepass_list │ │ ├── info.json │ │ ├── label.txt │ │ ├── train.txt │ │ └── val.txt │ ├── dp13_dataset.py │ ├── s2d3d_pan_list │ │ ├── info.json │ │ ├── label.txt │ │ ├── train.txt │ │ ├── trainval.txt │ │ └── val.txt │ ├── s2d3d_pin_list │ │ ├── colors.npy │ │ ├── label.txt │ │ ├── name2label.json │ │ ├── semantic_labels.json │ │ ├── train.txt │ │ ├── trainval.txt │ │ └── val.txt │ ├── s3d8_dataset.py │ ├── sp13_dataset.py │ ├── sp22_dataset.py │ ├── stanford_pan8_dataset.py │ ├── stanford_pan_dataset.py │ ├── stanford_pin8_dataset.py │ ├── stanford_pin_dataset.py │ ├── structured3d_list │ │ ├── label.txt │ │ ├── test.txt │ │ ├── train.txt │ │ └── val.txt │ └── synpass_list │ │ ├── label.txt │ │ ├── test.txt │ │ ├── train.txt │ │ └── val.txt ├── datasets ├── evaluate.py ├── evaluate_out13.py ├── gen_pseudo_label.py ├── init_memory_joint_ms.npy ├── init_memory_joint_ms_Trans4PASS_plus_v2_CS13.npy ├── model │ ├── __init__.py │ ├── discriminator.py │ ├── memory.py │ ├── trans4pass.py │ └── trans4passplus.py ├── train_mpa.py ├── train_mpa_out_p2p.py ├── train_ssl.py ├── train_warm.py ├── train_warm_out_p2p.py ├── utils │ ├── __init__.py │ ├── constant.py │ ├── functions.py │ ├── loss.py │ └── transform.py └── workdirs ├── colors.npy ├── configs ├── cityscapes │ ├── trans4pass_plus_small_512x512.yaml │ ├── trans4pass_plus_tiny_512x512.yaml │ ├── trans4pass_small_512x512.yaml │ └── trans4pass_tiny_512x512.yaml ├── cityscapes13 │ ├── trans4pass_plus_small_512x512.yaml │ └── trans4pass_plus_tiny_512x512.yaml ├── stanford2d3d │ ├── trans4pass_plus_small_1080x1080.yaml │ ├── trans4pass_plus_tiny_1080x1080.yaml │ ├── trans4pass_small_1080x1080.yaml │ └── trans4pass_tiny_1080x1080.yaml ├── stanford2d3d8 │ ├── trans4pass_plus_small_1080x1080.yaml │ └── trans4pass_plus_tiny_1080x1080.yaml ├── stanford2d3d_pan │ ├── trans4pass_plus_small_1080x1080.yaml │ ├── trans4pass_small_1080x1080.yaml │ └── trans4pass_tiny_1080x1080.yaml ├── structured3d8 │ ├── trans4pass_plus_small_512x512.yaml │ └── trans4pass_plus_tiny_512x512.yaml ├── synpass │ ├── trans4pass_plus_small_512x512.yaml │ ├── trans4pass_plus_tiny_512x512.yaml │ ├── trans4pass_small_512x512.yaml │ └── trans4pass_tiny_512x512.yaml └── synpass13 │ ├── trans4pass_plus_small_512x512.yaml │ └── trans4pass_plus_tiny_512x512.yaml ├── figs ├── fig_trans4pass.png ├── fig_trans4pass_plus.png └── fig_vis_synpass.png ├── name2label.json ├── requirements.txt ├── segmentron ├── __init__.py ├── config │ ├── __init__.py │ ├── config.py │ └── settings.py ├── data │ ├── __init__.py │ ├── dataloader │ │ ├── __init__.py │ │ ├── cityscapes.py │ │ ├── cityscapes13.py │ │ ├── densepass.py │ │ ├── densepass13.py │ │ ├── seg_data_base.py │ │ ├── stanford2d3d.py │ │ ├── stanford2d3d8.py │ │ ├── stanford2d3d_pan.py │ │ ├── stanford2d3d_pan8.py │ │ ├── structured3d8.py │ │ ├── synpass.py │ │ ├── synpass13.py │ │ └── utils.py │ └── downloader │ │ ├── __init__.py │ │ ├── ade20k.py │ │ ├── cityscapes.py │ │ ├── mscoco.py │ │ ├── pascal_voc.py │ │ └── sbu_shadow.py ├── models │ ├── __init__.py │ ├── backbones │ │ ├── __init__.py │ │ ├── build.py │ │ ├── eespnet.py │ │ ├── hrnet.py │ │ ├── mobilenet.py │ │ ├── resnet.py │ │ ├── trans4pass.py │ │ └── xception.py │ ├── model_zoo.py │ ├── segbase.py │ └── trans4pass.py ├── modules │ ├── __init__.py │ ├── basic.py │ ├── batch_norm.py │ ├── csrc │ │ ├── criss_cross_attention │ │ │ ├── ca.h │ │ │ └── ca_cuda.cu │ │ └── vision.cpp │ ├── dmlp.py │ ├── dmlpv2.py │ ├── drop.py │ ├── module.py │ ├── norm.py │ └── sync_bn │ │ └── syncbn.py ├── solver │ ├── __init__.py │ ├── loss.py │ ├── lovasz_losses.py │ ├── lr_scheduler.py │ └── optimizer.py └── utils │ ├── __init__.py │ ├── default_setup.py │ ├── distributed.py │ ├── download.py │ ├── env.py │ ├── filesystem.py │ ├── logger.py │ ├── options.py │ ├── parallel.py │ ├── registry.py │ ├── score.py │ └── visualize.py ├── semantic_labels.json ├── setup.py └── tools ├── eval_dp.py ├── eval_dp13.py ├── eval_s2d3d.py ├── eval_s2d3d8.py ├── eval_sp.py ├── train_cs.py ├── train_cs13.py ├── train_s2d3d.py ├── train_s2d3d8.py ├── train_s2d3d_span.py └── train_sp.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/README.md -------------------------------------------------------------------------------- /adaptations/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /adaptations/compute_iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/compute_iou.py -------------------------------------------------------------------------------- /adaptations/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /adaptations/dataset/cityscapes_list/info.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/cityscapes_list/info.json -------------------------------------------------------------------------------- /adaptations/dataset/cityscapes_list/label.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/cityscapes_list/label.txt -------------------------------------------------------------------------------- /adaptations/dataset/cityscapes_list/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/cityscapes_list/train.txt -------------------------------------------------------------------------------- /adaptations/dataset/cityscapes_list/val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/cityscapes_list/val.txt -------------------------------------------------------------------------------- /adaptations/dataset/cs13_dataset_src.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/cs13_dataset_src.py -------------------------------------------------------------------------------- /adaptations/dataset/cs_dataset_src.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/cs_dataset_src.py -------------------------------------------------------------------------------- /adaptations/dataset/densepass_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/densepass_dataset.py -------------------------------------------------------------------------------- /adaptations/dataset/densepass_list/info.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/densepass_list/info.json -------------------------------------------------------------------------------- /adaptations/dataset/densepass_list/label.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/densepass_list/label.txt -------------------------------------------------------------------------------- /adaptations/dataset/densepass_list/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/densepass_list/train.txt -------------------------------------------------------------------------------- /adaptations/dataset/densepass_list/val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/densepass_list/val.txt -------------------------------------------------------------------------------- /adaptations/dataset/dp13_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/dp13_dataset.py -------------------------------------------------------------------------------- /adaptations/dataset/s2d3d_pan_list/info.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/s2d3d_pan_list/info.json -------------------------------------------------------------------------------- /adaptations/dataset/s2d3d_pan_list/label.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/s2d3d_pan_list/label.txt -------------------------------------------------------------------------------- /adaptations/dataset/s2d3d_pan_list/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/s2d3d_pan_list/train.txt -------------------------------------------------------------------------------- /adaptations/dataset/s2d3d_pan_list/trainval.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/s2d3d_pan_list/trainval.txt -------------------------------------------------------------------------------- /adaptations/dataset/s2d3d_pan_list/val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/s2d3d_pan_list/val.txt -------------------------------------------------------------------------------- /adaptations/dataset/s2d3d_pin_list/colors.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/s2d3d_pin_list/colors.npy -------------------------------------------------------------------------------- /adaptations/dataset/s2d3d_pin_list/label.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/s2d3d_pin_list/label.txt -------------------------------------------------------------------------------- /adaptations/dataset/s2d3d_pin_list/name2label.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/s2d3d_pin_list/name2label.json -------------------------------------------------------------------------------- /adaptations/dataset/s2d3d_pin_list/semantic_labels.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/s2d3d_pin_list/semantic_labels.json -------------------------------------------------------------------------------- /adaptations/dataset/s2d3d_pin_list/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/s2d3d_pin_list/train.txt -------------------------------------------------------------------------------- /adaptations/dataset/s2d3d_pin_list/trainval.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/s2d3d_pin_list/trainval.txt -------------------------------------------------------------------------------- /adaptations/dataset/s2d3d_pin_list/val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/s2d3d_pin_list/val.txt -------------------------------------------------------------------------------- /adaptations/dataset/s3d8_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/s3d8_dataset.py -------------------------------------------------------------------------------- /adaptations/dataset/sp13_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/sp13_dataset.py -------------------------------------------------------------------------------- /adaptations/dataset/sp22_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/sp22_dataset.py -------------------------------------------------------------------------------- /adaptations/dataset/stanford_pan8_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/stanford_pan8_dataset.py -------------------------------------------------------------------------------- /adaptations/dataset/stanford_pan_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/stanford_pan_dataset.py -------------------------------------------------------------------------------- /adaptations/dataset/stanford_pin8_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/stanford_pin8_dataset.py -------------------------------------------------------------------------------- /adaptations/dataset/stanford_pin_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/stanford_pin_dataset.py -------------------------------------------------------------------------------- /adaptations/dataset/structured3d_list/label.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/structured3d_list/label.txt -------------------------------------------------------------------------------- /adaptations/dataset/structured3d_list/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/structured3d_list/test.txt -------------------------------------------------------------------------------- /adaptations/dataset/structured3d_list/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/structured3d_list/train.txt -------------------------------------------------------------------------------- /adaptations/dataset/structured3d_list/val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/structured3d_list/val.txt -------------------------------------------------------------------------------- /adaptations/dataset/synpass_list/label.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/synpass_list/label.txt -------------------------------------------------------------------------------- /adaptations/dataset/synpass_list/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/synpass_list/test.txt -------------------------------------------------------------------------------- /adaptations/dataset/synpass_list/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/synpass_list/train.txt -------------------------------------------------------------------------------- /adaptations/dataset/synpass_list/val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/dataset/synpass_list/val.txt -------------------------------------------------------------------------------- /adaptations/datasets: -------------------------------------------------------------------------------- 1 | ../datasets -------------------------------------------------------------------------------- /adaptations/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/evaluate.py -------------------------------------------------------------------------------- /adaptations/evaluate_out13.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/evaluate_out13.py -------------------------------------------------------------------------------- /adaptations/gen_pseudo_label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/gen_pseudo_label.py -------------------------------------------------------------------------------- /adaptations/init_memory_joint_ms.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/init_memory_joint_ms.npy -------------------------------------------------------------------------------- /adaptations/init_memory_joint_ms_Trans4PASS_plus_v2_CS13.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/init_memory_joint_ms_Trans4PASS_plus_v2_CS13.npy -------------------------------------------------------------------------------- /adaptations/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /adaptations/model/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/model/discriminator.py -------------------------------------------------------------------------------- /adaptations/model/memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/model/memory.py -------------------------------------------------------------------------------- /adaptations/model/trans4pass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/model/trans4pass.py -------------------------------------------------------------------------------- /adaptations/model/trans4passplus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/model/trans4passplus.py -------------------------------------------------------------------------------- /adaptations/train_mpa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/train_mpa.py -------------------------------------------------------------------------------- /adaptations/train_mpa_out_p2p.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/train_mpa_out_p2p.py -------------------------------------------------------------------------------- /adaptations/train_ssl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/train_ssl.py -------------------------------------------------------------------------------- /adaptations/train_warm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/train_warm.py -------------------------------------------------------------------------------- /adaptations/train_warm_out_p2p.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/train_warm_out_p2p.py -------------------------------------------------------------------------------- /adaptations/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /adaptations/utils/constant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/utils/constant.py -------------------------------------------------------------------------------- /adaptations/utils/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/utils/functions.py -------------------------------------------------------------------------------- /adaptations/utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/utils/loss.py -------------------------------------------------------------------------------- /adaptations/utils/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/adaptations/utils/transform.py -------------------------------------------------------------------------------- /adaptations/workdirs: -------------------------------------------------------------------------------- 1 | ../workdirs -------------------------------------------------------------------------------- /colors.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/colors.npy -------------------------------------------------------------------------------- /configs/cityscapes/trans4pass_plus_small_512x512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/configs/cityscapes/trans4pass_plus_small_512x512.yaml -------------------------------------------------------------------------------- /configs/cityscapes/trans4pass_plus_tiny_512x512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/configs/cityscapes/trans4pass_plus_tiny_512x512.yaml -------------------------------------------------------------------------------- /configs/cityscapes/trans4pass_small_512x512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/configs/cityscapes/trans4pass_small_512x512.yaml -------------------------------------------------------------------------------- /configs/cityscapes/trans4pass_tiny_512x512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/configs/cityscapes/trans4pass_tiny_512x512.yaml -------------------------------------------------------------------------------- /configs/cityscapes13/trans4pass_plus_small_512x512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/configs/cityscapes13/trans4pass_plus_small_512x512.yaml -------------------------------------------------------------------------------- /configs/cityscapes13/trans4pass_plus_tiny_512x512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/configs/cityscapes13/trans4pass_plus_tiny_512x512.yaml -------------------------------------------------------------------------------- /configs/stanford2d3d/trans4pass_plus_small_1080x1080.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/configs/stanford2d3d/trans4pass_plus_small_1080x1080.yaml -------------------------------------------------------------------------------- /configs/stanford2d3d/trans4pass_plus_tiny_1080x1080.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/configs/stanford2d3d/trans4pass_plus_tiny_1080x1080.yaml -------------------------------------------------------------------------------- /configs/stanford2d3d/trans4pass_small_1080x1080.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/configs/stanford2d3d/trans4pass_small_1080x1080.yaml -------------------------------------------------------------------------------- /configs/stanford2d3d/trans4pass_tiny_1080x1080.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/configs/stanford2d3d/trans4pass_tiny_1080x1080.yaml -------------------------------------------------------------------------------- /configs/stanford2d3d8/trans4pass_plus_small_1080x1080.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/configs/stanford2d3d8/trans4pass_plus_small_1080x1080.yaml -------------------------------------------------------------------------------- /configs/stanford2d3d8/trans4pass_plus_tiny_1080x1080.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/configs/stanford2d3d8/trans4pass_plus_tiny_1080x1080.yaml -------------------------------------------------------------------------------- /configs/stanford2d3d_pan/trans4pass_plus_small_1080x1080.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/configs/stanford2d3d_pan/trans4pass_plus_small_1080x1080.yaml -------------------------------------------------------------------------------- /configs/stanford2d3d_pan/trans4pass_small_1080x1080.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/configs/stanford2d3d_pan/trans4pass_small_1080x1080.yaml -------------------------------------------------------------------------------- /configs/stanford2d3d_pan/trans4pass_tiny_1080x1080.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/configs/stanford2d3d_pan/trans4pass_tiny_1080x1080.yaml -------------------------------------------------------------------------------- /configs/structured3d8/trans4pass_plus_small_512x512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/configs/structured3d8/trans4pass_plus_small_512x512.yaml -------------------------------------------------------------------------------- /configs/structured3d8/trans4pass_plus_tiny_512x512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/configs/structured3d8/trans4pass_plus_tiny_512x512.yaml -------------------------------------------------------------------------------- /configs/synpass/trans4pass_plus_small_512x512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/configs/synpass/trans4pass_plus_small_512x512.yaml -------------------------------------------------------------------------------- /configs/synpass/trans4pass_plus_tiny_512x512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/configs/synpass/trans4pass_plus_tiny_512x512.yaml -------------------------------------------------------------------------------- /configs/synpass/trans4pass_small_512x512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/configs/synpass/trans4pass_small_512x512.yaml -------------------------------------------------------------------------------- /configs/synpass/trans4pass_tiny_512x512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/configs/synpass/trans4pass_tiny_512x512.yaml -------------------------------------------------------------------------------- /configs/synpass13/trans4pass_plus_small_512x512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/configs/synpass13/trans4pass_plus_small_512x512.yaml -------------------------------------------------------------------------------- /configs/synpass13/trans4pass_plus_tiny_512x512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/configs/synpass13/trans4pass_plus_tiny_512x512.yaml -------------------------------------------------------------------------------- /figs/fig_trans4pass.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/figs/fig_trans4pass.png -------------------------------------------------------------------------------- /figs/fig_trans4pass_plus.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/figs/fig_trans4pass_plus.png -------------------------------------------------------------------------------- /figs/fig_vis_synpass.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/figs/fig_vis_synpass.png -------------------------------------------------------------------------------- /name2label.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/name2label.json -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/requirements.txt -------------------------------------------------------------------------------- /segmentron/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/__init__.py -------------------------------------------------------------------------------- /segmentron/config/__init__.py: -------------------------------------------------------------------------------- 1 | from .settings import cfg -------------------------------------------------------------------------------- /segmentron/config/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/config/config.py -------------------------------------------------------------------------------- /segmentron/config/settings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/config/settings.py -------------------------------------------------------------------------------- /segmentron/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /segmentron/data/dataloader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/data/dataloader/__init__.py -------------------------------------------------------------------------------- /segmentron/data/dataloader/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/data/dataloader/cityscapes.py -------------------------------------------------------------------------------- /segmentron/data/dataloader/cityscapes13.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/data/dataloader/cityscapes13.py -------------------------------------------------------------------------------- /segmentron/data/dataloader/densepass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/data/dataloader/densepass.py -------------------------------------------------------------------------------- /segmentron/data/dataloader/densepass13.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/data/dataloader/densepass13.py -------------------------------------------------------------------------------- /segmentron/data/dataloader/seg_data_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/data/dataloader/seg_data_base.py -------------------------------------------------------------------------------- /segmentron/data/dataloader/stanford2d3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/data/dataloader/stanford2d3d.py -------------------------------------------------------------------------------- /segmentron/data/dataloader/stanford2d3d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/data/dataloader/stanford2d3d8.py -------------------------------------------------------------------------------- /segmentron/data/dataloader/stanford2d3d_pan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/data/dataloader/stanford2d3d_pan.py -------------------------------------------------------------------------------- /segmentron/data/dataloader/stanford2d3d_pan8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/data/dataloader/stanford2d3d_pan8.py -------------------------------------------------------------------------------- /segmentron/data/dataloader/structured3d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/data/dataloader/structured3d8.py -------------------------------------------------------------------------------- /segmentron/data/dataloader/synpass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/data/dataloader/synpass.py -------------------------------------------------------------------------------- /segmentron/data/dataloader/synpass13.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/data/dataloader/synpass13.py -------------------------------------------------------------------------------- /segmentron/data/dataloader/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/data/dataloader/utils.py -------------------------------------------------------------------------------- /segmentron/data/downloader/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /segmentron/data/downloader/ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/data/downloader/ade20k.py -------------------------------------------------------------------------------- /segmentron/data/downloader/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/data/downloader/cityscapes.py -------------------------------------------------------------------------------- /segmentron/data/downloader/mscoco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/data/downloader/mscoco.py -------------------------------------------------------------------------------- /segmentron/data/downloader/pascal_voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/data/downloader/pascal_voc.py -------------------------------------------------------------------------------- /segmentron/data/downloader/sbu_shadow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/data/downloader/sbu_shadow.py -------------------------------------------------------------------------------- /segmentron/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/models/__init__.py -------------------------------------------------------------------------------- /segmentron/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/models/backbones/__init__.py -------------------------------------------------------------------------------- /segmentron/models/backbones/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/models/backbones/build.py -------------------------------------------------------------------------------- /segmentron/models/backbones/eespnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/models/backbones/eespnet.py -------------------------------------------------------------------------------- /segmentron/models/backbones/hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/models/backbones/hrnet.py -------------------------------------------------------------------------------- /segmentron/models/backbones/mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/models/backbones/mobilenet.py -------------------------------------------------------------------------------- /segmentron/models/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/models/backbones/resnet.py -------------------------------------------------------------------------------- /segmentron/models/backbones/trans4pass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/models/backbones/trans4pass.py -------------------------------------------------------------------------------- /segmentron/models/backbones/xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/models/backbones/xception.py -------------------------------------------------------------------------------- /segmentron/models/model_zoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/models/model_zoo.py -------------------------------------------------------------------------------- /segmentron/models/segbase.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/models/segbase.py -------------------------------------------------------------------------------- /segmentron/models/trans4pass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/models/trans4pass.py -------------------------------------------------------------------------------- /segmentron/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/modules/__init__.py -------------------------------------------------------------------------------- /segmentron/modules/basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/modules/basic.py -------------------------------------------------------------------------------- /segmentron/modules/batch_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/modules/batch_norm.py -------------------------------------------------------------------------------- /segmentron/modules/csrc/criss_cross_attention/ca.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/modules/csrc/criss_cross_attention/ca.h -------------------------------------------------------------------------------- /segmentron/modules/csrc/criss_cross_attention/ca_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/modules/csrc/criss_cross_attention/ca_cuda.cu -------------------------------------------------------------------------------- /segmentron/modules/csrc/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/modules/csrc/vision.cpp -------------------------------------------------------------------------------- /segmentron/modules/dmlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/modules/dmlp.py -------------------------------------------------------------------------------- /segmentron/modules/dmlpv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/modules/dmlpv2.py -------------------------------------------------------------------------------- /segmentron/modules/drop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/modules/drop.py -------------------------------------------------------------------------------- /segmentron/modules/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/modules/module.py -------------------------------------------------------------------------------- /segmentron/modules/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/modules/norm.py -------------------------------------------------------------------------------- /segmentron/modules/sync_bn/syncbn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/modules/sync_bn/syncbn.py -------------------------------------------------------------------------------- /segmentron/solver/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /segmentron/solver/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/solver/loss.py -------------------------------------------------------------------------------- /segmentron/solver/lovasz_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/solver/lovasz_losses.py -------------------------------------------------------------------------------- /segmentron/solver/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/solver/lr_scheduler.py -------------------------------------------------------------------------------- /segmentron/solver/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/solver/optimizer.py -------------------------------------------------------------------------------- /segmentron/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/utils/__init__.py -------------------------------------------------------------------------------- /segmentron/utils/default_setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/utils/default_setup.py -------------------------------------------------------------------------------- /segmentron/utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/utils/distributed.py -------------------------------------------------------------------------------- /segmentron/utils/download.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/utils/download.py -------------------------------------------------------------------------------- /segmentron/utils/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/utils/env.py -------------------------------------------------------------------------------- /segmentron/utils/filesystem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/utils/filesystem.py -------------------------------------------------------------------------------- /segmentron/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/utils/logger.py -------------------------------------------------------------------------------- /segmentron/utils/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/utils/options.py -------------------------------------------------------------------------------- /segmentron/utils/parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/utils/parallel.py -------------------------------------------------------------------------------- /segmentron/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/utils/registry.py -------------------------------------------------------------------------------- /segmentron/utils/score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/utils/score.py -------------------------------------------------------------------------------- /segmentron/utils/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/segmentron/utils/visualize.py -------------------------------------------------------------------------------- /semantic_labels.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/semantic_labels.json -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/setup.py -------------------------------------------------------------------------------- /tools/eval_dp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/tools/eval_dp.py -------------------------------------------------------------------------------- /tools/eval_dp13.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/tools/eval_dp13.py -------------------------------------------------------------------------------- /tools/eval_s2d3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/tools/eval_s2d3d.py -------------------------------------------------------------------------------- /tools/eval_s2d3d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/tools/eval_s2d3d8.py -------------------------------------------------------------------------------- /tools/eval_sp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/tools/eval_sp.py -------------------------------------------------------------------------------- /tools/train_cs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/tools/train_cs.py -------------------------------------------------------------------------------- /tools/train_cs13.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/tools/train_cs13.py -------------------------------------------------------------------------------- /tools/train_s2d3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/tools/train_s2d3d.py -------------------------------------------------------------------------------- /tools/train_s2d3d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/tools/train_s2d3d8.py -------------------------------------------------------------------------------- /tools/train_s2d3d_span.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/tools/train_s2d3d_span.py -------------------------------------------------------------------------------- /tools/train_sp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jamycheung/Trans4PASS/HEAD/tools/train_sp.py --------------------------------------------------------------------------------