├── .gitignore ├── LICENSE ├── README.md ├── detecting ├── __init__.py ├── build │ ├── __init__.py │ └── fasterrcnn.py ├── config │ ├── __init__.py │ └── defaults.py ├── datasets │ ├── __init__.py │ ├── coco.py │ ├── data_generator.py │ ├── datasets.py │ ├── mydata.py │ ├── transforms.py │ ├── utils.py │ └── voc.py ├── loss │ ├── __init__.py │ └── losses.py ├── models │ ├── __init__.py │ ├── anchors │ │ ├── __init__.py │ │ ├── anchor_generator.py │ │ └── anchor_target.py │ ├── backbones │ │ ├── __init__.py │ │ ├── get_backbone.py │ │ └── resnet_v1.py │ ├── bbox_heads │ │ ├── __init__.py │ │ ├── bbox_head.py │ │ └── bbox_target.py │ ├── detectors │ │ ├── __init__.py │ │ └── faster_rcnn.py │ ├── roi_extractors │ │ ├── __init__.py │ │ └── roi_align.py │ └── rpn_heads │ │ ├── __init__.py │ │ └── rpn_head.py ├── solver │ ├── __init__.py │ ├── lr_schedules.py │ └── optimizer.py └── utils │ ├── __init__.py │ ├── colors.py │ ├── compute_map.py │ ├── eval_coco.py │ ├── iou.py │ ├── logger.py │ ├── misc.py │ ├── model_util.py │ ├── transforms.py │ └── visualize.py ├── setup.py ├── test_images ├── 000000018380.jpg ├── 000000540414.jpg ├── Untitled.ipynb ├── Untitled1.ipynb └── test.jpg └── tutorial ├── 1.下载COCO预训练ResNet模型进行预测 ├── 1.下载COCO预训练模型进行预测.ipynb └── 1.下载COCO预训练模型进行预测.py ├── 2.使用COCO预训练模型对COCO数据集进行预测 ├── 2.使用COCO预训练模型对COCO数据集进行预测.ipynb ├── 2.使用COCO预训练模型对COCO数据集进行预测.py └── test.yml ├── 3.训练VOC数据集 ├── 3.训练VOC数据集.ipynb ├── 3.训练VOC数据集.py └── train.yml ├── 4.训练自己的数据集 ├── 4.训练自己的数据集.ipynb ├── 4.训练自己的数据集.py └── train.yml ├── 5.VOC数据集模型评估 ├── 5.VOC数据集模型评估.ipynb ├── 5.VOC数据集模型评估.py └── eval.yml └── 6.COCO数据集模型评估 ├── 6.COCO数据集模型评估.ipynb ├── 6.COCO数据集模型评估.py └── eval.yml /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/README.md -------------------------------------------------------------------------------- /detecting/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/__init__.py -------------------------------------------------------------------------------- /detecting/build/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /detecting/build/fasterrcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/build/fasterrcnn.py -------------------------------------------------------------------------------- /detecting/config/__init__.py: -------------------------------------------------------------------------------- 1 | from .defaults import _C as cfg -------------------------------------------------------------------------------- /detecting/config/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/config/defaults.py -------------------------------------------------------------------------------- /detecting/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /detecting/datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/datasets/coco.py -------------------------------------------------------------------------------- /detecting/datasets/data_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/datasets/data_generator.py -------------------------------------------------------------------------------- /detecting/datasets/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/datasets/datasets.py -------------------------------------------------------------------------------- /detecting/datasets/mydata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/datasets/mydata.py -------------------------------------------------------------------------------- /detecting/datasets/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/datasets/transforms.py -------------------------------------------------------------------------------- /detecting/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/datasets/utils.py -------------------------------------------------------------------------------- /detecting/datasets/voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/datasets/voc.py -------------------------------------------------------------------------------- /detecting/loss/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /detecting/loss/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/loss/losses.py -------------------------------------------------------------------------------- /detecting/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /detecting/models/anchors/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /detecting/models/anchors/anchor_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/models/anchors/anchor_generator.py -------------------------------------------------------------------------------- /detecting/models/anchors/anchor_target.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/models/anchors/anchor_target.py -------------------------------------------------------------------------------- /detecting/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/models/backbones/__init__.py -------------------------------------------------------------------------------- /detecting/models/backbones/get_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/models/backbones/get_backbone.py -------------------------------------------------------------------------------- /detecting/models/backbones/resnet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/models/backbones/resnet_v1.py -------------------------------------------------------------------------------- /detecting/models/bbox_heads/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /detecting/models/bbox_heads/bbox_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/models/bbox_heads/bbox_head.py -------------------------------------------------------------------------------- /detecting/models/bbox_heads/bbox_target.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/models/bbox_heads/bbox_target.py -------------------------------------------------------------------------------- /detecting/models/detectors/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /detecting/models/detectors/faster_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/models/detectors/faster_rcnn.py -------------------------------------------------------------------------------- /detecting/models/roi_extractors/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /detecting/models/roi_extractors/roi_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/models/roi_extractors/roi_align.py -------------------------------------------------------------------------------- /detecting/models/rpn_heads/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /detecting/models/rpn_heads/rpn_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/models/rpn_heads/rpn_head.py -------------------------------------------------------------------------------- /detecting/solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/solver/__init__.py -------------------------------------------------------------------------------- /detecting/solver/lr_schedules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/solver/lr_schedules.py -------------------------------------------------------------------------------- /detecting/solver/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/solver/optimizer.py -------------------------------------------------------------------------------- /detecting/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /detecting/utils/colors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/utils/colors.py -------------------------------------------------------------------------------- /detecting/utils/compute_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/utils/compute_map.py -------------------------------------------------------------------------------- /detecting/utils/eval_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/utils/eval_coco.py -------------------------------------------------------------------------------- /detecting/utils/iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/utils/iou.py -------------------------------------------------------------------------------- /detecting/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/utils/logger.py -------------------------------------------------------------------------------- /detecting/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/utils/misc.py -------------------------------------------------------------------------------- /detecting/utils/model_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/utils/model_util.py -------------------------------------------------------------------------------- /detecting/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/utils/transforms.py -------------------------------------------------------------------------------- /detecting/utils/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/detecting/utils/visualize.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/setup.py -------------------------------------------------------------------------------- /test_images/000000018380.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/test_images/000000018380.jpg -------------------------------------------------------------------------------- /test_images/000000540414.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/test_images/000000540414.jpg -------------------------------------------------------------------------------- /test_images/Untitled.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/test_images/Untitled.ipynb -------------------------------------------------------------------------------- /test_images/Untitled1.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/test_images/Untitled1.ipynb -------------------------------------------------------------------------------- /test_images/test.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/test_images/test.jpg -------------------------------------------------------------------------------- /tutorial/1.下载COCO预训练ResNet模型进行预测/1.下载COCO预训练模型进行预测.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/tutorial/1.下载COCO预训练ResNet模型进行预测/1.下载COCO预训练模型进行预测.ipynb -------------------------------------------------------------------------------- /tutorial/1.下载COCO预训练ResNet模型进行预测/1.下载COCO预训练模型进行预测.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/tutorial/1.下载COCO预训练ResNet模型进行预测/1.下载COCO预训练模型进行预测.py -------------------------------------------------------------------------------- /tutorial/2.使用COCO预训练模型对COCO数据集进行预测/2.使用COCO预训练模型对COCO数据集进行预测.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/tutorial/2.使用COCO预训练模型对COCO数据集进行预测/2.使用COCO预训练模型对COCO数据集进行预测.ipynb -------------------------------------------------------------------------------- /tutorial/2.使用COCO预训练模型对COCO数据集进行预测/2.使用COCO预训练模型对COCO数据集进行预测.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/tutorial/2.使用COCO预训练模型对COCO数据集进行预测/2.使用COCO预训练模型对COCO数据集进行预测.py -------------------------------------------------------------------------------- /tutorial/2.使用COCO预训练模型对COCO数据集进行预测/test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/tutorial/2.使用COCO预训练模型对COCO数据集进行预测/test.yml -------------------------------------------------------------------------------- /tutorial/3.训练VOC数据集/3.训练VOC数据集.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/tutorial/3.训练VOC数据集/3.训练VOC数据集.ipynb -------------------------------------------------------------------------------- /tutorial/3.训练VOC数据集/3.训练VOC数据集.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/tutorial/3.训练VOC数据集/3.训练VOC数据集.py -------------------------------------------------------------------------------- /tutorial/3.训练VOC数据集/train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/tutorial/3.训练VOC数据集/train.yml -------------------------------------------------------------------------------- /tutorial/4.训练自己的数据集/4.训练自己的数据集.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/tutorial/4.训练自己的数据集/4.训练自己的数据集.ipynb -------------------------------------------------------------------------------- /tutorial/4.训练自己的数据集/4.训练自己的数据集.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/tutorial/4.训练自己的数据集/4.训练自己的数据集.py -------------------------------------------------------------------------------- /tutorial/4.训练自己的数据集/train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/tutorial/4.训练自己的数据集/train.yml -------------------------------------------------------------------------------- /tutorial/5.VOC数据集模型评估/5.VOC数据集模型评估.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/tutorial/5.VOC数据集模型评估/5.VOC数据集模型评估.ipynb -------------------------------------------------------------------------------- /tutorial/5.VOC数据集模型评估/5.VOC数据集模型评估.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/tutorial/5.VOC数据集模型评估/5.VOC数据集模型评估.py -------------------------------------------------------------------------------- /tutorial/5.VOC数据集模型评估/eval.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/tutorial/5.VOC数据集模型评估/eval.yml -------------------------------------------------------------------------------- /tutorial/6.COCO数据集模型评估/6.COCO数据集模型评估.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/tutorial/6.COCO数据集模型评估/6.COCO数据集模型评估.ipynb -------------------------------------------------------------------------------- /tutorial/6.COCO数据集模型评估/6.COCO数据集模型评估.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/tutorial/6.COCO数据集模型评估/6.COCO数据集模型评估.py -------------------------------------------------------------------------------- /tutorial/6.COCO数据集模型评估/eval.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qinbf/Detecting/HEAD/tutorial/6.COCO数据集模型评估/eval.yml --------------------------------------------------------------------------------