def test_efficientnetlite0_model_with_model_maker_retraining_lib(self):
model = image_classifier.create(self.train_data,
model_spec.efficientnet_lite0_spec(),
epochs=1,
batch_size=1,
shuffle=True,
use_hub_library=False)
self._test_accuracy(model)
self._test_export_to_tflite(model)
def test_efficientnetlite0_model(self):
model = image_classifier.create(self.train_data,
model_spec.efficientnet_lite0_spec(),
epochs=1,
batch_size=1,
shuffle=True)
self._test_accuracy(model)
self._test_export_to_tflite(model)
self._test_export_to_tflite_quantized(model, self.train_data)
self._test_export_to_tflite_with_metadata(
model, expected_json_file='efficientnet_lite0_metadata.json')
def test_efficientnetlite0_model_without_training(self):
model = image_classifier.create(
self.train_data, model_spec.efficientnet_lite0_spec(), do_train=False)
self._test_accuracy(model, threshold=0.0)
self._test_export_to_tflite(model, threshold=0.0)