第10.4篇 使用预训练的目标检测网络

在PyTorch提供的已经训练好的图像目标检测中,均是R-CNN系列

的网络,并且针对目标检测和人体关键点检测分别提供了容易调用的方

法。针对目标检测的网络,输入图像均要求使用相同的预处理方式,即先将每张图像的像素值预处理到0~1之间,且输人的图像尺寸不是很小即可直接

调用。已经预训练的可供使用的网络模型如表10-2所示。

首先定义每个类别所对应的标签COCO_INSTANCE_CATEGORY_NAMES,程序如下:

上面的程序在可视化图像时,使用ImageDraw.Draw(image)方法,表示要在原始的image图像上相应的位置添加一些元素,draw.rectangle(表示要添加矩形框,draw.text()表示在图像上指定位置添加文本。运行程序后,可得到图10-14所示的目标检测结果。

相关推荐
没事勤琢磨32 分钟前
如何让 OpenClaw 控制使用浏览器:让 AI 像真人一样操控你的浏览器
人工智能
用户51914958484536 分钟前
CrushFTP 认证绕过漏洞利用工具 (CVE-2024-4040)
人工智能·aigc
牛马摆渡人5281 小时前
OpenClaw实战--Day1: 本地化
人工智能
前端小豆1 小时前
玩转 OpenClaw:打造你的私有 AI 助手网关
人工智能
BugShare1 小时前
写一个你自己的Agent Skills
人工智能·程序员
机器之心1 小时前
英伟达护城河被AI攻破,字节清华CUDA Agent,让人人能搓CUDA内核
人工智能·openai
后端小肥肠3 小时前
公众号躺更神器!OpenClaw+Claude Skill 实现自动读对标 + 写文 + 配图 + 存入草稿箱
人工智能·aigc·agent
爱可生开源社区3 小时前
SCALE | 重构 AI 时代数据库能力的全新评估标准
人工智能
Jahzo3 小时前
openclaw本地化部署体验与踩坑记录--飞书机器人配置
人工智能·开源
Narrastory3 小时前
明日香 - Pytorch 快速入门保姆级教程(一)
人工智能·pytorch·深度学习