使用pytorch加载并读取COCO数据集的详细操作

更新时间：2022年05月12日 17:03:08 作者：爱学习的小登西

这篇文章主要介绍了使用pytorch加载并读取COCO数据集,基础知识包括元祖、字典、数组，本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下

环境配置

看pytorch入门教程

基础知识：元祖、字典、数组

# 元祖
a = (1, 2)
# 字典
b = {'username': 'peipeiwang', 'code': '111'}
# 数组
c = [1, 2, 3]
print(a[0])
print(c[0])
print(b["username"])

输出：

利用PyTorch读取COCO数据集

import torchvision
from PIL import ImageDraw
# 导入coco 2017 验证集和对应annotations
coco_dataset = torchvision.datasets.CocoDetection(root="COCO_dataset_val_2017/val2017",
                                                  annFile="COCO_dataset_val_2017/annotations_trainval2017/annotations/instances_val2017.json")
# 图像和annotation分开读取
image, info = coco_dataset[0]
# ImageDraw 画图工具
image_handler = ImageDraw.ImageDraw(image)
for annotation in info:
    # bbox为检测框的位置坐标
    x_min, y_min, width, height = annotation['bbox']
    # ((), ())分别为左上角的坐标对和右上角的坐标对，image_handler.rectangle是指在图片是绘制方框
    image_handler.rectangle(((x_min, y_min), (x_min + width, y_min + height)))
image.show()

结果：

利用PyTorch读取自己制作的数据集

使用cvat工具创建自己的数据集标注，导出为coco格式并读取
结果：

到此这篇关于使用pytorch加载并读取COCO数据集的文章就介绍到这了,更多相关pytorch 读取COCO数据集内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

python中print输出有空格如何解决
这篇文章主要介绍了python中print输出有空格的解决方案,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
2024-02-02
python矩阵转换为一维数组的实例
今天小编就为大家分享一篇python矩阵转换为一维数组的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2018-06-06
python实现从ftp服务器下载文件的方法
这篇文章主要介绍了python实现从ftp服务器下载文件的方法,涉及Python操作FTP的相关技巧,非常具有实用价值,需要的朋友可以参考下
2015-04-04
Python使用pymongo库操作MongoDB数据库的方法实例
今天小编就为大家分享一篇关于Python使用pymongo库操作MongoDB数据库的方法实例，小编觉得内容挺不错的，现在分享给大家，具有很好的参考价值，需要的朋友一起跟随小编来看看吧
2019-02-02
详解Tensorflow不同版本要求与CUDA及CUDNN版本对应关系
这篇文章主要介绍了详解Tensorflow不同版本要求与CUDA及CUDNN版本对应关系，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-08-08
Python3.10接入ChatGPT实现逐句回答流式返回
这篇文章主为大家要介绍了Python3.10接入ChatGPT实现逐句回答流式返回示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2023-03-03
Python生成并下载文件后端代码实例
这篇文章主要介绍了Python生成并下载文件后端代码实例,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2020-08-08
python35种绘图函数详细总结
Python有许多用于绘图的函数和库,比如Matplotlib,Plotly,Bokeh,Seaborn等,这只是一些常用的绘图函数和库,Python还有其他绘图工具,如Pandas、ggplot等,选择适合你需求的库,可以根据你的数据类型、图形需求和个人偏好来决定,本文给大家总结了python35种绘图函数
2023-08-08
python中翻译功能translate模块实现方法
在本篇文章中小编给各位整理了一篇关于python中翻译功能translate模块实现方法，有需要的朋友们可以参考下。
2020-12-12
python中upper是做什么用的
在本篇文章里小编给大家整理的是一篇关于python中upper的作用的相关文章，有需要的朋友们可以参考下。
2020-07-07