发布时间:2023-10-25 18:30
如何使用pytorch加载并读取COCO数据集 环境配置基础知识:元祖、字典、数组利用PyTorch读取COCO数据集利用PyTorch读取自己制作的数据集
看pytorch入门教程
# 元祖 a = (1, 2) # 字典 b = {\'username\': \'peipeiwang\', \'code\': \'111\'} # 数组 c = [1, 2, 3] print(a[0]) print(c[0]) print(b[\"username\"])
输出:
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()
结果:
使用cvat工具创建自己的数据集标注,导出为coco格式并读取
结果:
到此这篇关于使用pytorch加载并读取COCO数据集的文章就介绍到这了,更多相关pytorch 读取COCO数据集内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!
[SpringBoot的@Async注解和关键字synchronized的使用]
【MQTT从入门到提高系列 | 03】一文掌握MQTT3.1.1协议框架
语料库数据处理个案实例(读取多个文本文件、读取一个文件夹下面指定的多个文件、解码错误、读取多个子文件夹文本、多个文件批量改名)
torch.Tensor.repeat是什么操作?对比numpy库和torch库中的tile和repeat操作
BI的边界:BI不适合做什么?主数据、MarTech?该如何扩展?
Python OpenCV中的drawMatches()关键匹配绘制方法
HCNP Routing&Switching之Super VLAN