百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 热门文章 > 正文

168.人工智能——基于PaddleX目标检测任务模型训练与部署

bigegpt 2024-09-29 09:19 3 浏览

167.人工智能——智能标注PaddleLabel的使用 文章中,有讲如何使用PaddleLabel来智能标注数据。本文主要使用标注好的目标检测数据,使用PaddleX来进行模型训练和部署。关于PaddleX的使用,可以参看165.人工智能——飞桨AI套件PaddleX体验使用(猫狗分类任务)。

登录到PP飞桨AI Studio,创建PaddleX项目,选择基础模型PP-PicoDet_l_640_LCNet(有免费GPU算力),

一、准备数据集

PaddleX针对目标检测任务定义的数据集,名称是COCODetDataset,组织结构和标注格式如下。


dataset_dir                  # 数据集根目录,目录名称可以改变
├── annotations              # 标注文件的保存目录,目录名称不可改变
│   ├── instance_train.json  # 训练集标注文件,文件名称不可改变,采用COCO标注格式
│   └── instance_val.json    # 验证集标注文件,文件名称不可改变,采用COCO标注格式
└── images                   # 图像的保存目录,目录名称不可改变

对于使用LabelMe标注的数据集,或是VOC格式的数据集,可以使用x2coco脚本将数据集转换成COCO格式,

注意:在PaddleLabel目标检测任务导出的coco格式后,不能直接在项目中使用(查看文件,发现格式写法不同)。

解决办法:先导出VOC格式 ,再转换成coco格式,可以参看:126.人工智能——PascalVOC和MSCOCO数据集格式相互转换

二、打开项目,校验数据



三、模型训练

这里调整批次大小和学习率


四、模型评估

五、导出预测模型

离线部署包里有,有完整示例和模型以及详细的说明文档。

六、预测部署

本地预测代码

import fastdeploy as fd
import cv2
import os

model_dir="picodet-bicycle-infer_model"
model_file = os.path.join(model_dir, "inference.pdmodel")
params_file = os.path.join(model_dir, "inference.pdiparams")
config_file = os.path.join(model_dir, "inference.yml")

model=fd.vision.detection.PicoDet(model_file,params_file,config_file, 
                                  runtime_option=None,model_format=fd.ModelFormat.PADDLE)     

# 加载图像
image_file = "img/test3.jpg"
img=cv2.imread(image_file)

# 预测结果
result=model.predict(img)
print(result)

#直接可视化输出结果
vis_img= fd.vision.vis_detection(img,result,score_threshold=0.5)
#保存结果
#cv2.imwrite("visualized_result.jpg", vis_img)

cv2.imshow("result",vis_img)
cv2.waitKey(0)
cv2.destroyAllWindows()    


对于人工智能计算机视频任务,PaddleX套件还是非常友好。适合初学者、小项目小任务的学习体验。

相关推荐

有些人能留在你的心里,但不能留在你生活里。

有时候,你必须要明白,有些人能留在你的心里,但不能留在你生活里。Sometimes,youhavetorealize,Somepeoplecanstayinyourheart,...

Python学不会来打我(34)python函数爬取百度图片_附源码

随着人工智能和大数据的发展,图像数据的获取变得越来越重要。作为Python初学者,掌握如何从网页中抓取图片并保存到本地是一项非常实用的技能。本文将手把手教你使用Python函数编写一个简单的百度图片...

软网推荐:图像变变变 一“软”见分晓

当我们仅需要改变一些图片的分辨率、裁减尺寸、添加水印、标注文本、更改图片颜色,或将一种图片转换为另一种格式时,总比较讨厌使用一些大型的图像处理软件,尤其是当尚未安装此类软件时,更是如此。实际上,只需一...

首款WP8.1图片搜索应用,搜照片得资料

首款WP8.1图片搜索应用,搜照片得资料出处:IT之家原创(天际)2014-11-1114:32:15评论WP之家报道,《反向图片搜索》(ReverseImageSearch)是Window...

分享一组美图(图片来自头条)(头条美女头像)

...

盗墓笔记电视剧精美海报 盗墓笔记电视剧全集高清种子下载

出身“老九门”世家的吴邪,因身为考古学家的父母在某次保护国家文物行动时被国外盗墓团伙杀害,吴家为保护吴邪安全将他送去德国读书,因而吴邪对“考古”事业有着与生俱来的兴趣。在一次护宝过程中他偶然获得一张...

微软调整Win11 24H2装机策略:6月起36款预装应用改为完整版

IT之家7月16日消息,微软公司今天(7月16日)发布公告,表示自今年6月更新开始,已默认更新Windows1124H2和WindowsServer2025系统中预装...

谷歌手把手教你成为谣言终结者 | 域外

刺猬公社出品,必属原创,严禁转载。合作事宜,请联系微信号:yunlugongby贾宸琰编译、整理11月23日,由谷歌新闻实验室(GoogleNewsLab)联合Bellingcat、DigD...

NAS 部署网盘资源搜索神器:全网资源一键搜,免费看剧听歌超爽!

还在为找不到想看的电影、电视剧、音乐而烦恼?还在各个网盘之间来回切换,浪费大量时间?今天就教你如何在NAS上部署aipan-netdisk-search,一款强大的网盘资源搜索神器,让你全网资源...

使用 Docker Compose 简化 INFINI Console 与 Easysearch 环境搭建

前言回顾在上一篇文章《搭建持久化的INFINIConsole与Easysearch容器环境》中,我们详细介绍了如何使用基础的dockerrun命令,手动启动和配置INFINICon...

为庆祝杜特尔特到访,这个国家宣布全国放假?

(观察者网讯)近日,一篇流传甚广的脸书推文称,为庆祝杜特尔特去年访问印度,印度宣布全国放假,并举办了街头集会以示欢迎。菲媒对此做出澄清,这则消息其实是“假新闻”。据《菲律宾世界日报》2日报道,该贴子...

一课译词:毛骨悚然(毛骨悚然的意思是?)

PhotobyMoosePhotosfromPexels“毛骨悚然”,汉语成语,意思是毛发竖起,脊梁骨发冷;形容恐惧惊骇的样子(withone'shairstandingonend...

Bing Overtakes Google in China's PC Search Market, Fueled by AI and Microsoft Ecosystem

ScreenshotofBingChinahomepageTMTPOST--Inastunningturnintheglobalsearchenginerace,Mic...

找图不求人!6个以图搜图的识图网站推荐

【本文由小黑盒作者@crystalz于03月08日发布,转载请标明出处!】前言以图搜图,专业说法叫“反向图片搜索引擎”,是专门用来搜索相似图片、原始图片或图片来源的方法。常用来寻找现有图片的原始发布出...

浏览器功能和“油管”有什么关联?为什么要下载

现在有没有一款插件可以实现全部的功能,同时占用又小呢,主题主要是网站的一个外观,而且插件则主要是实现wordpress网站的一些功能,它不仅仅可以定制网站的外观,还可以实现很多插件的功能,搭载chro...