├── .gitignore ├── LICENSE ├── README.md ├── data ├── .gitignore ├── demo │ ├── 000456.jpg │ ├── 000542.jpg │ ├── 001150.jpg │ ├── 001763.jpg │ └── 004545.jpg ├── imgs │ ├── gt.png │ └── pred.png └── scripts │ └── fetch_faster_rcnn_models.sh ├── docker ├── Dockerfile.cuda-7.5 └── Dockerfile.cuda-8.0 ├── experiments ├── cfgs │ ├── mobile.yml │ ├── res101-lg.yml │ ├── res101.yml │ ├── res50.yml │ └── vgg16.yml ├── logs │ └── .gitignore └── scripts │ ├── convert_vgg16.sh │ ├── test_faster_rcnn.sh │ └── train_faster_rcnn.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 │ ├── 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 │ ├── mobilenet_v1.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 │ ├── timer.py │ └── visualization.py └── tools ├── _init_paths.py ├── convert_from_depre.py ├── demo.py ├── reval.py ├── test_net.py └── trainval_net.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/README.md -------------------------------------------------------------------------------- /data/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/data/.gitignore -------------------------------------------------------------------------------- /data/demo/000456.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/data/demo/000456.jpg -------------------------------------------------------------------------------- /data/demo/000542.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/data/demo/000542.jpg -------------------------------------------------------------------------------- /data/demo/001150.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/data/demo/001150.jpg -------------------------------------------------------------------------------- /data/demo/001763.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/data/demo/001763.jpg -------------------------------------------------------------------------------- /data/demo/004545.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/data/demo/004545.jpg -------------------------------------------------------------------------------- /data/imgs/gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/data/imgs/gt.png -------------------------------------------------------------------------------- /data/imgs/pred.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/data/imgs/pred.png -------------------------------------------------------------------------------- /data/scripts/fetch_faster_rcnn_models.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/data/scripts/fetch_faster_rcnn_models.sh -------------------------------------------------------------------------------- /docker/Dockerfile.cuda-7.5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/docker/Dockerfile.cuda-7.5 -------------------------------------------------------------------------------- /docker/Dockerfile.cuda-8.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/docker/Dockerfile.cuda-8.0 -------------------------------------------------------------------------------- /experiments/cfgs/mobile.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/experiments/cfgs/mobile.yml -------------------------------------------------------------------------------- /experiments/cfgs/res101-lg.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/experiments/cfgs/res101-lg.yml -------------------------------------------------------------------------------- /experiments/cfgs/res101.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/experiments/cfgs/res101.yml -------------------------------------------------------------------------------- /experiments/cfgs/res50.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/experiments/cfgs/res50.yml -------------------------------------------------------------------------------- /experiments/cfgs/vgg16.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/experiments/cfgs/vgg16.yml -------------------------------------------------------------------------------- /experiments/logs/.gitignore: -------------------------------------------------------------------------------- 1 | *.txt.* 2 | -------------------------------------------------------------------------------- /experiments/scripts/convert_vgg16.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/experiments/scripts/convert_vgg16.sh -------------------------------------------------------------------------------- /experiments/scripts/test_faster_rcnn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/experiments/scripts/test_faster_rcnn.sh -------------------------------------------------------------------------------- /experiments/scripts/train_faster_rcnn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/experiments/scripts/train_faster_rcnn.sh -------------------------------------------------------------------------------- /lib/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/Makefile -------------------------------------------------------------------------------- /lib/datasets/VOCdevkit-matlab-wrapper/get_voc_opts.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/datasets/VOCdevkit-matlab-wrapper/get_voc_opts.m -------------------------------------------------------------------------------- /lib/datasets/VOCdevkit-matlab-wrapper/voc_eval.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/datasets/VOCdevkit-matlab-wrapper/voc_eval.m -------------------------------------------------------------------------------- /lib/datasets/VOCdevkit-matlab-wrapper/xVOCap.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/datasets/VOCdevkit-matlab-wrapper/xVOCap.m -------------------------------------------------------------------------------- /lib/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/datasets/__init__.py -------------------------------------------------------------------------------- /lib/datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/datasets/coco.py -------------------------------------------------------------------------------- /lib/datasets/ds_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/datasets/ds_utils.py -------------------------------------------------------------------------------- /lib/datasets/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/datasets/factory.py -------------------------------------------------------------------------------- /lib/datasets/imdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/datasets/imdb.py -------------------------------------------------------------------------------- /lib/datasets/pascal_voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/datasets/pascal_voc.py -------------------------------------------------------------------------------- /lib/datasets/tools/mcg_munge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/datasets/tools/mcg_munge.py -------------------------------------------------------------------------------- /lib/datasets/voc_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/datasets/voc_eval.py -------------------------------------------------------------------------------- /lib/layer_utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/layer_utils/anchor_target_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/layer_utils/anchor_target_layer.py -------------------------------------------------------------------------------- /lib/layer_utils/generate_anchors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/layer_utils/generate_anchors.py -------------------------------------------------------------------------------- /lib/layer_utils/proposal_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/layer_utils/proposal_layer.py -------------------------------------------------------------------------------- /lib/layer_utils/proposal_target_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/layer_utils/proposal_target_layer.py -------------------------------------------------------------------------------- /lib/layer_utils/proposal_top_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/layer_utils/proposal_top_layer.py -------------------------------------------------------------------------------- /lib/layer_utils/snippets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/layer_utils/snippets.py -------------------------------------------------------------------------------- /lib/model/__init__.py: -------------------------------------------------------------------------------- 1 | from . import config 2 | -------------------------------------------------------------------------------- /lib/model/bbox_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/model/bbox_transform.py -------------------------------------------------------------------------------- /lib/model/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/model/config.py -------------------------------------------------------------------------------- /lib/model/nms_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/model/nms_wrapper.py -------------------------------------------------------------------------------- /lib/model/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/model/test.py -------------------------------------------------------------------------------- /lib/model/train_val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/model/train_val.py -------------------------------------------------------------------------------- /lib/nets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/nets/mobilenet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/nets/mobilenet_v1.py -------------------------------------------------------------------------------- /lib/nets/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/nets/network.py -------------------------------------------------------------------------------- /lib/nets/resnet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/nets/resnet_v1.py -------------------------------------------------------------------------------- /lib/nets/vgg16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/nets/vgg16.py -------------------------------------------------------------------------------- /lib/nms/.gitignore: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/nms/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/nms/cpu_nms.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/nms/cpu_nms.c -------------------------------------------------------------------------------- /lib/nms/cpu_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/nms/cpu_nms.pyx -------------------------------------------------------------------------------- /lib/nms/gpu_nms.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/nms/gpu_nms.cpp -------------------------------------------------------------------------------- /lib/nms/gpu_nms.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/nms/gpu_nms.hpp -------------------------------------------------------------------------------- /lib/nms/gpu_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/nms/gpu_nms.pyx -------------------------------------------------------------------------------- /lib/nms/nms_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/nms/nms_kernel.cu -------------------------------------------------------------------------------- /lib/nms/py_cpu_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/nms/py_cpu_nms.py -------------------------------------------------------------------------------- /lib/roi_data_layer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/roi_data_layer/__init__.py -------------------------------------------------------------------------------- /lib/roi_data_layer/layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/roi_data_layer/layer.py -------------------------------------------------------------------------------- /lib/roi_data_layer/minibatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/roi_data_layer/minibatch.py -------------------------------------------------------------------------------- /lib/roi_data_layer/roidb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/roi_data_layer/roidb.py -------------------------------------------------------------------------------- /lib/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/setup.py -------------------------------------------------------------------------------- /lib/utils/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/utils/.gitignore -------------------------------------------------------------------------------- /lib/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/utils/__init__.py -------------------------------------------------------------------------------- /lib/utils/bbox.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/utils/bbox.pyx -------------------------------------------------------------------------------- /lib/utils/blob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/utils/blob.py -------------------------------------------------------------------------------- /lib/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/utils/timer.py -------------------------------------------------------------------------------- /lib/utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/lib/utils/visualization.py -------------------------------------------------------------------------------- /tools/_init_paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/tools/_init_paths.py -------------------------------------------------------------------------------- /tools/convert_from_depre.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/tools/convert_from_depre.py -------------------------------------------------------------------------------- /tools/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/tools/demo.py -------------------------------------------------------------------------------- /tools/reval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/tools/reval.py -------------------------------------------------------------------------------- /tools/test_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/tools/test_net.py -------------------------------------------------------------------------------- /tools/trainval_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/endernewton/tf-faster-rcnn/HEAD/tools/trainval_net.py --------------------------------------------------------------------------------