机器学习工具训练出js版模型,然后安装tfjs-converter转换成tensorflow模型
# Ubuntu 2004
pip install tensorflowjs[wizard]
tensorflowjs_converter --input_format=tfjs_layers_model --output_format=keras_saved_model tfjs-model/model.json keras-model
进行验证(图像分类)
# tensorflow 2.8
from tensorflow import keras
import cv2
import numpy as np
model = keras.models.load_model('keras-saved-model/')
print(model)
def predict(file):
image = cv2.imread(file)
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
image = cv2.resize(
image, (224, 224)).astype('float32')
input_data = np.expand_dims(image, axis=0)
input_mean = 127.5
input_std = 127.5
input_data = (np.float32(input_data) - input_mean) / input_std
print(model.predict(input_data, steps=1))
if __name__ == '__main__':
predict('1.jpg')
predict('2.jpg')
predict('3.jpg')
predict('4.jpg')
命令词识别