ContextQMD
Back to Tensorflow

Module: tflite_model_maker.object_detector

tensorflow/lite/g3doc/api_docs/python/tflite_model_maker/object_detector.md

2.21.02.7 KB
Original Source

page_type: reference description: APIs to train an object detection model.

<link rel="stylesheet" href="/site-assets/css/style.css"> <!-- DO NOT EDIT! Automatically generated file. --> <div itemscope itemtype="http://developers.google.com/ReferenceObject"> <meta itemprop="name" content="tflite_model_maker.object_detector" /> <meta itemprop="path" content="Stable" /> </div>

Module: tflite_model_maker.object_detector

<!-- Insert buttons and diff --> <table class="tfo-notebook-buttons tfo-api nocontent" align="left"> <td> <a target="_blank" href="https://github.com/tensorflow/examples/blob/tflmm/v0.4.2/tensorflow_examples/lite/model_maker/public/object_detector/__init__.py"> 
View source on GitHub
</a> </td> </table>

APIs to train an object detection model.

Classes

class DataLoader: DataLoader for object detector.

class EfficientDetSpec: A specification of the EfficientDet model.

class ObjectDetector: ObjectDetector class for inference and exporting to tflite.

Functions

EfficientDetLite0Spec(...): Creates EfficientDet-Lite0 model spec. See also: <a href="../tflite_model_maker/object_detector/EfficientDetSpec"><code>tflite_model_maker.object_detector.EfficientDetSpec</code></a>.

EfficientDetLite1Spec(...): Creates EfficientDet-Lite1 model spec. See also: <a href="../tflite_model_maker/object_detector/EfficientDetSpec"><code>tflite_model_maker.object_detector.EfficientDetSpec</code></a>.

EfficientDetLite2Spec(...): Creates EfficientDet-Lite2 model spec. See also: <a href="../tflite_model_maker/object_detector/EfficientDetSpec"><code>tflite_model_maker.object_detector.EfficientDetSpec</code></a>.

EfficientDetLite3Spec(...): Creates EfficientDet-Lite3 model spec. See also: <a href="../tflite_model_maker/object_detector/EfficientDetSpec"><code>tflite_model_maker.object_detector.EfficientDetSpec</code></a>.

EfficientDetLite4Spec(...): Creates EfficientDet-Lite4 model spec. See also: <a href="../tflite_model_maker/object_detector/EfficientDetSpec"><code>tflite_model_maker.object_detector.EfficientDetSpec</code></a>.

create(...): Loads data and train the model for object detection.