【BML全功能AI开发平台初体验】两小时搞定目标检测全流程!
作者:互联网
BML全功能AI开发平台初体验】两小时搞定目标检测全流程!
最近发现了一个全功能AI开发平台,叫做BML(Baidu Machine Learning),这是一个面向企业和个人开发者的机器学习集成开发环境,为经典机器学习和深度学习提供了从数据处理、模型训练、模型管理到模型推理的全生命周期管理服务。
BML全功能AI开发平台官网链接:https://ai.baidu.com/bml/
最近找到了一个目标检测数据集,同时也是了解到BML(Baidu Machine Learning)这个平台,就想用这个平台实验一下目标检测任务,于是就开始我们的入坑之旅。具体开发流程(全流程)如下:
一、准备开始
在BML平台上点击"立即使用",即将进行我们的目标检测任务。
二、创建并标注数据集
数据在人工智能项目开发的过程中起着至关重要的作用,数据的好坏通常也决定着最终的模型效果。
本次使用螺丝螺母数据集,不会数据集流程的请点击这里:
【BML全功能AI开发平台初体验】目标检测数据集——智能标注
超详细的教程,智能标注省你百分之70的标注时间!
三、新建一个项目
首先打开BML管理页面,在这个界面中可以看到当前支持的深度学习任务类型,也可在左侧目录选择不同深度学习方向。
在左侧目录中找到“计算机视觉模型”,点击“物体检测”,写个名称加段描述就能新建一个项目了。
点击“新建”后,可以在下方找到刚刚创建的项目:
点击“新建任务”可管理本项目的“基本信息”、“配置任务类型”、“添加数据”、“配置网络”等。
四、导入数据集
将"第一步""创建的数据集导入:
这里我们直接选择数据集点击添加即可。
五、配置网络
在配置网络的过程中,可以选择使用预训练模型百度超大规模数据集-通用检测预训练模型,公开数据集-常规预训练模型。这里我们选择公开数据集-常规预训练模型进行训练,同时网络也分为很多种,这里就不再一一叙述,我们选择SSD进行训练模型:
平台提供了脚本编辑的工具,点击“立即编辑”进行脚本编辑:
一般可以在里面配置一些参数,开发者可以自己修改里面的超参数,这里我们把轮数改为10轮训练,试试效果。
六、配置资源
训练模型需要租用服务器或者用自己的服务器,BML平台直接提供了一个配置好的环境,不需要自己买服务器、自己配置环境了,可以说是非常地方便!
这里根据自己的需求选择即可。
点击“提交训练任务”就可以开始训练啦:
训练过程中可以查看自己的训练日志:
当感觉看log枯燥无味的时候,BML也为你提供了可视化展示:
为大家清晰的展示我们的训练过程~,真的是太棒了!
"训练结束"后还能出一份评估报告:
当你有事外出的时候,训练完成之后也可以给你发送短信,这里自己进行选择。
真的是爱了爱了!
七、模型部署
大家参考我写的另一篇文章,都是通用的换成你现在的模型就行,就不在细讲了,直接展示我部署完成的界面!
【BML全功能AI开发平台初体验】没有服务器也可以实现公有云部署!
至此,公有云部署就完成啦,完全不用自己的服务器,哈哈!
八、模型调用
第一步:点击查看详情
记录下你的调用接口地址,在下方代码部分需要修改API调用接口地址。
第二步:点击控制台,对接口进行赋权
第三步:创建应用
填写你的应用名称、应用归属、公司名称等信息。
创建完成之后会显示你创建的应用信息,也就是对接口进行赋权的应用。
第四步:记录API Key和Secret Key
记录好API Key和Secret Key,下方代码调用的时候需要进行输入授权。
第五步:使用代码调用API接口
在这里我们使用python3调用进行演示。
复制代码到自己本地开始进行调用:
"""
BML 物体检测 调用模型公有云API Python3实现
"""
import json
import base64
import requests
"""
使用 requests 库发送请求
使用 pip(或者 pip3)检查我的 python3 环境是否安装了该库,执行命令
pip freeze | grep requests
若返回值为空,则安装该库
pip install requests
"""
# 目标图片的 本地文件路径,支持jpg/png/bmp格式
IMAGE_FILEPATH = "【您的测试图片地址,例如:./example.jpg】"
# 可选的请求参数
# threshold: 默认值为建议阈值,请在 我的模型-模型效果-完整评估结果-详细评估 查看建议阈值
PARAMS = {"threshold": 0.3}
# 服务详情 中的 接口地址
MODEL_API_URL = "【您的API地址】"
# 调用 API 需要 ACCESS_TOKEN。若已有 ACCESS_TOKEN 则于下方填入该字符串
# 否则,留空 ACCESS_TOKEN,于下方填入 该模型部署的 API_KEY 以及 SECRET_KEY,会自动申请并显示新 ACCESS_TOKEN
ACCESS_TOKEN = "【您的ACESS_TOKEN】"
API_KEY = "【您的API_KEY】"
SECRET_KEY = "【您的SECRET_KEY】"
print("1. 读取目标图片 '{}'".format(IMAGE_FILEPATH))
with open(IMAGE_FILEPATH, 'rb') as f:
base64_data = base64.b64encode(f.read())
base64_str = base64_data.decode('UTF8')
print("将 BASE64 编码后图片的字符串填入 PARAMS 的 'image' 字段")
PARAMS["image"] = base64_str
if not ACCESS_TOKEN:
print("2. ACCESS_TOKEN 为空,调用鉴权接口获取TOKEN")
auth_url = "https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials" "&client_id={}&client_secret={}".format(API_KEY, SECRET_KEY)
auth_resp = requests.get(auth_url)
auth_resp_json = auth_resp.json()
ACCESS_TOKEN = auth_resp_json["access_token"]
print("新 ACCESS_TOKEN: {}".format(ACCESS_TOKEN))
else:
print("2. 使用已有 ACCESS_TOKEN")
print("3. 向模型接口 'MODEL_API_URL' 发送请求")
request_url = "{}?access_token={}".format(MODEL_API_URL, ACCESS_TOKEN)
response = requests.post(url=request_url, json=PARAMS)
response_json = response.json()
response_str = json.dumps(response_json, indent=4, ensure_ascii=False)
print("结果:{}".format(response_str))
我们只需要修改五个地方即可完成调用
- IMAGE_FILEPATH:设置为自己本地的图片
- MODEL_API_URL:为API接口的地址
- ACCESS_TOKEN:留空即可(一定要把他的删除留空,或者是你自己计* 算出了ACCESS_TOKEN,写在这里也行)
- API_KEY:创建应用的时候产生了,复制过来即可
- SECRET_KEY:创建应用的时候产生了,复制过来即可
第六步:运行代码
数据集展示(网上随便找的一个图):
修改完毕之后,运行代码,可以看下方图片,已经运行成功,识别出螺丝数据。
至此,模型的调用也就成功了,大家赶紧自己去试试把
九、温馨提示
模型不用的时候,赶紧停止,毕竟免费额度有限哈哈
十、总结
体验了BML全功能AI开发平台的物体检测全流程,从数据集标注-新建任务-模型选择-模型训练-模型部署-模型调用,总共用了不到两个小时就完成了,而且没有过多的打代码操作,只是修改了一下轮数,100轮感觉有点多,训练了10轮MAP已经达到了99.3,完全够用了,模型部署阶段也是相当的给力,点击两下就好了,模型调用的时候,实例代码已经给出,只需要简单的改改就行了,真的是太方便了,大家也赶紧去试试把!
标签:初体验,AI,模型,调用,BML,ACCESS,TOKEN,API 来源: https://blog.csdn.net/deadline_zs/article/details/118312304