使用pytorch加载并读取COCO数据集的详细操作
作者：爱学习的小登西 发布时间：2021-01-04 21:21:24
标签：pytorch,COCO,数据集
如何使用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"])
输出：
利用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格式并读取
结果：
来源：https://blog.csdn.net/m0_52127604/article/details/124698579
0
投稿猜你喜欢
- 1.字符强转一切如果是字符串进行强制转换, 仅仅就是在原数据类型的两边套上引号2.list : 强制转换成列表""&qu
- folder.htm<html><head><title>闪亮日子之在线创建文件夹</title&
PHP children() 函数实例查找 note 节点的子节点：<?php $note=<<<XML<no- 本文实例讲述了Selenium定位元素操作。分享给大家供大家参考，具体如下：Selenium是一个用于Web应用程序测试的工具。Seleni
- 1、GIL简介GIL的全称为Global Interpreter Lock，全局解释器锁。1.1 GIL设计理念与限制python的代码执行
- 定义一个变量，直接输出会输出变量的属性，并不能输出变量值。那么怎么输出变量值呢？请看下面得意import tensorflow as tf
- 如下所示：Numpy中reshape的使用方法为:numpy.reshape(a, newshape, order='C')
- 高效的css写法中的一条就是使用简写。通过简写可以让你的CSS文件更小，更易读。而了解CSS属性简写也是前端开发工程师的基本功之一。今天我们
- Mysqli是php5之后才有的功能，没有开启扩展的朋友可以打开您的php.ini的配置文件。 查找下面的语句：;extension=php
- 实现爬虫的增量式爬取有两种方法，一是在获得页面解析的内容后判断该内容是否已经被爬取过，二是在发送请求之前判断要被请求的url是否已经被爬取过
- 环境：ubuntu 16.04 python3.5 pycharm包 ： wave pyaudio sys上代码：AudioPlayer.p
- 1. SELECT…FOR UPDATE 是什么？作用是什么？select for update 即排他锁，排他锁又称
- 最近工作上有个需求，当爬虫程序遇到异常的时候，需要通知相应的人员进行修复。如果是国外可能是通过邮件的方式来通知，但国内除了万年不变的 qq
- 本文实例讲述了Python事务操作实现方法。分享给大家供大家参考，具体如下：#coding=utf-8import sysimport My
- Python绘制双轴组合的关键在plt库的twinx()函数，具体流程：1.先建立坐标系，然后绘制主坐标轴上的图表；2.再调用plt.twi
- pygame.mixer是一个用来处理声音的模块，其含义为“混音器”。游戏中对声音的处理一般包括制造声音和播放声音两部分，这里仅学习了播放声
- 方法说明： 同步版的fs.writeFile() 。语法：fs.writeFileSync(filename, data, [o
- 前言本文主要给大家介绍了关于Django跨域请求问题解决的几种方法，分享出来供大家参考学习，下面话不多说了，来一起看看详细的介绍吧。几种方法
- 先看一个实例这是我用asp写的一个搜索一个字符串里面第一张图片地址的函数（当然你可以将values那里换一个得到所有图片地址）functio
- 秒杀活动可以说在互联网上随处可见，从12306抢票，到聚划算抢购，我们生活的方方面面都可以看到秒杀的身影。秒杀的架构设计也是对于一个架构师架
