机器学习工具训练出js版模型,然后安装tfjs-converter转换成tensorflow模型
# Ubuntu 2004pip install tensorflowjs[wizard]tensorflowjs_converter --input_format=tfjs_layers_model --output_format=keras_saved_model tfjs-model/model.json keras-model
进行验证(图像分类)
# tensorflow 2.8from tensorflow import kerasimport cv2import numpy as npmodel = 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.5input_std = 127.5input_data = (np.float32(input_data) - input_mean) / input_stdprint(model.predict(input_data, steps=1))if __name__ == '__main__':predict('1.jpg')predict('2.jpg')predict('3.jpg')predict('4.jpg')
命令词识别
