import tensorflow
import numpy as np
from PIL import Image


def main():
    print("Загрузка модели ...")
    model = tensorflow.keras.applications.MobileNetV2(weights="imagenet")

    img = Image.open("test.jpg").resize((224, 224))
    img_array = np.array(img)
    img_array = np.expand_dims(img_array, axis=0)
    img_array = tensorflow.keras.applications.mobilenet_v2.preprocess_input(img_array)
    predictions = model.predict(img_array)
    decoded_predictions = tensorflow.keras.applications.mobilenet_v2.decode_predictions(
        predictions, top=3
    )[0]
    print("Результаты распознавания: ")
    for i, (imagenet_id, label, score) in enumerate(decoded_predictions):
        print(f"{i+1}. {label}, ({score:.2f})")


if __name__ == "__main__":
    main()