- Faster_RCNN is a two-stage detector, it consists of feature extraction, proposal, classification and refinement processes. This module is trained on COCO2017 dataset, and can be used for object detection.
## II.Installation
## II.Installation
...
@@ -73,7 +75,7 @@
...
@@ -73,7 +75,7 @@
visualization=True)
visualization=True)
```
```
- 预测API,检测输入图片中的所有目标的位置.
- Detection API, detect positions of all objects in image
- **Parameters**
- **Parameters**
...
@@ -82,22 +84,22 @@
...
@@ -82,22 +84,22 @@
- batch_size (int): the size of batch;
- batch_size (int): the size of batch;
- use_gpu (bool): use GPU or not; **set the CUDA_VISIBLE_DEVICES environment variable first if you are using GPU**
- use_gpu (bool): use GPU or not; **set the CUDA_VISIBLE_DEVICES environment variable first if you are using GPU**
- Faster_RCNN is a two-stage detector, it consists of feature extraction, proposal, classification and refinement processes. This module is trained on COCO2017 dataset, and can be used for object detection.
## II.Installation
## II.Installation
...
@@ -73,7 +73,7 @@
...
@@ -73,7 +73,7 @@
visualization=True)
visualization=True)
```
```
- 预测API,检测输入图片中的所有目标的位置.
- Detection API, detect positions of all objects in image
- **Parameters**
- **Parameters**
...
@@ -82,22 +82,22 @@
...
@@ -82,22 +82,22 @@
- batch_size (int): the size of batch;
- batch_size (int): the size of batch;
- use_gpu (bool): use GPU or not; **set the CUDA_VISIBLE_DEVICES environment variable first if you are using GPU**
- use_gpu (bool): use GPU or not; **set the CUDA_VISIBLE_DEVICES environment variable first if you are using GPU**
- Faster_RCNN is a two-stage detector, it consists of feature extraction, proposal, classification and refinement processes. This module is trained on Baidu Detection Dataset, which contains 170w pictures and 1000w+ boxes, and improve the accuracy on 8 test datasets with average 2.06%. Besides, this module supports to fine-tune model, and can achieve faster convergence and better performance.
## II.Installation
## II.Installation
...
@@ -44,38 +44,38 @@
...
@@ -44,38 +44,38 @@
phase='train')
phase='train')
```
```
- 提取特征,用于迁移学习.
- Extract features, and do transfer learning
- **Parameters**
- **Parameters**
- num\_classes (int): 类别数;<br/>
- num\_classes (int): number of classes;<br/>
- trainable (bool): Parameters是否可训练;<br/>
- trainable (bool): whether parameters trainable or not;<br/>
- pretrained (bool): 是否加载预训练模型;<br/>
- pretrained (bool): whether load pretrained model or not
- Single Shot MultiBox Detector (SSD) 是一种单阶段的目标检测器.与两阶段的检测方法不同,单阶段目标检测并不进行区域推荐,而是直接从特征图回归出目标的边界框和分类概率.SSD 运用了这种单阶段检测的思想,并且对其进行改进:在不同尺度的特征图上检测对应尺度的目标.该PaddleHub Module的基网络为MobileNet-v1模型,在Pascal数据集上预训练得到,目前仅支持预测.
- Single Shot MultiBox Detector (SSD) is a one-stage detector. Different from two-stage detector, SSD frames object detection as a re- gression problem to spatially separated bounding boxes and associated class probabilities. This module is based on MobileNet-v1, trained on Pascal dataset, and can be used for object detection.
## II.Installation
## II.Installation
...
@@ -73,7 +73,7 @@
...
@@ -73,7 +73,7 @@
)
)
```
```
- 预测API,检测输入图片中的所有目标的位置.
- Detection API, detect positions of all objects in image
- **Parameters**
- **Parameters**
...
@@ -90,15 +90,15 @@
...
@@ -90,15 +90,15 @@
- **Return**
- **Return**
- res (list\[dict\]): classication results, each element in the list is dict, key is the label name, and value is the corresponding probability
- res (list\[dict\]): results
- data (list): detection results, each element in the list is dict
- data (list): detection results, each element in the list is dict
- confidence (float): the confidence of the result
- confidence (float): the confidence of the result
- label (str): 标签
- label (str): label
- left (int): the upper left corner x coordinate of the detection box
- left (int): the upper left corner x coordinate of the detection box
- top (int): the upper left corner y coordinate of the detection box
- top (int): the upper left corner y coordinate of the detection box
- right (int): the lower right corner x coordinate of the detection box
- right (int): the lower right corner x coordinate of the detection box
- bottom (int): the lower right corner y coordinate of the detection box
- bottom (int): the lower right corner y coordinate of the detection box
- Single Shot MultiBox Detector (SSD) 是一种单阶段的目标检测器.与两阶段的检测方法不同,单阶段目标检测并不进行区域推荐,而是直接从特征图回归出目标的边界框和分类概率.SSD 运用了这种单阶段检测的思想,并且对其进行改进:在不同尺度的特征图上检测对应尺度的目标.该PaddleHub Module的基网络为VGG16模型,在Pascal数据集上预训练得到,目前仅支持预测.
- Single Shot MultiBox Detector (SSD) is a one-stage detector. Different from two-stage detector, SSD frames object detection as a re- gression problem to spatially separated bounding boxes and associated class probabilities. This module is based on VGG16, trained on COCO2017 dataset, and can be used for object detection.
## II.Installation
## II.Installation
...
@@ -72,7 +73,7 @@
...
@@ -72,7 +73,7 @@
visualization=True)
visualization=True)
```
```
- 预测API,检测输入图片中的所有目标的位置.
- Detection API, detect positions of all objects in image
- **Parameters**
- **Parameters**
...
@@ -81,22 +82,22 @@
...
@@ -81,22 +82,22 @@
- batch_size (int): the size of batch;
- batch_size (int): the size of batch;
- use_gpu (bool): use GPU or not; **set the CUDA_VISIBLE_DEVICES environment variable first if you are using GPU**
- use_gpu (bool): use GPU or not; **set the CUDA_VISIBLE_DEVICES environment variable first if you are using GPU**
- YOLOv3 is a one-stage detector proposed by Joseph Redmon and Ali Farhadi, which can reach comparable accuracy but twice as fast as traditional methods. This module is based on YOLOv3, trained on COCO2017, and can be used for object detection.
## II.Installation
## II.Installation
...
@@ -72,7 +72,7 @@
...
@@ -72,7 +72,7 @@
visualization=True)
visualization=True)
```
```
- 预测API,检测输入图片中的所有目标的位置.
- Detection API, detect positions of all objects in image
- **Parameters**
- **Parameters**
...
@@ -81,22 +81,22 @@
...
@@ -81,22 +81,22 @@
- batch_size (int): the size of batch;
- batch_size (int): the size of batch;
- use_gpu (bool): use GPU or not; **set the CUDA_VISIBLE_DEVICES environment variable first if you are using GPU**
- use_gpu (bool): use GPU or not; **set the CUDA_VISIBLE_DEVICES environment variable first if you are using GPU**
-YOLOv3 is a one-stage detector proposed by Joseph Redmon and Ali Farhadi, which can reach comparable accuracy but twice as fast as traditional methods. This module is based on YOLOv3, trained on Baidu Pedestrian Dataset, and can be used for pedestrian detection.
## II.Installation
## II.Installation
...
@@ -72,7 +72,7 @@
...
@@ -72,7 +72,7 @@
visualization=True)
visualization=True)
```
```
- 预测API,检测输入图片中的所有行人的位置.
- Detection API, detect positions of all pedestrian in image
- **Parameters**
- **Parameters**
...
@@ -81,7 +81,7 @@
...
@@ -81,7 +81,7 @@
- batch_size (int): the size of batch;
- batch_size (int): the size of batch;
- use_gpu (bool): use GPU or not; **set the CUDA_VISIBLE_DEVICES environment variable first if you are using GPU**
- use_gpu (bool): use GPU or not; **set the CUDA_VISIBLE_DEVICES environment variable first if you are using GPU**
- YOLOv3 is a one-stage detector proposed by Joseph Redmon and Ali Farhadi, which can reach comparable accuracy but twice as fast as traditional methods. This module is based on YOLOv3, trained on Baidu Vehicle Dataset, and can be used for vehicle detection.
## II.Installation
## II.Installation
...
@@ -72,7 +73,7 @@
...
@@ -72,7 +73,7 @@
visualization=True)
visualization=True)
```
```
- 预测API,检测输入图片中的所有车辆的位置.
- Detection API, detect positions of all vehicles in image
- **Parameters**
- **Parameters**
...
@@ -81,22 +82,22 @@
...
@@ -81,22 +82,22 @@
- batch_size (int): the size of batch;
- batch_size (int): the size of batch;
- use_gpu (bool): use GPU or not; **set the CUDA_VISIBLE_DEVICES environment variable first if you are using GPU**
- use_gpu (bool): use GPU or not; **set the CUDA_VISIBLE_DEVICES environment variable first if you are using GPU**
- YOLOv3 is a one-stage detector proposed by Joseph Redmon and Ali Farhadi, which can reach comparable accuracy but twice as fast as traditional methods. This module is based on YOLOv3, trained on Baidu Vehicle Dataset which consists of 170w pictures and 1000w+ boxes, improve the accuracy on 8 test datasets for average 5.36%, and can be used for vehicle detection.
## II.Installation
## II.Installation
...
@@ -43,20 +43,20 @@
...
@@ -43,20 +43,20 @@
get_prediction=False)
get_prediction=False)
```
```
- 提取特征,用于迁移学习.
- Extract features, and do transfer learning
- **Parameters**
- **Parameters**
- trainable(bool): Parameters是否可训练;<br/>
- trainable(bool): whether parameters trainable or not
- pretrained (bool): 是否加载预训练模型;<br/>
- pretrained (bool): whether load pretrained model or not
- YOLOv3 is a one-stage detector proposed by Joseph Redmon and Ali Farhadi, which can reach comparable accuracy but twice as fast as traditional methods. This module is based on YOLOv3, trained on COCO2017, and can be used for object detection.
## II.Installation
## II.Installation
...
@@ -73,7 +73,7 @@
...
@@ -73,7 +73,7 @@
visualization=True)
visualization=True)
```
```
- 预测API,检测输入图片中的所有目标的位置.
- Detection API, detect positions of all objects in image
- **Parameters**
- **Parameters**
...
@@ -82,22 +82,22 @@
...
@@ -82,22 +82,22 @@
- batch_size (int): the size of batch;
- batch_size (int): the size of batch;
- use_gpu (bool): use GPU or not; **set the CUDA_VISIBLE_DEVICES environment variable first if you are using GPU**
- use_gpu (bool): use GPU or not; **set the CUDA_VISIBLE_DEVICES environment variable first if you are using GPU**
- YOLOv3 is a one-stage detector proposed by Joseph Redmon and Ali Farhadi, which can reach comparable accuracy but twice as fast as traditional methods. This module is based on YOLOv3, trained on COCO2017, and can be used for object detection.
## II.Installation
## II.Installation
...
@@ -72,7 +72,7 @@
...
@@ -72,7 +72,7 @@
visualization=True)
visualization=True)
```
```
- 预测API,检测输入图片中的所有目标的位置.
- Detection API, detect positions of all objects in image
- **Parameters**
- **Parameters**
...
@@ -81,22 +81,22 @@
...
@@ -81,22 +81,22 @@
- batch_size (int): the size of batch;
- batch_size (int): the size of batch;
- use_gpu (bool): use GPU or not; **set the CUDA_VISIBLE_DEVICES environment variable first if you are using GPU**
- use_gpu (bool): use GPU or not; **set the CUDA_VISIBLE_DEVICES environment variable first if you are using GPU**
- YOLOv3 is a one-stage detector proposed by Joseph Redmon and Ali Farhadi, which can reach comparable accuracy but twice as fast as traditional methods. This module is based on YOLOv3, trained on COCO2017, and can be used for object detection.
## II.Installation
## II.Installation
...
@@ -72,7 +72,7 @@
...
@@ -72,7 +72,7 @@
visualization=True)
visualization=True)
```
```
- 预测API,检测输入图片中的所有目标的位置.
- Detection API, detect positions of all objects in image
- **Parameters**
- **Parameters**
...
@@ -81,22 +81,22 @@
...
@@ -81,22 +81,22 @@
- batch_size (int): the size of batch;
- batch_size (int): the size of batch;
- use_gpu (bool): use GPU or not; **set the CUDA_VISIBLE_DEVICES environment variable first if you are using GPU**
- use_gpu (bool): use GPU or not; **set the CUDA_VISIBLE_DEVICES environment variable first if you are using GPU**