请帮我写一份人工智能识别中药的python代码
时间: 2023-03-19 07:22:20 浏览: 255
好的,以下是一份基本的中药识别 Python 代码:
```
import cv2
import numpy as np
import tensorflow as tf
from PIL import Image
# 加载模型
model = tf.keras.models.load_model('model.h5')
# 定义标签
labels = ['当归', '川芎', '党参', '地黄', '枸杞子', '黄芪', '甘草', '白芍', '白术', '熟地黄', '天麻', '桂枝', '茯苓', '龙眼肉']
# 读取图片
img_path = 'test.jpg'
img = cv2.imread(img_path)
# 图像预处理
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
img = cv2.resize(img, (224, 224))
img = np.expand_dims(img, axis=0)
img = img / 255.0
# 预测
pred = model.predict(img)[0]
idx = np.argmax(pred)
name = labels[idx]
confidence = pred[idx] * 100
# 输出结果
print('预测结果:', name)
print('置信度:%.2f%%' % confidence)
```
请注意,这只是一个基本的代码示例,它可能无法完全识别所有中药,并且可能需要更多的数据和模型优化来实现更准确的识别。此外,您还需要准备一组中药图像和相应的标签来训练您自己的模型。
阅读全文