pytorch读取本地数据集,pytorch 读取数据
本文主要介绍使用pytorch加载和读取COCO数据集,包括祖先、字典和数组。本文通过示例代码向您详细介绍,对您的学习或工作有一定的参考价值。有需要的朋友可以参考一下。
00-1010环境配置基础知识:祖先、字典、数组、用PyTorch读取COCO数据集、用PyTorch读取自制数据集、如何用PyTorch加载读取COCO数据集、环境配置基础知识:祖先、字典、数组、用PyTorch读取COCO数据集、用pytorch读取自制数据集
目录
参见pytorch入门教程
环境配置
#元祖
a=(1,2)
#字典
b={ 用户名 : 陪陪网,代码 : 111}
#数组
c=[1,2,3]
打印(a[0])
print(c[0])
打印(b[用户名])
输出:
基础知识:元祖、字典、数组
进口火炬视觉
从PIL进口ImageDraw
#导入coco 2017验证集和相应的标注
COCO _ dataset=torch vision . datasets . COCO detection(root= COCO _ dataset _ val _ 2017/val 2017 ,
ann file= COCO _ dataset _ val _ 2017/annotations _ train val 2017/annotations/instances _ val 2017 . JSON )
#图像和注释分开读取。
image,info=coco_dataset[0]
# ImageDraw绘图工具
image_handler=ImageDraw。ImageDraw(图像)
对于info:中的注释
# bbox是检测框的位置坐标。
x_min,y_min,width,height=annotation[bbox]
#((),())分别是左上角的坐标对和右上角的坐标对。image_handler.rectangle是指在图片中画一个方框。
image _ handler . rectangle((x_min,y_min),(x _ min宽度,y_min高度)))
image.show()
结果:
利用PyTorch读取COCO数据集
使用cvat工具创建自己的数据集标注,导出为coco格式并读取。
结果:
关于用pytorch加载和读取COCO数据集的这篇文章到此为止。有关pytorch阅读COCO数据集的更多信息,请搜索热门IT软件开发工作室之前的文章或继续浏览下面的相关文章。希望大家以后多多支持热门IT软件开发工作室!
郑重声明:本文由网友发布,不代表盛行IT的观点,版权归原作者所有,仅为传播更多信息之目的,如有侵权请联系,我们将第一时间修改或删除,多谢。