Skip to content

Commit 2222705

Browse files
committed
add results
1 parent 72b66a9 commit 2222705

14 files changed

+22
-1
lines changed

README.md

+22-1
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ This repository contains codes of the reimplementation of [SSD: Single Shot Mult
44

55
There are already some TensorFlow based SSD reimplementation codes on GitHub, the main special features of this repo inlcudes:
66

7-
- state of the art performance when training from VGG-16 pre-trained model (trained model will be released later).
7+
- state of the art performance(~77%mAP) when training from VGG-16 pre-trained model (SSD300-VGG16).
88
- the model is trained using TensorFlow high level API [tf.estimator](https://www.tensorflow.org/api_docs/python/tf/estimator/Estimator). Although TensorFlow provides many APIs, the Estimator API is highly recommended to yield scalable, high-performance models.
99
- all codes were writen by TensorFlow ops (no numpy operation) to ensure the performance and portability.
1010
- using ssd augmentation pipeline discribed in the original paper.
@@ -55,5 +55,26 @@ All the codes was tested under TensorFlow 1.6, Python 3.5, Ubuntu 16.04 with CUD
5555

5656
This repo is just created recently, any contribution will be welcomed.
5757

58+
## Results (VOC07 Metric)
59+
60+
This implementation(SSD300-VGG16) yield **mAP 76.94%** on PASCAL VOC 2007 test dataset, the details are as follows:
61+
62+
| sofa | bird | pottedplant | bus | diningtable | cow | bottle | horse | aeroplane | motorbike
63+
|:-------|:-----:|:-------:|:-------:|:-------:|:-------:|:-------:|:-------:|:-------:|:-------:|
64+
| 80.6 | 75.3 | 51.8 | 85.1 | 77.1 | 81.7 | 49.3 | 85.5 | 80.1 | 83.9 |
65+
| sheep | train | boat | bicycle | chair | cat | tvmonitor | person | car | dog |
66+
| 79.1 | 86.4 | 70.3 | 82.5 | 61.9 | 87.8 | 73.7 | 78.5 | 82.7 | 85.5 |
67+
68+
You can download the trained model(VOC07+12 Train) from [GoogleDrive](https://drive.google.com/open?id=1yeYcfcOURcZ4DaElEn9C2xY1NymGzG5W) for further research.
69+
70+
Here is the training logs and some detection results:
71+
72+
![](logs/loss.jpg "loss")
73+
![](logs/celoss.jpg "celoss")
74+
![](logs/locloss.jpg "locloss")
75+
![](demo/demo1.jpg "demo1")
76+
![](demo/demo2.jpg "demo2")
77+
![](demo/demo3.jpg "demo3")
78+
5879
## ##
5980
Apache License, Version 2.0

demo/2007_000063.jpg

-123 KB
Binary file not shown.

demo/2007_000068.jpg

-98.8 KB
Binary file not shown.

demo/2007_000121.jpg

-78 KB
Binary file not shown.

demo/2007_000123.jpg

-77.9 KB
Binary file not shown.

demo/2007_000129.jpg

-103 KB
Binary file not shown.

demo/2007_000170.jpg

-76.2 KB
Binary file not shown.

demo/demo1.jpg

26.9 KB
Loading

demo/demo2.jpg

31.2 KB
Loading

demo/demo3.jpg

34.2 KB
Loading

demo/test.jpg

-71.4 KB
Binary file not shown.

logs/celoss.JPG

25.5 KB
Loading

logs/locloss.JPG

25.1 KB
Loading

logs/loss.JPG

25.1 KB
Loading

0 commit comments

Comments
 (0)