HuaTuoAI

HuaTuoAI

基于人工智能的中医图像分类, 本存储库包含一个针对中药的人工智能图像分类系统。该项目的目标是通过输入图像准确识别和分类各种中草药和成分。这个仓库里藏着一个神秘的宝藏——一个专为中药打造的人工智能图像分类系统。就像一位奇幻冒险中的导航者,这个项目的任务是将神秘的图像输入,变幻成准确的中草药和成分分类。让我们一起揭开这个数字世界中的迷雾,解锁植物的秘密,用技术和智能描绘中药的未知领域。

Stars: 83

Visit
 screenshot

HuaTuoAI is an artificial intelligence image classification system specifically designed for traditional Chinese medicine. It utilizes deep learning techniques, such as Convolutional Neural Networks (CNN), to accurately classify Chinese herbs and ingredients based on input images. The project aims to unlock the secrets of plants, depict the unknown realm of Chinese medicine using technology and intelligence, and perpetuate ancient cultural heritage.

README:

HuaTuo AI Banner

华佗AI

《支持中医,永远传承古老文化》

这个仓库里藏着一个神秘的宝藏——一个专为中药打造的人工智能图像分类系统。就像一位奇幻冒险中的导航者,这个项目的任务是将神秘的图像输入,变幻成准确的中草药和成分分类。让我们一起揭开这个数字世界中的迷雾,解锁植物的秘密,用技术和智能描绘中药的未知领域。

Python 3.10 TensorFlow License

个人授权许可证

版权所有 2023至2050

特此授予任何获得华佗AI应用程序(以下简称“软件”)副本的人免费许可,可根据以下条件使用软件:

- 使用者被允许复制、修改、合并、出版发行、散布、再授权和/或销售本软件的副本。
- 在使用、复制、修改和分发软件的副本时,使用者必须在显著位置保留原始许可声明,包括对华佗AI应用程序的适当署名,
  并特别标明原作者的姓名为钟智强。
- 在使用者派生的作品中,如果使用了本软件的代码或借鉴了本软件的思想,使用者必须在相关代码、文档或其他材料中明确
  指出华佗AI应用程序及其对应的贡献,并提供华佗AI应用程序原作者钟智强的适当署名。
- 使用者不得将华佗AI应用程序标记为自己的作品,或以任何方式暗示华佗AI应用程序对派生作品的认可或支持。
- 如果使用者希望在本软件的基础上进行盈利或生产产品,必须获得原作者钟智强的书面许可。
- 华佗AI应用程序提供的是按"原样"提供的,不提供任何明示或暗示的担保或条件,包括但不限于对适销性、特定用途适用性
  和非侵权性的担保或条件。在任何情况下,华佗AI应用程序的作者或版权持有人均不承担因使用或无法使用本软件所引起的
  任何索赔、损害或其他责任。
- 华佗AI应用程序的作者或版权持有人不对因使用、复制、修改、合并、出版发行、散布、再授权和/或销售本软件而产生的
  任何索赔、损害或其他责任承担责任,无论是合同责任、侵权行为或其他原因,即使事先已被告知发生此类损害的可能性。 
  
  以上许可条款和限制适用于使用华佗AI应用程序的全部或部分功能。使用华佗AI应用程序即表示接受本许可证的条款和条件。

功能特点

  • 图像分类:该项目的核心功能是图像分类模型,利用深度学习技术对中草药和成分进行分类。模型使用了一种称为卷积神经网络(Convolutional Neural Network,CNN) 的算法。具体而言,模型通过一系列卷积层、池化层和全连接层来提取图像中的特征,并利用softmax函数将图像分为不同的类别。模型的数学表示如下:

在这个公式中,(f * g)(n) 表示在位置 n 对两个函数 fg 进行卷积操作的结果。符号 * 表示卷积操作, 求和符号 \summ 的范围进行求和计算。该公式在每个 m 值处将 f(m)g(n - m) 相乘,并将它们相加, 以得到卷积结果 (f * g)(n)

  • 预处理和增强:为提高模型的性能,该项目采用了强大的预处理技术来准备训练和推理的输入图像。此外,还使用数据增强方法来扩充数据集,改善模型对中草药和成分的泛化能力。
  • 模型评估:本存储库提供了评估脚本,用于评估图像分类模型的性能。计算准确率、精确率、召回率和F1分数等指标,以衡量分类系统的有效性。
  • 部署:存储库中提供了部署指南和资源,以便在生产环境中部署图像分类模型。这使得训练好的模型可以用于实时推理,并集成到现有的系统或应用程序中。

