这篇文章介绍如何用Python快速实现一个植物识别的app,家里养了几盆多肉还叫不上名字,正好拿来识别一下。实现这样一个app只需要20行左右的代码,先来看下效果:
另外,我也开发了微信小程序版本,大家可以体验一下。
实现该app主要包含两步,前端界面开发和后端植物识别服务,下面来分别介绍一下。
前端的实现方式有很多种,刚刚说的小程序是一种,但对于习惯用Python的开发者来说,我们还是希望能通过Python语言来开发界面。果然真有这样的工具,叫streamlit
,它是专门为数据科学家、机器学习工程师而开发的,使用它的时候不需要关心布局、样式、服务部署等web相关的知识,而就像开发普通Python程序一样,快速构建优美的app。streamlit
的安装也很简单,执行pip install streamlit
命令即可。想入门streamlit
的朋友可以看次条的文章。
接下来,我们把前端界面开发出来,从上面动图可以看到,最核心逻辑为接收用户输入的图片,并将其显示出来。
import streamlit as st
# 设置网站标题
st.title('植物识别')
# 图片选择框
uploaded_file = st.file_uploader('选择一张图片', type=['jpg', 'png'])
if uploaded_file is not None:
# 显示已选的文件
st.image(uploaded_file, caption='已选文件', use_column_width=True)
界面完成后,我们再来考虑植物图片的识别服务,这里我用的是百度AI的服务
执行pip install baidu-aip
命令即可安装百度AI的Python SDK。然后,再去百度AI开放平台注册一个账号获得相应的APP_KEY和SCRET_KEY。编写代码调用植物识别服务
from aip import AipImageClassify
APP_ID = 'xxx' # 换成自己的 APP_ID
API_KEY = 'xxx' # 换成自己的 API_KEY
SECRET_KEY = 'xxx' # 换成自己的 SECRET_KEY
client = AipImageClassify(APP_ID, API_KEY, SECRET_KEY)
""" 植物识别结果 """
res = client.plantDetect(image) # 调用百度api识别植物
最后将返回的结果在app上展示即可,完整代码如下
import streamlit as st
from aip import AipImageClassify
APP_ID = 'xxx'
API_KEY = 'xxx'
SECRET_KEY = 'xxx'
client = AipImageClassify(APP_ID, API_KEY, SECRET_KEY)
# 设置网站标题
st.title('植物识别')
# 图片选择框
uploaded_file = st.file_uploader('选择一张图片', type=['jpg', 'png'])
if uploaded_file is not None:
# 显示已选的文件
st.image(uploaded_file, caption='已选文件', use_column_width=True)
bs = uploaded_file.read()
""" 植物识别结果 """
res = client.plantDetect(bs) # 调用百度api识别植物
res['result'] # 显示输出结果
""" 该植物最有可能是 """, res['result'][0]['name'] # 取预测概率最大的结果
执行streamlit run plant_detect.py
命令启动app,看到有如下输出
You can now view your Streamlit app in your browser.
Local URL: http://localhost:8501
Network URL: http://192.168.1.3:8501
在浏览器访问指定的地址即可。
希望这篇文章的内容能对你有用,接下来我会定期分享这种AI小应用希望大家能够喜欢。