机器学习工具训练出js版模型,然后安装tfjs-converter转换成tensorflow模型

    1. # Ubuntu 2004
    2. pip install tensorflowjs[wizard]
    3. tensorflowjs_converter --input_format=tfjs_layers_model --output_format=keras_saved_model tfjs-model/model.json keras-model

    进行验证(图像分类)

    1. # tensorflow 2.8
    2. from tensorflow import keras
    3. import cv2
    4. import numpy as np
    5. model = keras.models.load_model('keras-saved-model/')
    6. print(model)
    7. def predict(file):
    8. image = cv2.imread(file)
    9. image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
    10. image = cv2.resize(
    11. image, (224, 224)).astype('float32')
    12. input_data = np.expand_dims(image, axis=0)
    13. input_mean = 127.5
    14. input_std = 127.5
    15. input_data = (np.float32(input_data) - input_mean) / input_std
    16. print(model.predict(input_data, steps=1))
    17. if __name__ == '__main__':
    18. predict('1.jpg')
    19. predict('2.jpg')
    20. predict('3.jpg')
    21. predict('4.jpg')

    命令词识别