├── .gitignore ├── LICENSE ├── README.md ├── choir ├── __init__.py ├── checkpoint │ ├── __init__.py │ ├── c2_model_loading.py │ ├── catalog.py │ └── detection_checkpoint.py ├── config │ ├── __init__.py │ ├── compat.py │ ├── config.py │ └── defaults.py ├── data │ ├── __init__.py │ ├── build.py │ ├── catalog.py │ ├── common.py │ ├── dataset_mapper.py │ ├── datasets │ │ ├── __init__.py │ │ ├── builtin.py │ │ ├── builtin_meta.py │ │ ├── doh.py │ │ ├── hico.py │ │ ├── hico_meta.py │ │ ├── swig.py │ │ └── swig_v1_meta.py │ ├── detection_utils.py │ ├── samplers │ │ ├── __init__.py │ │ └── distributed_sampler.py │ └── transforms │ │ ├── __init__.py │ │ ├── augmentation.py │ │ ├── augmentation_impl.py │ │ └── transform.py ├── engine │ ├── __init__.py │ ├── defaults.py │ ├── hooks.py │ ├── launch.py │ └── train_loop.py ├── evaluation │ ├── HICO_MATLAB_EVAL │ │ ├── VOCevaldet_bboxpair.m │ │ ├── cell_find_string.m │ │ ├── hico_eval_wrapper.m │ │ └── hico_official_eval.m │ ├── __init__.py │ ├── evaluator.py │ ├── fast_eval_api.py │ ├── hico_evaluation.py │ ├── swig_evaluation.py │ └── testing.py ├── layers │ ├── __init__.py │ ├── batch_norm.py │ ├── blocks.py │ ├── csrc │ │ ├── README.md │ │ ├── ROIAlign │ │ │ ├── ROIAlign.h │ │ │ ├── ROIAlign_cpu.cpp │ │ │ └── ROIAlign_cuda.cu │ │ ├── cocoeval │ │ │ ├── cocoeval.cpp │ │ │ └── cocoeval.h │ │ ├── cuda_version.cu │ │ ├── deformable │ │ │ ├── deform_conv.h │ │ │ ├── deform_conv_cuda.cu │ │ │ └── deform_conv_cuda_kernel.cu │ │ └── vision.cpp │ ├── deform_conv.py │ ├── nms.py │ ├── roi_align.py │ ├── shape_spec.py │ └── wrappers.py ├── modeling │ ├── __init__.py │ ├── backbone │ │ ├── __init__.py │ │ ├── backbone.py │ │ ├── build.py │ │ ├── fpn.py │ │ ├── resnet.py │ │ └── swin_transformer.py │ ├── box_regression.py │ ├── matcher.py │ ├── meta_arch │ │ ├── __init__.py │ │ ├── build.py │ │ ├── cascade_hoi_detector.py │ │ └── hoi_detector.py │ ├── proposal_generator │ │ ├── __init__.py │ │ ├── anchor_generator.py │ │ ├── build.py │ │ ├── proposal_utils.py │ │ └── rpn.py │ ├── roi_heads │ │ ├── __init__.py │ │ ├── box_head.py │ │ ├── fast_rcnn.py │ │ ├── poolers.py │ │ └── roi_heads.py │ ├── sampling.py │ ├── test_time_augmentation.py │ └── transformers │ │ ├── __init__.py │ │ ├── build.py │ │ ├── cascade_hoi_transformer.py │ │ ├── hoi_transformer.py │ │ ├── position_encoding.py │ │ ├── set_criterion.py │ │ └── transformer.py ├── solver │ ├── __init__.py │ ├── build.py │ └── lr_scheduler.py ├── structures │ ├── __init__.py │ ├── boxes.py │ ├── image_list.py │ └── instances.py └── utils │ ├── __init__.py │ ├── analysis.py │ ├── box_ops.py │ ├── collect_env.py │ ├── colormap.py │ ├── comm.py │ ├── env.py │ ├── events.py │ ├── logger.py │ ├── memory.py │ ├── misc.py │ ├── registry.py │ ├── serialize.py │ └── visualizer.py ├── configs ├── hico_det │ └── choir_swin.yaml └── swig_hoi │ └── hoir_swin.yaml ├── data ├── assembled_datasets.md └── swig_hoi │ └── put_json_here.txt ├── figures ├── example1.jpg ├── example2.jpg ├── example3.jpg ├── example4.jpg ├── example5.jpg ├── example6.jpg └── teaser.jpg ├── models └── model_zoo.md ├── requirements.txt ├── setup.py └── tools ├── demo.py ├── train_net.py └── visualize_hois.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/README.md -------------------------------------------------------------------------------- /choir/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/__init__.py -------------------------------------------------------------------------------- /choir/checkpoint/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/checkpoint/__init__.py -------------------------------------------------------------------------------- /choir/checkpoint/c2_model_loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/checkpoint/c2_model_loading.py -------------------------------------------------------------------------------- /choir/checkpoint/catalog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/checkpoint/catalog.py -------------------------------------------------------------------------------- /choir/checkpoint/detection_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/checkpoint/detection_checkpoint.py -------------------------------------------------------------------------------- /choir/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/config/__init__.py -------------------------------------------------------------------------------- /choir/config/compat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/config/compat.py -------------------------------------------------------------------------------- /choir/config/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/config/config.py -------------------------------------------------------------------------------- /choir/config/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/config/defaults.py -------------------------------------------------------------------------------- /choir/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/data/__init__.py -------------------------------------------------------------------------------- /choir/data/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/data/build.py -------------------------------------------------------------------------------- /choir/data/catalog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/data/catalog.py -------------------------------------------------------------------------------- /choir/data/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/data/common.py -------------------------------------------------------------------------------- /choir/data/dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/data/dataset_mapper.py -------------------------------------------------------------------------------- /choir/data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/data/datasets/__init__.py -------------------------------------------------------------------------------- /choir/data/datasets/builtin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/data/datasets/builtin.py -------------------------------------------------------------------------------- /choir/data/datasets/builtin_meta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/data/datasets/builtin_meta.py -------------------------------------------------------------------------------- /choir/data/datasets/doh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/data/datasets/doh.py -------------------------------------------------------------------------------- /choir/data/datasets/hico.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/data/datasets/hico.py -------------------------------------------------------------------------------- /choir/data/datasets/hico_meta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/data/datasets/hico_meta.py -------------------------------------------------------------------------------- /choir/data/datasets/swig.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/data/datasets/swig.py -------------------------------------------------------------------------------- /choir/data/datasets/swig_v1_meta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/data/datasets/swig_v1_meta.py -------------------------------------------------------------------------------- /choir/data/detection_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/data/detection_utils.py -------------------------------------------------------------------------------- /choir/data/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/data/samplers/__init__.py -------------------------------------------------------------------------------- /choir/data/samplers/distributed_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/data/samplers/distributed_sampler.py -------------------------------------------------------------------------------- /choir/data/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/data/transforms/__init__.py -------------------------------------------------------------------------------- /choir/data/transforms/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/data/transforms/augmentation.py -------------------------------------------------------------------------------- /choir/data/transforms/augmentation_impl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/data/transforms/augmentation_impl.py -------------------------------------------------------------------------------- /choir/data/transforms/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/data/transforms/transform.py -------------------------------------------------------------------------------- /choir/engine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/engine/__init__.py -------------------------------------------------------------------------------- /choir/engine/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/engine/defaults.py -------------------------------------------------------------------------------- /choir/engine/hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/engine/hooks.py -------------------------------------------------------------------------------- /choir/engine/launch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/engine/launch.py -------------------------------------------------------------------------------- /choir/engine/train_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/engine/train_loop.py -------------------------------------------------------------------------------- /choir/evaluation/HICO_MATLAB_EVAL/VOCevaldet_bboxpair.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/evaluation/HICO_MATLAB_EVAL/VOCevaldet_bboxpair.m -------------------------------------------------------------------------------- /choir/evaluation/HICO_MATLAB_EVAL/cell_find_string.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/evaluation/HICO_MATLAB_EVAL/cell_find_string.m -------------------------------------------------------------------------------- /choir/evaluation/HICO_MATLAB_EVAL/hico_eval_wrapper.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/evaluation/HICO_MATLAB_EVAL/hico_eval_wrapper.m -------------------------------------------------------------------------------- /choir/evaluation/HICO_MATLAB_EVAL/hico_official_eval.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/evaluation/HICO_MATLAB_EVAL/hico_official_eval.m -------------------------------------------------------------------------------- /choir/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/evaluation/__init__.py -------------------------------------------------------------------------------- /choir/evaluation/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/evaluation/evaluator.py -------------------------------------------------------------------------------- /choir/evaluation/fast_eval_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/evaluation/fast_eval_api.py -------------------------------------------------------------------------------- /choir/evaluation/hico_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/evaluation/hico_evaluation.py -------------------------------------------------------------------------------- /choir/evaluation/swig_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/evaluation/swig_evaluation.py -------------------------------------------------------------------------------- /choir/evaluation/testing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/evaluation/testing.py -------------------------------------------------------------------------------- /choir/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/layers/__init__.py -------------------------------------------------------------------------------- /choir/layers/batch_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/layers/batch_norm.py -------------------------------------------------------------------------------- /choir/layers/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/layers/blocks.py -------------------------------------------------------------------------------- /choir/layers/csrc/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/layers/csrc/README.md -------------------------------------------------------------------------------- /choir/layers/csrc/ROIAlign/ROIAlign.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/layers/csrc/ROIAlign/ROIAlign.h -------------------------------------------------------------------------------- /choir/layers/csrc/ROIAlign/ROIAlign_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/layers/csrc/ROIAlign/ROIAlign_cpu.cpp -------------------------------------------------------------------------------- /choir/layers/csrc/ROIAlign/ROIAlign_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/layers/csrc/ROIAlign/ROIAlign_cuda.cu -------------------------------------------------------------------------------- /choir/layers/csrc/cocoeval/cocoeval.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/layers/csrc/cocoeval/cocoeval.cpp -------------------------------------------------------------------------------- /choir/layers/csrc/cocoeval/cocoeval.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/layers/csrc/cocoeval/cocoeval.h -------------------------------------------------------------------------------- /choir/layers/csrc/cuda_version.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/layers/csrc/cuda_version.cu -------------------------------------------------------------------------------- /choir/layers/csrc/deformable/deform_conv.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/layers/csrc/deformable/deform_conv.h -------------------------------------------------------------------------------- /choir/layers/csrc/deformable/deform_conv_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/layers/csrc/deformable/deform_conv_cuda.cu -------------------------------------------------------------------------------- /choir/layers/csrc/deformable/deform_conv_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/layers/csrc/deformable/deform_conv_cuda_kernel.cu -------------------------------------------------------------------------------- /choir/layers/csrc/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/layers/csrc/vision.cpp -------------------------------------------------------------------------------- /choir/layers/deform_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/layers/deform_conv.py -------------------------------------------------------------------------------- /choir/layers/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/layers/nms.py -------------------------------------------------------------------------------- /choir/layers/roi_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/layers/roi_align.py -------------------------------------------------------------------------------- /choir/layers/shape_spec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/layers/shape_spec.py -------------------------------------------------------------------------------- /choir/layers/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/layers/wrappers.py -------------------------------------------------------------------------------- /choir/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/__init__.py -------------------------------------------------------------------------------- /choir/modeling/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/backbone/__init__.py -------------------------------------------------------------------------------- /choir/modeling/backbone/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/backbone/backbone.py -------------------------------------------------------------------------------- /choir/modeling/backbone/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/backbone/build.py -------------------------------------------------------------------------------- /choir/modeling/backbone/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/backbone/fpn.py -------------------------------------------------------------------------------- /choir/modeling/backbone/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/backbone/resnet.py -------------------------------------------------------------------------------- /choir/modeling/backbone/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/backbone/swin_transformer.py -------------------------------------------------------------------------------- /choir/modeling/box_regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/box_regression.py -------------------------------------------------------------------------------- /choir/modeling/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/matcher.py -------------------------------------------------------------------------------- /choir/modeling/meta_arch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/meta_arch/__init__.py -------------------------------------------------------------------------------- /choir/modeling/meta_arch/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/meta_arch/build.py -------------------------------------------------------------------------------- /choir/modeling/meta_arch/cascade_hoi_detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/meta_arch/cascade_hoi_detector.py -------------------------------------------------------------------------------- /choir/modeling/meta_arch/hoi_detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/meta_arch/hoi_detector.py -------------------------------------------------------------------------------- /choir/modeling/proposal_generator/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/proposal_generator/__init__.py -------------------------------------------------------------------------------- /choir/modeling/proposal_generator/anchor_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/proposal_generator/anchor_generator.py -------------------------------------------------------------------------------- /choir/modeling/proposal_generator/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/proposal_generator/build.py -------------------------------------------------------------------------------- /choir/modeling/proposal_generator/proposal_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/proposal_generator/proposal_utils.py -------------------------------------------------------------------------------- /choir/modeling/proposal_generator/rpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/proposal_generator/rpn.py -------------------------------------------------------------------------------- /choir/modeling/roi_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/roi_heads/__init__.py -------------------------------------------------------------------------------- /choir/modeling/roi_heads/box_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/roi_heads/box_head.py -------------------------------------------------------------------------------- /choir/modeling/roi_heads/fast_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/roi_heads/fast_rcnn.py -------------------------------------------------------------------------------- /choir/modeling/roi_heads/poolers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/roi_heads/poolers.py -------------------------------------------------------------------------------- /choir/modeling/roi_heads/roi_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/roi_heads/roi_heads.py -------------------------------------------------------------------------------- /choir/modeling/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/sampling.py -------------------------------------------------------------------------------- /choir/modeling/test_time_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/test_time_augmentation.py -------------------------------------------------------------------------------- /choir/modeling/transformers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/transformers/__init__.py -------------------------------------------------------------------------------- /choir/modeling/transformers/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/transformers/build.py -------------------------------------------------------------------------------- /choir/modeling/transformers/cascade_hoi_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/transformers/cascade_hoi_transformer.py -------------------------------------------------------------------------------- /choir/modeling/transformers/hoi_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/transformers/hoi_transformer.py -------------------------------------------------------------------------------- /choir/modeling/transformers/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/transformers/position_encoding.py -------------------------------------------------------------------------------- /choir/modeling/transformers/set_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/transformers/set_criterion.py -------------------------------------------------------------------------------- /choir/modeling/transformers/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/modeling/transformers/transformer.py -------------------------------------------------------------------------------- /choir/solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/solver/__init__.py -------------------------------------------------------------------------------- /choir/solver/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/solver/build.py -------------------------------------------------------------------------------- /choir/solver/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/solver/lr_scheduler.py -------------------------------------------------------------------------------- /choir/structures/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/structures/__init__.py -------------------------------------------------------------------------------- /choir/structures/boxes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/structures/boxes.py -------------------------------------------------------------------------------- /choir/structures/image_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/structures/image_list.py -------------------------------------------------------------------------------- /choir/structures/instances.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/structures/instances.py -------------------------------------------------------------------------------- /choir/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /choir/utils/analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/utils/analysis.py -------------------------------------------------------------------------------- /choir/utils/box_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/utils/box_ops.py -------------------------------------------------------------------------------- /choir/utils/collect_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/utils/collect_env.py -------------------------------------------------------------------------------- /choir/utils/colormap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/utils/colormap.py -------------------------------------------------------------------------------- /choir/utils/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/utils/comm.py -------------------------------------------------------------------------------- /choir/utils/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/utils/env.py -------------------------------------------------------------------------------- /choir/utils/events.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/utils/events.py -------------------------------------------------------------------------------- /choir/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/utils/logger.py -------------------------------------------------------------------------------- /choir/utils/memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/utils/memory.py -------------------------------------------------------------------------------- /choir/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/utils/misc.py -------------------------------------------------------------------------------- /choir/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/utils/registry.py -------------------------------------------------------------------------------- /choir/utils/serialize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/utils/serialize.py -------------------------------------------------------------------------------- /choir/utils/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/choir/utils/visualizer.py -------------------------------------------------------------------------------- /configs/hico_det/choir_swin.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/configs/hico_det/choir_swin.yaml -------------------------------------------------------------------------------- /configs/swig_hoi/hoir_swin.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/configs/swig_hoi/hoir_swin.yaml -------------------------------------------------------------------------------- /data/assembled_datasets.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/data/assembled_datasets.md -------------------------------------------------------------------------------- /data/swig_hoi/put_json_here.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/data/swig_hoi/put_json_here.txt -------------------------------------------------------------------------------- /figures/example1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/figures/example1.jpg -------------------------------------------------------------------------------- /figures/example2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/figures/example2.jpg -------------------------------------------------------------------------------- /figures/example3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/figures/example3.jpg -------------------------------------------------------------------------------- /figures/example4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/figures/example4.jpg -------------------------------------------------------------------------------- /figures/example5.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/figures/example5.jpg -------------------------------------------------------------------------------- /figures/example6.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/figures/example6.jpg -------------------------------------------------------------------------------- /figures/teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/figures/teaser.jpg -------------------------------------------------------------------------------- /models/model_zoo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/models/model_zoo.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/setup.py -------------------------------------------------------------------------------- /tools/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/tools/demo.py -------------------------------------------------------------------------------- /tools/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/tools/train_net.py -------------------------------------------------------------------------------- /tools/visualize_hois.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/scwangdyd/large_vocabulary_hoi_detection/HEAD/tools/visualize_hois.py --------------------------------------------------------------------------------