├── .gitattributes ├── LICENSE ├── README.md ├── data ├── __init__.py ├── dataset.py ├── util.py └── voc_dataset.py ├── demo.ipynb ├── imgs ├── faster-speed.jpg ├── model_all.png └── visdom-fasterrcnn.png ├── misc ├── convert_caffe_pretrain.py ├── demo.jpg └── train_fast.py ├── model ├── __init__.py ├── faster_rcnn.py ├── faster_rcnn_vgg16.py ├── region_proposal_network.py ├── roi_module.py └── utils │ ├── __init__.py │ ├── bbox_tools.py │ ├── creator_tool.py │ ├── nms │ ├── __init__.py │ ├── _nms_gpu_post.pyx │ ├── _nms_gpu_post_py.py │ ├── build.py │ └── non_maximum_suppression.py │ └── roi_cupy.py ├── requirements.txt ├── train.py ├── trainer.py └── utils ├── __init__.py ├── array_tool.py ├── config.py ├── eval_tool.py └── vis_tool.py /.gitattributes: -------------------------------------------------------------------------------- 1 | *.ipynb binary 2 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/README.md -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/data/dataset.py -------------------------------------------------------------------------------- /data/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/data/util.py -------------------------------------------------------------------------------- /data/voc_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/data/voc_dataset.py -------------------------------------------------------------------------------- /demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/demo.ipynb -------------------------------------------------------------------------------- /imgs/faster-speed.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/imgs/faster-speed.jpg -------------------------------------------------------------------------------- /imgs/model_all.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/imgs/model_all.png -------------------------------------------------------------------------------- /imgs/visdom-fasterrcnn.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/imgs/visdom-fasterrcnn.png -------------------------------------------------------------------------------- /misc/convert_caffe_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/misc/convert_caffe_pretrain.py -------------------------------------------------------------------------------- /misc/demo.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/misc/demo.jpg -------------------------------------------------------------------------------- /misc/train_fast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/misc/train_fast.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/faster_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/model/faster_rcnn.py -------------------------------------------------------------------------------- /model/faster_rcnn_vgg16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/model/faster_rcnn_vgg16.py -------------------------------------------------------------------------------- /model/region_proposal_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/model/region_proposal_network.py -------------------------------------------------------------------------------- /model/roi_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/model/roi_module.py -------------------------------------------------------------------------------- /model/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/utils/bbox_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/model/utils/bbox_tools.py -------------------------------------------------------------------------------- /model/utils/creator_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/model/utils/creator_tool.py -------------------------------------------------------------------------------- /model/utils/nms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/model/utils/nms/__init__.py -------------------------------------------------------------------------------- /model/utils/nms/_nms_gpu_post.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/model/utils/nms/_nms_gpu_post.pyx -------------------------------------------------------------------------------- /model/utils/nms/_nms_gpu_post_py.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/model/utils/nms/_nms_gpu_post_py.py -------------------------------------------------------------------------------- /model/utils/nms/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/model/utils/nms/build.py -------------------------------------------------------------------------------- /model/utils/nms/non_maximum_suppression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/model/utils/nms/non_maximum_suppression.py -------------------------------------------------------------------------------- /model/utils/roi_cupy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/model/utils/roi_cupy.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/train.py -------------------------------------------------------------------------------- /trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/trainer.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/array_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/utils/array_tool.py -------------------------------------------------------------------------------- /utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/utils/config.py -------------------------------------------------------------------------------- /utils/eval_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/utils/eval_tool.py -------------------------------------------------------------------------------- /utils/vis_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/simple-faster-rcnn-explain/HEAD/utils/vis_tool.py --------------------------------------------------------------------------------