1
1
# Tensorflow detection model zoo
2
2
3
3
We provide a collection of detection models pre-trained on the [ COCO
4
- dataset] ( http://mscoco.org ) and the [ Kitti dataset] ( http://www.cvlibs.net/datasets/kitti/ ) .
5
- These models can be useful for
4
+ dataset] ( http://mscoco.org ) , the [ Kitti dataset] ( http://www.cvlibs.net/datasets/kitti/ ) , and the
5
+ [ Open Images dataset] ( https://github.com/openimages/dataset ) . These models can
6
+ be useful for
6
7
out-of-the-box inference if you are interested in categories already in COCO
7
- (e.g., humans, cars, etc). They are also useful for initializing your models when
8
+ (e.g., humans, cars, etc) or in Open Images (e.g.,
9
+ surfboard, jacuzzi, etc). They are also useful for initializing your models when
8
10
training on novel datasets.
9
11
10
12
In the table below, we list each such pre-trained model including:
@@ -18,7 +20,7 @@ In the table below, we list each such pre-trained model including:
18
20
configuration (these timings were performed using an Nvidia
19
21
GeForce GTX TITAN X card) and should be treated more as relative timings in
20
22
many cases.
21
- * detector performance on subset of the COCO validation set.
23
+ * detector performance on subset of the COCO validation set or Open Images test split as measured by the dataset-specific mAP measure .
22
24
Here, higher is better, and we only report bounding box mAP rounded to the
23
25
nearest integer.
24
26
* Output types (currently only ` Boxes ` )
@@ -86,5 +88,14 @@ Model name
86
88
----------------------------------------------------------------------------------------------------------------------------------------------------------------- | :---: | :-------------: | :-----:
87
89
[ faster_rcnn_resnet101_kitti] ( http://download.tensorflow.org/models/object_detection/faster_rcnn_resnet101_kitti_2017_11_08.tar.gz ) | 79 | 87 | Boxes
88
90
91
+ ## Open Images-trained models {#open-images-models}
92
+
93
+ Model name | Speed (ms) | Open Images
[email protected] [ ^ 2 ] | Outputs
94
+ ----------------------------------------------------------------------------------------------------------------------------------------------------------------- | :---: | :-------------: | :-----:
95
+ [ faster_rcnn_inception_resnet_v2_atrous_oid] ( http://download.tensorflow.org/models/object_detection/faster_rcnn_inception_resnet_v2_atrous_oid_2017_11_08.tar.gz ) | 727 | 37 | Boxes
96
+ [ faster_rcnn_inception_resnet_v2_atrous_lowproposals_oid] ( http://download.tensorflow.org/models/object_detection/faster_rcnn_inception_resnet_v2_atrous_lowproposals_oid_2017_11_08.tar.gz ) | 347 | | Boxes
97
+
98
+
89
99
[ ^ 1 ] : See [ MSCOCO evaluation protocol] ( http://cocodataset.org/#detections-eval ) .
100
+ [ ^ 2 ] : This is PASCAL mAP with a slightly different way of true positives computation: see [ Open Images evaluation protocol] ( evaluation_protocols.md#open-images ) .
90
101
0 commit comments