毛笔字识别

import cv2
import tensorflow as tf

# 加载预训练模型
model = tf.keras.models.load_model('calligraphy_recognition_model.h5')

def preprocess_image(image_path):
    img = cv2.imread(image_path, cv2.IMREAD_GRAYSCALE)
    img = cv2.resize(img, (64, 64))  # 调整图像大小
    img = img / 255.0  # 归一化
    img = img.reshape(1, 64, 64, 1)  # 增加通道维度
    return img

def recognize_calligraphy(image_path):
    processed_img = preprocess_image(image_path)
    predictions = model.predict(processed_img)
    predicted_label = tf.argmax(predictions, axis=1).numpy()[0]
    return predicted_label

# 使用示例
image_path = 'path_to_your_image.jpg'
result = recognize_calligraphy(image_path)
print(f'Recognized character: {result}')