Python: Advanced Guide to Artificial Intelligence

Pre-trained models

There are several models that are pre-trained and made available. All these models are trained on the COCO dataset and can be used for detecting the objects that are available in the COCO dataset such as humans and cars. These models are also useful for transfer learning for a new task such as traffic sign detection. A table of pre-trained models is shown here with relative speed and mAP on the COCO dataset. Various algorithms are trained with different CNN and are depicted in the names:

Model name	Speed	COCO mAP
ssd_mobilenet_v1_coco	fast	21
ssd_inception_v2_coco	fast	24
rfcn_resnet101_coco	medium	30
faster_rcnn_resnet101_coco	medium	32
faster_rcnn_inception_resnet_v2_atrous_coco ...