├── .gitignore ├── LICENSE ├── README.md ├── data ├── .gitignore ├── demo │ ├── 000456.jpg │ ├── 000542.jpg │ ├── 001150.jpg │ ├── 001763.jpg │ └── 004545.jpg └── imgs │ ├── gt.png │ └── pred.png ├── experiments ├── cfgs │ ├── res101.yml │ ├── res50.yml │ ├── res50_idn_coco.yml │ ├── res50_idn_pascal_voc.yml │ ├── vgg16.yml │ ├── vgg16_idn_coco.yml │ ├── vgg16_idn_pascal_voc.yml │ └── vgg16_idn_pascal_voc_0712.yml ├── logs │ └── .gitignore └── scripts │ ├── test_faster_rcnn.sh │ ├── test_idnet.sh │ ├── test_nms.sh │ ├── train_faster_rcnn.sh │ └── train_idnet.sh ├── lib ├── Makefile ├── datasets │ ├── VOCdevkit-matlab-wrapper │ │ ├── get_voc_opts.m │ │ ├── voc_eval.m │ │ └── xVOCap.m │ ├── __init__.py │ ├── coco.py │ ├── ds_utils.py │ ├── factory.py │ ├── imdb.py │ ├── pascal_voc.py │ ├── tools │ │ └── mcg_munge.py │ └── voc_eval.py ├── layer_utils │ ├── __init__.py │ ├── anchor_target_layer.py │ ├── generate_anchors.py │ ├── idn_proposal_layer.py │ ├── proposal_layer.py │ ├── proposal_target_layer.py │ ├── proposal_top_layer.py │ └── snippets.py ├── model │ ├── __init__.py │ ├── bbox_transform.py │ ├── config.py │ ├── nms_wrapper.py │ ├── test.py │ └── train_val.py ├── nets │ ├── __init__.py │ ├── network.py │ ├── resnet_v1.py │ └── vgg16.py ├── nms │ ├── .gitignore │ ├── __init__.py │ ├── cpu_nms.c │ ├── cpu_nms.pyx │ ├── gpu_nms.cpp │ ├── gpu_nms.hpp │ ├── gpu_nms.pyx │ ├── nms_kernel.cu │ └── py_cpu_nms.py ├── roi_data_layer │ ├── __init__.py │ ├── layer.py │ ├── minibatch.py │ └── roidb.py ├── setup.py └── utils │ ├── .gitignore │ ├── __init__.py │ ├── bbox.pyx │ ├── blob.py │ ├── boxTools.py │ ├── myutils.py │ ├── tf_utils.py │ ├── timer.py │ └── visualization.py └── tools ├── _init_paths.py ├── reval.py ├── test_net.py └── trainval_net.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/README.md -------------------------------------------------------------------------------- /data/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/data/.gitignore -------------------------------------------------------------------------------- /data/demo/000456.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/data/demo/000456.jpg -------------------------------------------------------------------------------- /data/demo/000542.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/data/demo/000542.jpg -------------------------------------------------------------------------------- /data/demo/001150.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/data/demo/001150.jpg -------------------------------------------------------------------------------- /data/demo/001763.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/data/demo/001763.jpg -------------------------------------------------------------------------------- /data/demo/004545.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/data/demo/004545.jpg -------------------------------------------------------------------------------- /data/imgs/gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/data/imgs/gt.png -------------------------------------------------------------------------------- /data/imgs/pred.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/data/imgs/pred.png -------------------------------------------------------------------------------- /experiments/cfgs/res101.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/experiments/cfgs/res101.yml -------------------------------------------------------------------------------- /experiments/cfgs/res50.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/experiments/cfgs/res50.yml -------------------------------------------------------------------------------- /experiments/cfgs/res50_idn_coco.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/experiments/cfgs/res50_idn_coco.yml -------------------------------------------------------------------------------- /experiments/cfgs/res50_idn_pascal_voc.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/experiments/cfgs/res50_idn_pascal_voc.yml -------------------------------------------------------------------------------- /experiments/cfgs/vgg16.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/experiments/cfgs/vgg16.yml -------------------------------------------------------------------------------- /experiments/cfgs/vgg16_idn_coco.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/experiments/cfgs/vgg16_idn_coco.yml -------------------------------------------------------------------------------- /experiments/cfgs/vgg16_idn_pascal_voc.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/experiments/cfgs/vgg16_idn_pascal_voc.yml -------------------------------------------------------------------------------- /experiments/cfgs/vgg16_idn_pascal_voc_0712.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/experiments/cfgs/vgg16_idn_pascal_voc_0712.yml -------------------------------------------------------------------------------- /experiments/logs/.gitignore: -------------------------------------------------------------------------------- 1 | *.txt.* 2 | -------------------------------------------------------------------------------- /experiments/scripts/test_faster_rcnn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/experiments/scripts/test_faster_rcnn.sh -------------------------------------------------------------------------------- /experiments/scripts/test_idnet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/experiments/scripts/test_idnet.sh -------------------------------------------------------------------------------- /experiments/scripts/test_nms.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/experiments/scripts/test_nms.sh -------------------------------------------------------------------------------- /experiments/scripts/train_faster_rcnn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/experiments/scripts/train_faster_rcnn.sh -------------------------------------------------------------------------------- /experiments/scripts/train_idnet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/experiments/scripts/train_idnet.sh -------------------------------------------------------------------------------- /lib/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/Makefile -------------------------------------------------------------------------------- /lib/datasets/VOCdevkit-matlab-wrapper/get_voc_opts.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/datasets/VOCdevkit-matlab-wrapper/get_voc_opts.m -------------------------------------------------------------------------------- /lib/datasets/VOCdevkit-matlab-wrapper/voc_eval.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/datasets/VOCdevkit-matlab-wrapper/voc_eval.m -------------------------------------------------------------------------------- /lib/datasets/VOCdevkit-matlab-wrapper/xVOCap.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/datasets/VOCdevkit-matlab-wrapper/xVOCap.m -------------------------------------------------------------------------------- /lib/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/datasets/__init__.py -------------------------------------------------------------------------------- /lib/datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/datasets/coco.py -------------------------------------------------------------------------------- /lib/datasets/ds_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/datasets/ds_utils.py -------------------------------------------------------------------------------- /lib/datasets/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/datasets/factory.py -------------------------------------------------------------------------------- /lib/datasets/imdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/datasets/imdb.py -------------------------------------------------------------------------------- /lib/datasets/pascal_voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/datasets/pascal_voc.py -------------------------------------------------------------------------------- /lib/datasets/tools/mcg_munge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/datasets/tools/mcg_munge.py -------------------------------------------------------------------------------- /lib/datasets/voc_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/datasets/voc_eval.py -------------------------------------------------------------------------------- /lib/layer_utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/layer_utils/anchor_target_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/layer_utils/anchor_target_layer.py -------------------------------------------------------------------------------- /lib/layer_utils/generate_anchors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/layer_utils/generate_anchors.py -------------------------------------------------------------------------------- /lib/layer_utils/idn_proposal_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/layer_utils/idn_proposal_layer.py -------------------------------------------------------------------------------- /lib/layer_utils/proposal_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/layer_utils/proposal_layer.py -------------------------------------------------------------------------------- /lib/layer_utils/proposal_target_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/layer_utils/proposal_target_layer.py -------------------------------------------------------------------------------- /lib/layer_utils/proposal_top_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/layer_utils/proposal_top_layer.py -------------------------------------------------------------------------------- /lib/layer_utils/snippets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/layer_utils/snippets.py -------------------------------------------------------------------------------- /lib/model/__init__.py: -------------------------------------------------------------------------------- 1 | from . import config 2 | -------------------------------------------------------------------------------- /lib/model/bbox_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/model/bbox_transform.py -------------------------------------------------------------------------------- /lib/model/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/model/config.py -------------------------------------------------------------------------------- /lib/model/nms_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/model/nms_wrapper.py -------------------------------------------------------------------------------- /lib/model/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/model/test.py -------------------------------------------------------------------------------- /lib/model/train_val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/model/train_val.py -------------------------------------------------------------------------------- /lib/nets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/nets/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/nets/network.py -------------------------------------------------------------------------------- /lib/nets/resnet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/nets/resnet_v1.py -------------------------------------------------------------------------------- /lib/nets/vgg16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/nets/vgg16.py -------------------------------------------------------------------------------- /lib/nms/.gitignore: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/nms/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/nms/cpu_nms.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/nms/cpu_nms.c -------------------------------------------------------------------------------- /lib/nms/cpu_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/nms/cpu_nms.pyx -------------------------------------------------------------------------------- /lib/nms/gpu_nms.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/nms/gpu_nms.cpp -------------------------------------------------------------------------------- /lib/nms/gpu_nms.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/nms/gpu_nms.hpp -------------------------------------------------------------------------------- /lib/nms/gpu_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/nms/gpu_nms.pyx -------------------------------------------------------------------------------- /lib/nms/nms_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/nms/nms_kernel.cu -------------------------------------------------------------------------------- /lib/nms/py_cpu_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/nms/py_cpu_nms.py -------------------------------------------------------------------------------- /lib/roi_data_layer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/roi_data_layer/__init__.py -------------------------------------------------------------------------------- /lib/roi_data_layer/layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/roi_data_layer/layer.py -------------------------------------------------------------------------------- /lib/roi_data_layer/minibatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/roi_data_layer/minibatch.py -------------------------------------------------------------------------------- /lib/roi_data_layer/roidb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/roi_data_layer/roidb.py -------------------------------------------------------------------------------- /lib/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/setup.py -------------------------------------------------------------------------------- /lib/utils/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/utils/.gitignore -------------------------------------------------------------------------------- /lib/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/utils/__init__.py -------------------------------------------------------------------------------- /lib/utils/bbox.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/utils/bbox.pyx -------------------------------------------------------------------------------- /lib/utils/blob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/utils/blob.py -------------------------------------------------------------------------------- /lib/utils/boxTools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/utils/boxTools.py -------------------------------------------------------------------------------- /lib/utils/myutils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/utils/myutils.py -------------------------------------------------------------------------------- /lib/utils/tf_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/utils/tf_utils.py -------------------------------------------------------------------------------- /lib/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/utils/timer.py -------------------------------------------------------------------------------- /lib/utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/lib/utils/visualization.py -------------------------------------------------------------------------------- /tools/_init_paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/tools/_init_paths.py -------------------------------------------------------------------------------- /tools/reval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/tools/reval.py -------------------------------------------------------------------------------- /tools/test_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/tools/test_net.py -------------------------------------------------------------------------------- /tools/trainval_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rllab-snu/IDNet/HEAD/tools/trainval_net.py --------------------------------------------------------------------------------