You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: fluid/object_detection/README.md
+95-3Lines changed: 95 additions & 3 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -2,7 +2,99 @@ The minimum PaddlePaddle version needed for the code sample in this directory is
2
2
3
3
---
4
4
5
-
#MobileNet-SSD
5
+
## SSD Object Detection
6
6
7
-
This model built with paddle fluid is still under active development and is not
8
-
the final version. We welcome feedbacks.
7
+
### Introduction
8
+
9
+
[Single Shot MultiBox Detector (SSD)](https://arxiv.org/abs/1512.02325) framework for object detection is based on a feed-forward convolutional network. The early network is a standard convolutional architecture for image classification, such as VGG, ResNet, or MobileNet, which is als called base network. In this tutorial we used [MobileNet](https://arxiv.org/abs/1704.04861).
10
+
11
+
### Data Preparation
12
+
13
+
You can use [PASCAL VOC dataset](http://host.robots.ox.ac.uk/pascal/VOC/) or [MS-COCO dataset](http://cocodataset.org/#download).
14
+
15
+
#### PASCAL VOC Dataset
16
+
17
+
If you want to train model on PASCAL VOC dataset, please download datset at first, skip this step if you already have one.
18
+
19
+
```bash
20
+
cd data/pascalvoc
21
+
./download.sh
22
+
```
23
+
24
+
The command `download.sh` also will create training and testing file lists.
25
+
26
+
#### MS-COCO Dataset
27
+
28
+
If you want to train model on MS-COCO dataset, please download datset at first, skip this step if you already have one.
29
+
30
+
```
31
+
cd data/coco
32
+
./download.sh
33
+
```
34
+
35
+
### Train
36
+
37
+
#### Download the Pre-trained Model.
38
+
39
+
We provide two pre-trained models. The one is MobileNet-v1 SSD trained on COCO dataset, but removed the convolutional predictors for COCO dataset. This model can be used to initialize the models when training other dataset, like PASCAL VOC. Then other pre-trained model is MobileNet v1 trained on ImageNet 2012 dataset, but removed the last weights and bias in Fully-Connected layer.
40
+
41
+
Declaration: the MobileNet-v1 SSD model is converted by [TensorFlow model](https://github.com/tensorflow/models/blob/f87a58cd96d45de73c9a8330a06b2ab56749a7fa/research/object_detection/g3doc/detection_model_zoo.md). The MobileNet v1 model is converted [Caffe](https://github.com/shicai/MobileNet-Caffe).
0 commit comments