├── .gitignore ├── README.md ├── __init__.py ├── eval.sh ├── experiments └── cfgs │ ├── faster_rcnn_end2end.yml │ └── scene_graph_generation.yml ├── faster_rcnn ├── Language_Model.py ├── MSDN.py ├── MSDN_base.py ├── RPN.py ├── __init__.py ├── datasets │ ├── __init__.py │ └── visual_genome_loader.py ├── fast_rcnn │ ├── __init__.py │ ├── bbox_transform.py │ ├── config.py │ ├── hierarchical_message_passing_structure.py │ ├── mps_base.py │ └── nms_wrapper.py ├── make.sh ├── network.py ├── nms │ ├── .gitignore │ ├── __init__.py │ ├── cpu_nms.pyx │ ├── gpu_nms.hpp │ ├── gpu_nms.pyx │ ├── nms_kernel.cu │ └── py_cpu_nms.py ├── pycocotools │ ├── UPSTREAM_REV │ ├── __init__.py │ ├── _mask.c │ ├── _mask.pyx │ ├── coco.py │ ├── cocoeval.py │ ├── license.txt │ ├── mask.py │ ├── maskApi.c │ └── maskApi.h ├── roi_pooling │ ├── __init__.py │ ├── _ext │ │ ├── __init__.py │ │ └── roi_pooling │ │ │ └── __init__.py │ ├── build.py │ ├── functions │ │ ├── __init__.py │ │ └── roi_pool.py │ ├── modules │ │ ├── __init__.py │ │ ├── roi_pool.py │ │ └── roi_pool_py.py │ └── src │ │ ├── cuda │ │ ├── roi_pooling_kernel.cu │ │ └── roi_pooling_kernel.h │ │ ├── roi_pooling.c │ │ ├── roi_pooling.h │ │ ├── roi_pooling_cuda.c │ │ └── roi_pooling_cuda.h ├── rpn_msr │ ├── __init__.py │ ├── anchor_target_layer.py │ ├── generate.py │ ├── generate_anchors.py │ ├── proposal_layer.py │ ├── proposal_target_layer.py │ └── proposal_target_layer_hdn.py ├── setup.py ├── utils │ ├── .gitignore │ ├── HDN_utils.py │ ├── __init__.py │ ├── bbox.pyx │ ├── blob.py │ ├── boxes_grid.py │ ├── nms.py │ ├── nms.pyx │ └── timer.py └── vgg16.py ├── scripts ├── preprocessing_data.py └── setup_eval.sh ├── tools ├── generate_anchor_scales_ratios.m └── kmeans_anchors.m ├── train_hdn.py └── train_rpn.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/eval.sh -------------------------------------------------------------------------------- /experiments/cfgs/faster_rcnn_end2end.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/experiments/cfgs/faster_rcnn_end2end.yml -------------------------------------------------------------------------------- /experiments/cfgs/scene_graph_generation.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/experiments/cfgs/scene_graph_generation.yml -------------------------------------------------------------------------------- /faster_rcnn/Language_Model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/Language_Model.py -------------------------------------------------------------------------------- /faster_rcnn/MSDN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/MSDN.py -------------------------------------------------------------------------------- /faster_rcnn/MSDN_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/MSDN_base.py -------------------------------------------------------------------------------- /faster_rcnn/RPN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/RPN.py -------------------------------------------------------------------------------- /faster_rcnn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /faster_rcnn/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/datasets/__init__.py -------------------------------------------------------------------------------- /faster_rcnn/datasets/visual_genome_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/datasets/visual_genome_loader.py -------------------------------------------------------------------------------- /faster_rcnn/fast_rcnn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/fast_rcnn/__init__.py -------------------------------------------------------------------------------- /faster_rcnn/fast_rcnn/bbox_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/fast_rcnn/bbox_transform.py -------------------------------------------------------------------------------- /faster_rcnn/fast_rcnn/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/fast_rcnn/config.py -------------------------------------------------------------------------------- /faster_rcnn/fast_rcnn/hierarchical_message_passing_structure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/fast_rcnn/hierarchical_message_passing_structure.py -------------------------------------------------------------------------------- /faster_rcnn/fast_rcnn/mps_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/fast_rcnn/mps_base.py -------------------------------------------------------------------------------- /faster_rcnn/fast_rcnn/nms_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/fast_rcnn/nms_wrapper.py -------------------------------------------------------------------------------- /faster_rcnn/make.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/make.sh -------------------------------------------------------------------------------- /faster_rcnn/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/network.py -------------------------------------------------------------------------------- /faster_rcnn/nms/.gitignore: -------------------------------------------------------------------------------- 1 | *.c 2 | *.cpp 3 | *.so 4 | -------------------------------------------------------------------------------- /faster_rcnn/nms/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /faster_rcnn/nms/cpu_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/nms/cpu_nms.pyx -------------------------------------------------------------------------------- /faster_rcnn/nms/gpu_nms.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/nms/gpu_nms.hpp -------------------------------------------------------------------------------- /faster_rcnn/nms/gpu_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/nms/gpu_nms.pyx -------------------------------------------------------------------------------- /faster_rcnn/nms/nms_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/nms/nms_kernel.cu -------------------------------------------------------------------------------- /faster_rcnn/nms/py_cpu_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/nms/py_cpu_nms.py -------------------------------------------------------------------------------- /faster_rcnn/pycocotools/UPSTREAM_REV: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/pycocotools/UPSTREAM_REV -------------------------------------------------------------------------------- /faster_rcnn/pycocotools/__init__.py: -------------------------------------------------------------------------------- 1 | __author__ = 'tylin' 2 | -------------------------------------------------------------------------------- /faster_rcnn/pycocotools/_mask.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/pycocotools/_mask.c -------------------------------------------------------------------------------- /faster_rcnn/pycocotools/_mask.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/pycocotools/_mask.pyx -------------------------------------------------------------------------------- /faster_rcnn/pycocotools/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/pycocotools/coco.py -------------------------------------------------------------------------------- /faster_rcnn/pycocotools/cocoeval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/pycocotools/cocoeval.py -------------------------------------------------------------------------------- /faster_rcnn/pycocotools/license.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/pycocotools/license.txt -------------------------------------------------------------------------------- /faster_rcnn/pycocotools/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/pycocotools/mask.py -------------------------------------------------------------------------------- /faster_rcnn/pycocotools/maskApi.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/pycocotools/maskApi.c -------------------------------------------------------------------------------- /faster_rcnn/pycocotools/maskApi.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/pycocotools/maskApi.h -------------------------------------------------------------------------------- /faster_rcnn/roi_pooling/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /faster_rcnn/roi_pooling/_ext/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /faster_rcnn/roi_pooling/_ext/roi_pooling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/roi_pooling/_ext/roi_pooling/__init__.py -------------------------------------------------------------------------------- /faster_rcnn/roi_pooling/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/roi_pooling/build.py -------------------------------------------------------------------------------- /faster_rcnn/roi_pooling/functions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /faster_rcnn/roi_pooling/functions/roi_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/roi_pooling/functions/roi_pool.py -------------------------------------------------------------------------------- /faster_rcnn/roi_pooling/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /faster_rcnn/roi_pooling/modules/roi_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/roi_pooling/modules/roi_pool.py -------------------------------------------------------------------------------- /faster_rcnn/roi_pooling/modules/roi_pool_py.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/roi_pooling/modules/roi_pool_py.py -------------------------------------------------------------------------------- /faster_rcnn/roi_pooling/src/cuda/roi_pooling_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/roi_pooling/src/cuda/roi_pooling_kernel.cu -------------------------------------------------------------------------------- /faster_rcnn/roi_pooling/src/cuda/roi_pooling_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/roi_pooling/src/cuda/roi_pooling_kernel.h -------------------------------------------------------------------------------- /faster_rcnn/roi_pooling/src/roi_pooling.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/roi_pooling/src/roi_pooling.c -------------------------------------------------------------------------------- /faster_rcnn/roi_pooling/src/roi_pooling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/roi_pooling/src/roi_pooling.h -------------------------------------------------------------------------------- /faster_rcnn/roi_pooling/src/roi_pooling_cuda.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/roi_pooling/src/roi_pooling_cuda.c -------------------------------------------------------------------------------- /faster_rcnn/roi_pooling/src/roi_pooling_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/roi_pooling/src/roi_pooling_cuda.h -------------------------------------------------------------------------------- /faster_rcnn/rpn_msr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/rpn_msr/__init__.py -------------------------------------------------------------------------------- /faster_rcnn/rpn_msr/anchor_target_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/rpn_msr/anchor_target_layer.py -------------------------------------------------------------------------------- /faster_rcnn/rpn_msr/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/rpn_msr/generate.py -------------------------------------------------------------------------------- /faster_rcnn/rpn_msr/generate_anchors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/rpn_msr/generate_anchors.py -------------------------------------------------------------------------------- /faster_rcnn/rpn_msr/proposal_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/rpn_msr/proposal_layer.py -------------------------------------------------------------------------------- /faster_rcnn/rpn_msr/proposal_target_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/rpn_msr/proposal_target_layer.py -------------------------------------------------------------------------------- /faster_rcnn/rpn_msr/proposal_target_layer_hdn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/rpn_msr/proposal_target_layer_hdn.py -------------------------------------------------------------------------------- /faster_rcnn/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/setup.py -------------------------------------------------------------------------------- /faster_rcnn/utils/.gitignore: -------------------------------------------------------------------------------- 1 | *.c 2 | *.cpp 3 | *.so 4 | -------------------------------------------------------------------------------- /faster_rcnn/utils/HDN_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/utils/HDN_utils.py -------------------------------------------------------------------------------- /faster_rcnn/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/utils/__init__.py -------------------------------------------------------------------------------- /faster_rcnn/utils/bbox.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/utils/bbox.pyx -------------------------------------------------------------------------------- /faster_rcnn/utils/blob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/utils/blob.py -------------------------------------------------------------------------------- /faster_rcnn/utils/boxes_grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/utils/boxes_grid.py -------------------------------------------------------------------------------- /faster_rcnn/utils/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/utils/nms.py -------------------------------------------------------------------------------- /faster_rcnn/utils/nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/utils/nms.pyx -------------------------------------------------------------------------------- /faster_rcnn/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/utils/timer.py -------------------------------------------------------------------------------- /faster_rcnn/vgg16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/faster_rcnn/vgg16.py -------------------------------------------------------------------------------- /scripts/preprocessing_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/scripts/preprocessing_data.py -------------------------------------------------------------------------------- /scripts/setup_eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/scripts/setup_eval.sh -------------------------------------------------------------------------------- /tools/generate_anchor_scales_ratios.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/tools/generate_anchor_scales_ratios.m -------------------------------------------------------------------------------- /tools/kmeans_anchors.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/tools/kmeans_anchors.m -------------------------------------------------------------------------------- /train_hdn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/train_hdn.py -------------------------------------------------------------------------------- /train_rpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yikang-li/MSDN/HEAD/train_rpn.py --------------------------------------------------------------------------------