【AIGC】全新领域!图像识别如何实现?小白也能轻松学会!

前言

上篇文章我们简单体验了一下OpenAI的使用和调试,今天我们继续带大家来进行练习和更加深入的了解。话不多说,让我们开始吧!

正文

机器学习任务

这次文章让我们来调试 一台能够完成我们需求的"机器 ",通过不断地调试让它进行学习,最终能够完成我们交给它的任务------识别图片上我们想要它识别出来的东西

首先让我们进入Colab ,新建一台笔记本供我们这次学习使用。(怎样开始我们的学习可以参考博主上一篇文章(【AIGC】全新领域!小白Openai初体验 带你轻松了解和运用! - 掘金 (juejin.cn))

随后,让我们安装一个transformers------也就是huggingface的核心库并点击运行:

python 复制代码
!pip install transformers

安装完之后先别急着弄完我们的"机器",先让我们看看支持派发那些任务。

python 复制代码
# pipeline --- 分发任务 --- pipeline(task, model)
# transformers pipeline 哪些任务?
from transformers.pipelines import SUPPORTED_TASKS
SUPPORTED_TASKS

运行完之后我们可以通过下拉下划线看见非常非常多能够支持的任务。继续执行下列代码并运行:

python 复制代码
print(SUPPORTED_TASKS.items()) # dick 字典 0(1)

接下来再让我们看看transformers 默认支持的nlp任务

python 复制代码
# es6 解构
for k, v in SUPPORTED_TASKS.items(): # 相当于块级作用域
  print(k) # 查看默认支持的nlp任务

我们可以看到很多常见需要用到的功能,像是question-answering问答功能,AI就是靠这个来进行训练的;还有我们接下来打算用到的image-classification图像分类功能啦;zero-shot-object-detection定位和识别目标功能啦等等。然后再执行下面的代码让我们能快速完成上面的任务:

python 复制代码
from transformers import pipeline # 让我们能快速完成上面的任务

通过运行代码下载图片识别模型:

python 复制代码
checkpoint = "google/owlvit-base-patch32" # 图片识别模型
detector = pipeline('zero-shot-object-detection',model=checkpoint) # (选择任务,传入模型)
# 然后就会开始下载图片识别模型

执行任务

我们的"机器"调试到这就快完成啦!要想它能够识别东西,我们得先让它能够拿到图片:

python 复制代码
import requests # 引入python的http请求库
from PIL import Image # PIL是python的图片的专业库

url = 'https://unsplash.com/photos/oj0zeY2Ltk4/download?ixid=MnwxMjA3fDB8MXxzZWFyY2h8MTR8fHBpY25pY3xlbnwwfHx8fDE2Nzc0OTE1NDk&force=true&w=640'
# requests.get---发送远程请求,这个过程会花费时间
# python是一门同步语言,会同步请求
# 等待图片下载完成;然后再Image.open;最后打印img
img = Image.open(requests.get(url, stream=True).raw)
# requsets --- 用于网络请求,http请求的方法,get表示明文请求
img

这样我们就拿到了一张"沙滩"的图片。接下来让我们向我们的"机器"传达 我们想要让它完成的任务,也就是识别哪些东西:

python 复制代码
predictions = detector(
    img,
    # 候选的对象
    candidate_labels=["hat","sunglasses","book"] 
    # 查找想要识别的对象
)
predictions

根据运行结果我们可以看到返回了JSON数组 ,这里代表的意思分别是:'score':相似度;'label':标签和'box':起始坐标。

查看结果

但这样似乎只是"机器"能够理解,在我们看来是不够直观的,那让我们再来一点点"小魔法":

python 复制代码
from PIL import ImageDraw # 引入"写模块"
draw = ImageDraw.Draw(img)

for prediction in predictions: # predictions里放着刚刚的JSON数组
  box = prediction["box"]
  label = prediction["label"]
  score = prediction["score"]
  xmin, ymin, xmax, ymax = box.values() 
  # 解构语法,这里是把一个数组里的值解构成四个变量
  draw.rectangle((xmin, ymin, xmax, ymax),outline="red",width=1) 
  # 在识别出来标签物品的始末坐标标出红色方框,宽度为1
  draw.text((xmin, ymin), f"{label}:{round(score, 2)}", fill="red") 
  # 在左上角写上标签名和相似值(保留两位小数),字体为红色

img

怎么样?现在是不是就足够直观了,我们可以直接通过肉眼看见"机器"展示给我们标记 出来的结果,甚至打上了红框 并打上了标签!得到这样的结果就说明我们的任务宣布成功啦~

结语

今天带大家体验的功能相信大家在生活中也能感受到各种各样的运用:像是相机的面部捕捉啦;手机的人脸识别啦;车库门口的车牌识别啦等等等等。AI已经越来越融入我们的生活了,如果喜欢的话还希望点点关注,之后继续给大家带来更多有趣的文章♡!

相关推荐
PetterHillWater3 分钟前
Trae中实现OOP原则工程重构
后端·aigc
程序员X小鹿15 分钟前
字节扣子空间的这个新功能,又原地封神了!打工人从此告别通宵!(附实测体验)
aigc·agent
奕川1 小时前
Spring AI 实战指南:模型集成与调优
后端·aigc
redreamSo3 小时前
AI Daily | AI日报:LinkedIn:90% 应用迁移,弃 Kafka 用 Northguard; 谷歌发布免费AI教育功能,重塑课堂!; 小扎143亿请新贵,Meta员工冰火两重天
程序员·aigc·资讯
降世神童3 小时前
华为云Flexus+DeepSeek征文| 使用华为云CCE容器部署Dify-LLM高可用方案的验证与测试
运维·华为云·aigc
降世神童3 小时前
华为云Flexus+DeepSeek征文| 基于华为云Dify-LLM高可用平台开发运维故障处理智能体
运维·华为云·aigc
知了一笑5 小时前
SpringBoot3集成多款主流大模型
spring boot·后端·openai
楽码5 小时前
终于说清楚!希腊字符如何进入数学或科学场景
openai·编程语言·trae
一眼万年046 小时前
每天都在使用的VS Code Copilot Chat 开源啦!
aigc·ai编程·visual studio code
不大姐姐AI智能体6 小时前
最新Coze(扣子)智能体工作流:1分钟生成10条合规美女热舞视频,流量暴涨300%
aigc