├── LICENSE ├── README.md ├── experiments ├── cfgs │ ├── fpn101.yml │ ├── fpn50.yml │ ├── mobile.yml │ ├── res101-lg.yml │ ├── res101.yml │ ├── res18.yml │ ├── res50.yml │ └── vgg16.yml └── logs │ └── .gitignore ├── lib ├── __init__.py ├── datasets │ ├── __init__.py │ ├── factory.py │ ├── imdb.py │ ├── pascal_voc.py │ └── voc_eval.py ├── make.sh ├── model │ ├── __init__.py │ ├── bbox_transform.py │ ├── bbox_transform_cpu.py │ ├── config.py │ ├── nms_wrapper.py │ ├── test.py │ ├── train_val.py │ └── train_val_ori.py ├── nets │ ├── __init__.py │ ├── fpn.py │ ├── layers_util.py │ ├── network.py │ ├── network_fpn.py │ ├── resnet.py │ └── vgg16.py ├── nms │ ├── __init__.py │ ├── build.py │ ├── nms_test │ │ ├── .gitignore │ │ ├── __init__.py │ │ ├── cpu_nms.pyx │ │ ├── gpu_nms.hpp │ │ ├── gpu_nms.pyx │ │ ├── nms_kernel.cu │ │ └── py_cpu_nms.py │ ├── pth_nms.py │ └── src │ │ ├── cuda │ │ ├── nms_kernel.cu │ │ └── nms_kernel.h │ │ ├── nms.c │ │ ├── nms.h │ │ ├── nms_cuda.c │ │ └── nms_cuda.h ├── roi_data_layer │ ├── __init__.py │ ├── layer.py │ ├── minibatch.py │ └── roidb.py ├── roi_pooling │ ├── __init__.py │ ├── _ext │ │ ├── __init__.py │ │ └── roi_pooling │ │ │ └── __init__.py │ ├── build.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 │ ├── __init__.py │ ├── anchor_target_layer_cpu.py │ ├── anchor_target_layer_gpu.py │ ├── fpn │ │ ├── __init__.py │ │ ├── anchor_target_layer_cpu.py │ │ ├── generate_anchors_global.py │ │ ├── proposal_layer.py │ │ └── proposal_target_layer.py │ ├── generate_anchors.py │ ├── generate_anchors_global.py │ ├── proposal_layer.py │ └── proposal_target_layer.py ├── setup.py └── utils │ ├── __init__.py │ ├── bbox.py │ ├── bbox.pyx │ ├── blob.py │ └── timer.py ├── log ├── test.py └── train.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /experiments/cfgs/fpn101.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/experiments/cfgs/fpn101.yml -------------------------------------------------------------------------------- /experiments/cfgs/fpn50.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/experiments/cfgs/fpn50.yml -------------------------------------------------------------------------------- /experiments/cfgs/mobile.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/experiments/cfgs/mobile.yml -------------------------------------------------------------------------------- /experiments/cfgs/res101-lg.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/experiments/cfgs/res101-lg.yml -------------------------------------------------------------------------------- /experiments/cfgs/res101.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/experiments/cfgs/res101.yml -------------------------------------------------------------------------------- /experiments/cfgs/res18.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/experiments/cfgs/res18.yml -------------------------------------------------------------------------------- /experiments/cfgs/res50.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/experiments/cfgs/res50.yml -------------------------------------------------------------------------------- /experiments/cfgs/vgg16.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/experiments/cfgs/vgg16.yml -------------------------------------------------------------------------------- /experiments/logs/.gitignore: -------------------------------------------------------------------------------- 1 | *.txt.* 2 | -------------------------------------------------------------------------------- /lib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/datasets/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/datasets/factory.py -------------------------------------------------------------------------------- /lib/datasets/imdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/datasets/imdb.py -------------------------------------------------------------------------------- /lib/datasets/pascal_voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/datasets/pascal_voc.py -------------------------------------------------------------------------------- /lib/datasets/voc_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/datasets/voc_eval.py -------------------------------------------------------------------------------- /lib/make.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/make.sh -------------------------------------------------------------------------------- /lib/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/model/bbox_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/model/bbox_transform.py -------------------------------------------------------------------------------- /lib/model/bbox_transform_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/model/bbox_transform_cpu.py -------------------------------------------------------------------------------- /lib/model/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/model/config.py -------------------------------------------------------------------------------- /lib/model/nms_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/model/nms_wrapper.py -------------------------------------------------------------------------------- /lib/model/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/model/test.py -------------------------------------------------------------------------------- /lib/model/train_val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/model/train_val.py -------------------------------------------------------------------------------- /lib/model/train_val_ori.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/model/train_val_ori.py -------------------------------------------------------------------------------- /lib/nets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/nets/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/nets/fpn.py -------------------------------------------------------------------------------- /lib/nets/layers_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/nets/layers_util.py -------------------------------------------------------------------------------- /lib/nets/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/nets/network.py -------------------------------------------------------------------------------- /lib/nets/network_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/nets/network_fpn.py -------------------------------------------------------------------------------- /lib/nets/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/nets/resnet.py -------------------------------------------------------------------------------- /lib/nets/vgg16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/nets/vgg16.py -------------------------------------------------------------------------------- /lib/nms/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/nms/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/nms/build.py -------------------------------------------------------------------------------- /lib/nms/nms_test/.gitignore: -------------------------------------------------------------------------------- 1 | *.c 2 | *.cpp 3 | *.so 4 | -------------------------------------------------------------------------------- /lib/nms/nms_test/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/nms/nms_test/cpu_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/nms/nms_test/cpu_nms.pyx -------------------------------------------------------------------------------- /lib/nms/nms_test/gpu_nms.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/nms/nms_test/gpu_nms.hpp -------------------------------------------------------------------------------- /lib/nms/nms_test/gpu_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/nms/nms_test/gpu_nms.pyx -------------------------------------------------------------------------------- /lib/nms/nms_test/nms_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/nms/nms_test/nms_kernel.cu -------------------------------------------------------------------------------- /lib/nms/nms_test/py_cpu_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/nms/nms_test/py_cpu_nms.py -------------------------------------------------------------------------------- /lib/nms/pth_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/nms/pth_nms.py -------------------------------------------------------------------------------- /lib/nms/src/cuda/nms_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/nms/src/cuda/nms_kernel.cu -------------------------------------------------------------------------------- /lib/nms/src/cuda/nms_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/nms/src/cuda/nms_kernel.h -------------------------------------------------------------------------------- /lib/nms/src/nms.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/nms/src/nms.c -------------------------------------------------------------------------------- /lib/nms/src/nms.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/nms/src/nms.h -------------------------------------------------------------------------------- /lib/nms/src/nms_cuda.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/nms/src/nms_cuda.c -------------------------------------------------------------------------------- /lib/nms/src/nms_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/nms/src/nms_cuda.h -------------------------------------------------------------------------------- /lib/roi_data_layer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/roi_data_layer/__init__.py -------------------------------------------------------------------------------- /lib/roi_data_layer/layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/roi_data_layer/layer.py -------------------------------------------------------------------------------- /lib/roi_data_layer/minibatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/roi_data_layer/minibatch.py -------------------------------------------------------------------------------- /lib/roi_data_layer/roidb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/roi_data_layer/roidb.py -------------------------------------------------------------------------------- /lib/roi_pooling/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/roi_pooling/_ext/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/roi_pooling/_ext/roi_pooling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/roi_pooling/_ext/roi_pooling/__init__.py -------------------------------------------------------------------------------- /lib/roi_pooling/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/roi_pooling/build.py -------------------------------------------------------------------------------- /lib/roi_pooling/roi_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/roi_pooling/roi_pool.py -------------------------------------------------------------------------------- /lib/roi_pooling/roi_pool_py.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/roi_pooling/roi_pool_py.py -------------------------------------------------------------------------------- /lib/roi_pooling/src/cuda/roi_pooling_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/roi_pooling/src/cuda/roi_pooling_kernel.cu -------------------------------------------------------------------------------- /lib/roi_pooling/src/cuda/roi_pooling_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/roi_pooling/src/cuda/roi_pooling_kernel.h -------------------------------------------------------------------------------- /lib/roi_pooling/src/roi_pooling.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/roi_pooling/src/roi_pooling.c -------------------------------------------------------------------------------- /lib/roi_pooling/src/roi_pooling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/roi_pooling/src/roi_pooling.h -------------------------------------------------------------------------------- /lib/roi_pooling/src/roi_pooling_cuda.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/roi_pooling/src/roi_pooling_cuda.c -------------------------------------------------------------------------------- /lib/roi_pooling/src/roi_pooling_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/roi_pooling/src/roi_pooling_cuda.h -------------------------------------------------------------------------------- /lib/rpn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/rpn/anchor_target_layer_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/rpn/anchor_target_layer_cpu.py -------------------------------------------------------------------------------- /lib/rpn/anchor_target_layer_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/rpn/anchor_target_layer_gpu.py -------------------------------------------------------------------------------- /lib/rpn/fpn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/rpn/fpn/anchor_target_layer_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/rpn/fpn/anchor_target_layer_cpu.py -------------------------------------------------------------------------------- /lib/rpn/fpn/generate_anchors_global.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/rpn/fpn/generate_anchors_global.py -------------------------------------------------------------------------------- /lib/rpn/fpn/proposal_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/rpn/fpn/proposal_layer.py -------------------------------------------------------------------------------- /lib/rpn/fpn/proposal_target_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/rpn/fpn/proposal_target_layer.py -------------------------------------------------------------------------------- /lib/rpn/generate_anchors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/rpn/generate_anchors.py -------------------------------------------------------------------------------- /lib/rpn/generate_anchors_global.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/rpn/generate_anchors_global.py -------------------------------------------------------------------------------- /lib/rpn/proposal_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/rpn/proposal_layer.py -------------------------------------------------------------------------------- /lib/rpn/proposal_target_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/rpn/proposal_target_layer.py -------------------------------------------------------------------------------- /lib/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/setup.py -------------------------------------------------------------------------------- /lib/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/utils/bbox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/utils/bbox.py -------------------------------------------------------------------------------- /lib/utils/bbox.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/utils/bbox.pyx -------------------------------------------------------------------------------- /lib/utils/blob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/utils/blob.py -------------------------------------------------------------------------------- /lib/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/lib/utils/timer.py -------------------------------------------------------------------------------- /log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/log -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yingxingde/FasterRCNN-pytorch/HEAD/train.py --------------------------------------------------------------------------------