├── .github └── pipeline.png ├── .gitignore ├── GET_STARTED.md ├── LICENSE ├── README.md ├── config.py ├── configs ├── deformable_detr.yaml ├── detection_r50_seqhead.yaml ├── detection_r50_seqhead_plus_box_refine.yaml ├── keypoint_r50_baseline_50e.yaml ├── keypoint_r50_seqhead_150e.yaml ├── keypoint_r50_seqhead_50e.yaml └── tasks │ ├── coco_detection.json │ ├── coco_hybrid.json │ ├── coco_pose_bce.json │ └── coco_pose_focal.json ├── datasets ├── __init__.py ├── coco.py ├── coco_eval.py ├── coco_hybrid.py ├── coco_mtl.py ├── coco_panoptic.py ├── data_prefetcher.py ├── dataloader.py ├── imnet.py ├── panoptic_eval.py ├── reader.py ├── samplers.py ├── torchvision_datasets │ ├── __init__.py │ └── coco.py ├── transform_pipelines.py ├── transforms.py └── voc_map.py ├── engine.py ├── main.py ├── models ├── __init__.py ├── backbone │ ├── __init__.py │ ├── resnet.py │ └── swin_transformer.py ├── losses │ ├── __init__.py │ ├── asl_losses.py │ ├── classwise_criterion.py │ ├── e2e_keypoint_criterion.py │ ├── losses.py │ ├── matcher.py │ ├── matcher_kps.py │ ├── set_criterion.py │ ├── unified_matcher.py │ └── unified_single_class_criterion.py ├── obj2seq.py ├── ops │ ├── functions │ │ ├── __init__.py │ │ └── ms_deform_attn_func.py │ ├── make.sh │ ├── modules │ │ ├── __init__.py │ │ └── ms_deform_attn.py │ ├── setup.py │ ├── src │ │ ├── cpu │ │ │ ├── ms_deform_attn_cpu.cpp │ │ │ └── ms_deform_attn_cpu.h │ │ ├── cuda │ │ │ ├── ms_deform_attn_cuda.cu │ │ │ ├── ms_deform_attn_cuda.h │ │ │ └── ms_deform_im2col_cuda.cuh │ │ ├── ms_deform_attn.h │ │ └── vision.cpp │ └── test.py ├── post_processor.py ├── predictors │ ├── __init__.py │ ├── classifiers │ │ ├── __init__.py │ │ ├── detr_multi_classifier.py │ │ └── label_classifier.py │ ├── separate_detect_head.py │ ├── separate_keypoint_head.py │ ├── seq_postprocess.py │ └── unified_seq_head.py └── transformer │ ├── __init__.py │ ├── attention_modules.py │ ├── class_criterion.py │ ├── encoder.py │ ├── object_decoder.py │ ├── position_encoding.py │ ├── prompt_indicator.py │ └── transformer.py ├── requirements.txt ├── scripts ├── convert_deformable_detr_weight.py ├── debug.sh ├── draw_kps.py ├── dump_clip_features.py ├── run.sh └── run_slurm.sh ├── util ├── __init__.py ├── box_ops.py ├── checkpoint.py ├── collate_fn.py ├── misc.py ├── plot_utils.py └── task_category.py └── word_arrays ├── coco_clip_v2.npy └── person_clip_v2.npy /.github/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/.github/pipeline.png -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/.gitignore -------------------------------------------------------------------------------- /GET_STARTED.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/GET_STARTED.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/README.md -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/config.py -------------------------------------------------------------------------------- /configs/deformable_detr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/configs/deformable_detr.yaml -------------------------------------------------------------------------------- /configs/detection_r50_seqhead.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/configs/detection_r50_seqhead.yaml -------------------------------------------------------------------------------- /configs/detection_r50_seqhead_plus_box_refine.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/configs/detection_r50_seqhead_plus_box_refine.yaml -------------------------------------------------------------------------------- /configs/keypoint_r50_baseline_50e.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/configs/keypoint_r50_baseline_50e.yaml -------------------------------------------------------------------------------- /configs/keypoint_r50_seqhead_150e.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/configs/keypoint_r50_seqhead_150e.yaml -------------------------------------------------------------------------------- /configs/keypoint_r50_seqhead_50e.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/configs/keypoint_r50_seqhead_50e.yaml -------------------------------------------------------------------------------- /configs/tasks/coco_detection.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/configs/tasks/coco_detection.json -------------------------------------------------------------------------------- /configs/tasks/coco_hybrid.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/configs/tasks/coco_hybrid.json -------------------------------------------------------------------------------- /configs/tasks/coco_pose_bce.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/configs/tasks/coco_pose_bce.json -------------------------------------------------------------------------------- /configs/tasks/coco_pose_focal.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/configs/tasks/coco_pose_focal.json -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/datasets/coco.py -------------------------------------------------------------------------------- /datasets/coco_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/datasets/coco_eval.py -------------------------------------------------------------------------------- /datasets/coco_hybrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/datasets/coco_hybrid.py -------------------------------------------------------------------------------- /datasets/coco_mtl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/datasets/coco_mtl.py -------------------------------------------------------------------------------- /datasets/coco_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/datasets/coco_panoptic.py -------------------------------------------------------------------------------- /datasets/data_prefetcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/datasets/data_prefetcher.py -------------------------------------------------------------------------------- /datasets/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/datasets/dataloader.py -------------------------------------------------------------------------------- /datasets/imnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/datasets/imnet.py -------------------------------------------------------------------------------- /datasets/panoptic_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/datasets/panoptic_eval.py -------------------------------------------------------------------------------- /datasets/reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/datasets/reader.py -------------------------------------------------------------------------------- /datasets/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/datasets/samplers.py -------------------------------------------------------------------------------- /datasets/torchvision_datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/datasets/torchvision_datasets/__init__.py -------------------------------------------------------------------------------- /datasets/torchvision_datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/datasets/torchvision_datasets/coco.py -------------------------------------------------------------------------------- /datasets/transform_pipelines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/datasets/transform_pipelines.py -------------------------------------------------------------------------------- /datasets/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/datasets/transforms.py -------------------------------------------------------------------------------- /datasets/voc_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/datasets/voc_map.py -------------------------------------------------------------------------------- /engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/engine.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/main.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/backbone/__init__.py -------------------------------------------------------------------------------- /models/backbone/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/backbone/resnet.py -------------------------------------------------------------------------------- /models/backbone/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/backbone/swin_transformer.py -------------------------------------------------------------------------------- /models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/losses/__init__.py -------------------------------------------------------------------------------- /models/losses/asl_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/losses/asl_losses.py -------------------------------------------------------------------------------- /models/losses/classwise_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/losses/classwise_criterion.py -------------------------------------------------------------------------------- /models/losses/e2e_keypoint_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/losses/e2e_keypoint_criterion.py -------------------------------------------------------------------------------- /models/losses/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/losses/losses.py -------------------------------------------------------------------------------- /models/losses/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/losses/matcher.py -------------------------------------------------------------------------------- /models/losses/matcher_kps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/losses/matcher_kps.py -------------------------------------------------------------------------------- /models/losses/set_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/losses/set_criterion.py -------------------------------------------------------------------------------- /models/losses/unified_matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/losses/unified_matcher.py -------------------------------------------------------------------------------- /models/losses/unified_single_class_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/losses/unified_single_class_criterion.py -------------------------------------------------------------------------------- /models/obj2seq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/obj2seq.py -------------------------------------------------------------------------------- /models/ops/functions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/ops/functions/__init__.py -------------------------------------------------------------------------------- /models/ops/functions/ms_deform_attn_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/ops/functions/ms_deform_attn_func.py -------------------------------------------------------------------------------- /models/ops/make.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/ops/make.sh -------------------------------------------------------------------------------- /models/ops/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/ops/modules/__init__.py -------------------------------------------------------------------------------- /models/ops/modules/ms_deform_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/ops/modules/ms_deform_attn.py -------------------------------------------------------------------------------- /models/ops/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/ops/setup.py -------------------------------------------------------------------------------- /models/ops/src/cpu/ms_deform_attn_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/ops/src/cpu/ms_deform_attn_cpu.cpp -------------------------------------------------------------------------------- /models/ops/src/cpu/ms_deform_attn_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/ops/src/cpu/ms_deform_attn_cpu.h -------------------------------------------------------------------------------- /models/ops/src/cuda/ms_deform_attn_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/ops/src/cuda/ms_deform_attn_cuda.cu -------------------------------------------------------------------------------- /models/ops/src/cuda/ms_deform_attn_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/ops/src/cuda/ms_deform_attn_cuda.h -------------------------------------------------------------------------------- /models/ops/src/cuda/ms_deform_im2col_cuda.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/ops/src/cuda/ms_deform_im2col_cuda.cuh -------------------------------------------------------------------------------- /models/ops/src/ms_deform_attn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/ops/src/ms_deform_attn.h -------------------------------------------------------------------------------- /models/ops/src/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/ops/src/vision.cpp -------------------------------------------------------------------------------- /models/ops/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/ops/test.py -------------------------------------------------------------------------------- /models/post_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/post_processor.py -------------------------------------------------------------------------------- /models/predictors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/predictors/__init__.py -------------------------------------------------------------------------------- /models/predictors/classifiers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/predictors/classifiers/__init__.py -------------------------------------------------------------------------------- /models/predictors/classifiers/detr_multi_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/predictors/classifiers/detr_multi_classifier.py -------------------------------------------------------------------------------- /models/predictors/classifiers/label_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/predictors/classifiers/label_classifier.py -------------------------------------------------------------------------------- /models/predictors/separate_detect_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/predictors/separate_detect_head.py -------------------------------------------------------------------------------- /models/predictors/separate_keypoint_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/predictors/separate_keypoint_head.py -------------------------------------------------------------------------------- /models/predictors/seq_postprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/predictors/seq_postprocess.py -------------------------------------------------------------------------------- /models/predictors/unified_seq_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/predictors/unified_seq_head.py -------------------------------------------------------------------------------- /models/transformer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/transformer/__init__.py -------------------------------------------------------------------------------- /models/transformer/attention_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/transformer/attention_modules.py -------------------------------------------------------------------------------- /models/transformer/class_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/transformer/class_criterion.py -------------------------------------------------------------------------------- /models/transformer/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/transformer/encoder.py -------------------------------------------------------------------------------- /models/transformer/object_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/transformer/object_decoder.py -------------------------------------------------------------------------------- /models/transformer/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/transformer/position_encoding.py -------------------------------------------------------------------------------- /models/transformer/prompt_indicator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/transformer/prompt_indicator.py -------------------------------------------------------------------------------- /models/transformer/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/models/transformer/transformer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/convert_deformable_detr_weight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/scripts/convert_deformable_detr_weight.py -------------------------------------------------------------------------------- /scripts/debug.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/scripts/debug.sh -------------------------------------------------------------------------------- /scripts/draw_kps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/scripts/draw_kps.py -------------------------------------------------------------------------------- /scripts/dump_clip_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/scripts/dump_clip_features.py -------------------------------------------------------------------------------- /scripts/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/scripts/run.sh -------------------------------------------------------------------------------- /scripts/run_slurm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/scripts/run_slurm.sh -------------------------------------------------------------------------------- /util/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/util/__init__.py -------------------------------------------------------------------------------- /util/box_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/util/box_ops.py -------------------------------------------------------------------------------- /util/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/util/checkpoint.py -------------------------------------------------------------------------------- /util/collate_fn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/util/collate_fn.py -------------------------------------------------------------------------------- /util/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/util/misc.py -------------------------------------------------------------------------------- /util/plot_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/util/plot_utils.py -------------------------------------------------------------------------------- /util/task_category.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/util/task_category.py -------------------------------------------------------------------------------- /word_arrays/coco_clip_v2.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/word_arrays/coco_clip_v2.npy -------------------------------------------------------------------------------- /word_arrays/person_clip_v2.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CASIA-LMC-Lab/Obj2Seq/HEAD/word_arrays/person_clip_v2.npy --------------------------------------------------------------------------------