@classmethod
def setUpClass(cls):
    super(ModelCorrectnessTest, cls).setUpClass()
    zip_file = tf.keras.utils.get_file(origin=DATASET_URL, fname='flower_photos.tgz', extract=True)
    cls.dataset_dir = os.path.join(os.path.dirname(zip_file), 'flower_photos')
    mobilenet_dir = tempfile.mkdtemp('tflite-transfer-test')
    mobilenet_keras = tf.keras.applications.MobileNetV2(input_shape=(IMAGE_SIZE, IMAGE_SIZE, 3), include_top=False, weights='imagenet')
    tfv1.keras.experimental.export_saved_model(mobilenet_keras, mobilenet_dir)
    cls.mobilenet_dir = mobilenet_dir
@classmethod
def setUpClass(cls):
    super(ModelCorrectnessTest, cls).setUpClass()
    zip_file = tf.keras.utils.get_file(origin=DATASET_URL, fname='flower_photos.tgz', extract=True)
    cls.dataset_dir = os.path.join(os.path.dirname(zip_file), 'flower_photos')
    mobilenet_dir = tempfile.mkdtemp('tflite-transfer-test')
    mobilenet_keras = tf.keras.applications.MobileNetV2(input_shape=(IMAGE_SIZE, IMAGE_SIZE, 3), include_top=False, weights='imagenet')
    tfv1.keras.experimental.export_saved_model(mobilenet_keras, mobilenet_dir)
    cls.mobilenet_dir = mobilenet_dir