README_cn.md 4.7 KB
Newer Older
1 2
# **该项目已被迁移到 https://github.com/PaddlePaddle/PaddleDetection**

W
wangguanzhong 已提交
3 4
[English](README.md) | 简体中文

W
wangguanzhong 已提交
5 6
# PaddleDetection

7
PaddleDetection的目的是为工业界和学术界提供丰富、易用的目标检测模型。不仅性能优越、易于部署,而且能够灵活的满足算法研究的需求。
W
wangguanzhong 已提交
8 9 10 11 12 13 14 15 16 17 18 19

<div align="center">
  <img src="demo/output/000000570688.jpg" />
</div>


## 简介

特性:

- 易部署:

20
  PaddleDetection的模型中使用的核心算子均通过C++或CUDA实现,同时基于PaddlePaddle的高性能推理引擎可以方便地部署在多种硬件平台上。
W
wangguanzhong 已提交
21 22 23

- 高灵活度:

24
  PaddleDetection通过模块化设计来解耦各个组件,基于配置文件可以轻松地搭建各种检测模型。
W
wangguanzhong 已提交
25 26 27

- 高性能:

28
  基于PaddlePaddle框架的高性能内核,在模型训练速度、显存占用上有一定的优势。例如,YOLOv3的训练速度快于其他框架,在Tesla V100 16GB环境下,Mask-RCNN(ResNet50)可以单卡Batch Size可以达到4 (甚至到5)。
W
wangguanzhong 已提交
29 30 31

支持的模型结构:

32 33 34 35 36 37
|                    | ResNet | ResNet-vd <sup>[1](#vd)</sup> | ResNeXt-vd | SENet | MobileNet | DarkNet | VGG |
|--------------------|:------:|------------------------------:|:----------:|:-----:|:---------:|:-------:|:---:|
| Faster R-CNN       | ✓      |                             ✓ | x          | ✓     | ✗         | ✗       | ✗   |
| Faster R-CNN + FPN | ✓      |                             ✓ | ✓          | ✓     | ✗         | ✗       | ✗   |
| Mask R-CNN         | ✓      |                             ✓ | x          | ✓     | ✗         | ✗       | ✗   |
| Mask R-CNN + FPN   | ✓      |                             ✓ | ✓          | ✓     | ✗         | ✗       | ✗   |
38 39 40 41
| Cascade Faster-CNN | ✓      |                             ✓ | ✓          | ✗     | ✗         | ✗       | ✗  |
| Cascade Mask-CNN   | ✓      |                             ✗ | ✗          | ✓     | ✗         | ✗       | ✗   |
| RetinaNet          | ✓      |                             ✗ | ✓          | ✗     | ✗         | ✗       | ✗   |
| YOLOv3             | ✓      |                             ✗ | ✗          | ✗     | ✓         | ✓       | ✗   |
42
| SSD                | ✗      |                             ✗ | ✗          | ✗     | ✓         | ✗       | ✓   |
W
wangguanzhong 已提交
43 44 45 46 47

<a name="vd">[1]</a> [ResNet-vd](https://arxiv.org/pdf/1812.01187) 模型提供了较大的精度提高和较少的性能损失。

扩展特性:

48 49 50 51
- [x] **Synchronized Batch Norm**: 目前在YOLOv3中使用。
- [x] **Group Norm**
- [x] **Modulated Deformable Convolution**
- [x] **Deformable PSRoI Pooling**
W
wangguanzhong 已提交
52 53 54 55

**注意:** Synchronized batch normalization 只能在多GPU环境下使用,不能在CPU环境或者单GPU环境下使用。


56
## 使用教程
W
wangguanzhong 已提交
57

58 59 60 61 62 63 64 65
- [安装说明](docs/INSTALL_cn.md)
- [快速开始](docs/QUICK_STARTED_cn.md)
- [训练、评估及参数说明](docs/GETTING_STARTED_cn.md)
- [数据预处理及自定义数据集](docs/DATA_cn.md)
- [配置模块设计和介绍](docs/CONFIG_cn.md)
- [详细的配置信息和参数说明示例](docs/config_example/)
- [IPython Notebook demo](demo/mask_rcnn_demo.ipynb)
- [迁移学习教程](docs/TRANSFER_LEARNING_cn.md)
W
wangguanzhong 已提交
66

67
## 模型库
W
wangguanzhong 已提交
68

69
- [模型库](docs/MODEL_ZOO_cn.md)
Q
qingqing01 已提交
70
- [人脸检测模型](configs/face_detection/README.md)
71
- [行人检测和车辆检测预训练模型](contrib/README_cn.md)
W
wangguanzhong 已提交
72 73


74 75 76
## 模型压缩
- [量化训练压缩示例](slim/quantization)
- [剪枝压缩示例](slim/prune)
77

78
## 推理部署
W
wangguanzhong 已提交
79

W
wangguanzhong 已提交
80
- [模型导出教程](docs/EXPORT_MODEL.md)
81
- [C++推理部署](inference/README.md)
W
wangguanzhong 已提交
82

83
## Benchmark
W
wangguanzhong 已提交
84

85
- [推理Benchmark](docs/BENCHMARK_INFER_cn.md)
W
wangguanzhong 已提交
86 87 88



89
## 版本更新
W
wangguanzhong 已提交
90

91
### 10/2019
W
wangguanzhong 已提交
92

93 94 95 96 97 98 99
- 增加人脸检测模型BlazeFace、Faceboxes。
- 丰富基于COCO的模型,精度高达51.9%。
- 增加Objects365 2019 Challenge上夺冠的最佳单模型之一CACascade-RCNN。
- 增加行人检测和车辆检测预训练模型。
- 支持FP16训练。
- 增加跨平台的C++推理部署方案。
- 增加模型压缩示例。
W
wangguanzhong 已提交
100 101


102 103 104
### 2/9/2019
- 增加GroupNorm模型。
- 增加CascadeRCNN+Mask模型。
W
wangguanzhong 已提交
105

106 107
#### 5/8/2019
- 增加Modulated Deformable Convolution系列模型。
W
wangguanzhong 已提交
108 109 110 111 112 113

#### 7/22/2019

- 增加检测库中文文档
- 修复R-CNN系列模型训练同时进行评估的问题
- 新增ResNext101-vd + Mask R-CNN + FPN模型
114
- 新增基于VOC数据集的YOLOv3模型
W
wangguanzhong 已提交
115 116 117 118 119

#### 7/3/2019

- 首次发布PaddleDetection检测库和检测模型库
- 模型包括:Faster R-CNN, Mask R-CNN, Faster R-CNN+FPN, Mask
120
  R-CNN+FPN, Cascade-Faster-RCNN+FPN, RetinaNet, YOLOv3, 和SSD.
W
wangguanzhong 已提交
121 122 123 124

## 如何贡献代码

我们非常欢迎你可以为PaddleDetection提供代码,也十分感谢你的反馈。