用Python快速实现一个动物识别的app,只需20行左右的代码,web app效果如下:
下面来看实现代码,前端用streamlit框架,用它就像开发普通Python程序一样,可以快速构建web app。
从上面动图可以看到,前端最核心逻辑为接收用户输入的图片,并显示。代码如下:
# 设置网站标题
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()
界面完成后,我们再来调用动物图片的识别服务,这里我用的是百度AI的服务
要调用该服务,首先需要去百度AI开放平台注册一个账号,并创建图像识别应用以获得相应的APP_KEY和SCRET_KEY。然后执行pip install baidu-aip命令即可安装百度AI的Python SDK。编写代码调用动物识别服务
from aip import AipI