模型架构

参数
损失函数 二元交叉熵
类别模式 Binary
优化器 RMSProp
深度学习模型 卷积神经网络

揭示准确度

Epoch 1/5
1/1 [==============] - 1s 756ms/step - loss: 4.8716 - accuracy: 0.0000e+00 - val_loss: 4.8392 - val_accuracy: 0.0000e+00
Epoch 2/5
1/1 [==============] - 0s 208ms/step - loss: 4.8392 - accuracy: 0.0000e+00 - val_loss: 4.8164 - val_accuracy: 0.2000
Epoch 3/5
1/1 [==============] - 0s 200ms/step - loss: 4.8164 - accuracy: 0.2000 - val_loss: 4.7990 - val_accuracy: 0.2000
Epoch 4/5
1/1 [==============] - 0s 202ms/step - loss: 4.7990 - accuracy: 0.2000 - val_loss: 4.7838 - val_accuracy: 0.2000
Epoch 5/5
1/1 [==============] - 0s 203ms/step - loss: 4.7838 - accuracy: 0.2000 - val_loss: 4.7683 - val_accuracy: 0.2000

18/18 [============] - 1s 23ms/step - loss: 0.7025 - accuracy: 0.1667 - val_loss: -9.6575 - val_accuracy: 0.2500

1️⃣ 安装依赖

pip3 install -r requirements.txt

如果您遇到以下问题 (MacOS):

ssl.SSLCertVerificationError: [SSL: CERTIFICATE_VERIFY_FAILED] certificate verify failed: 
unable to get local issuer certificate

请执行以下命令:

/Applications/Python\{您的python版本}/Install\ Certificates.command ; exit;

# 目前我的版本是 Python 3.10 所以以下的是我命令:
/Applications/Python\3.10/Install\ Certificates.command ; exit;

为了训练更多数据,请将文件放置在数据集 [data/images] 文件夹中。 为了进行验证,请将您的输入数据添加到[data/input]文件夹中。

当前数据集如下:

HuaTuoAI/
:
:.. data/
:    :.. chinese_medicine.txt
:    :.. images/
:         :.. 金银花/
:             :.. 1.png
:             :.. 2.png
:             :.. 3.png
:         :.. 丁公藤/
:             :.. 1.png
:             :.. 2.png
:             :.. 3.png
:         :.. 罗汉果/
:             :.. 1.png
:             :.. 2.png
:             :.. 3.png
:         :.. 人参片/
:             :.. 1.png
:             :.. 2.png
:             :.. 3.png
:         :.. 绿豆/
:             :.. 1.png
:             :.. 2.png
:             :.. 3.png
:    :.. input 
:         :.. 金银花/
:             :.. 1.png
:         :.. 丁公藤/
:             :.. 1.png
:         :.. 罗汉果/
:             :.. 1.png
:         :.. 人参片/
:             :.. 1.png
:         :.. 绿豆/
:             :.. 1.png

若您对中药完全陌生,无需担心,我已经收集了一份中药清单。请点击此处链接查看。

2️⃣ 运行程序

python3 run.py

3️⃣ 运行测试

python3 -m unittest tests/test.py

演示效果

以下是使用 chinese_medicine.h5 二进制文件进行程序演示的示例,以及输出结果:

Image 2Image 1

🌟 开源项目赞助计划

用捐赠助力发展

感谢您使用本项目!您的支持是开源持续发展的核心动力。
每一份捐赠都将直接用于:
✅ 服务器与基础设施维护
✅ 新功能开发与版本迭代
✅ 文档优化与社区建设

点滴支持皆能汇聚成海,让我们共同打造更强大的开源工具!


🌐 全球捐赠通道

国内用户


🔵 支付宝

🟢 微信支付

国际用户


📌 开发者社交图谱

技术交流

社交互动


Star History Chart

🙌 感谢您成为开源社区的重要一员!
💬 捐赠后欢迎通过社交平台与我联系,您的名字将出现在项目致谢列表!

For Tasks:

Click tags to check more tools for each tasks

For Jobs:

Alternative AI tools for HuaTuoAI

Similar Open Source Tools

For similar tasks

For similar jobs