這篇文章主要介紹了使用pytorch加載并讀取COCO數(shù)據(jù)集,基礎(chǔ)知識(shí)包括元祖、字典、數(shù)組，本文通過(guò)實(shí)例代碼給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下

環(huán)境配置

看pytorch入門(mén)教程

基礎(chǔ)知識(shí)：元祖、字典、數(shù)組

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

輸出：

利用PyTorch讀取COCO數(shù)據(jù)集

import torchvision
from PIL import ImageDraw
# 導(dǎo)入coco 2017 驗(yàn)證集和對(duì)應(yīng)annotations
coco_dataset = torchvision.datasets.CocoDetection(root="COCO_dataset_val_2017/val2017",
                                                  annFile="COCO_dataset_val_2017/annotations_trainval2017/annotations/instances_val2017.json")
# 圖像和annotation分開(kāi)讀取
image, info = coco_dataset[0]
# ImageDraw 畫(huà)圖工具
image_handler = ImageDraw.ImageDraw(image)
for annotation in info:
    # bbox為檢測(cè)框的位置坐標(biāo)
    x_min, y_min, width, height = annotation['bbox']
    # ((), ())分別為左上角的坐標(biāo)對(duì)和右上角的坐標(biāo)對(duì)，image_handler.rectangle是指在圖片是繪制方框
    image_handler.rectangle(((x_min, y_min), (x_min + width, y_min + height)))
image.show()

結(jié)果：

利用PyTorch讀取自己制作的數(shù)據(jù)集

使用cvat工具創(chuàng)建自己的數(shù)據(jù)集標(biāo)注，導(dǎo)出為coco格式并讀取
結(jié)果：

到此這篇關(guān)于使用pytorch加載并讀取COCO數(shù)據(jù)集的文章就介紹到這了,更多相關(guān)pytorch 讀取COCO數(shù)據(jù)集內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章:

相關(guān)文章

Python數(shù)據(jù)可視化探索實(shí)例分享
這篇文章主要介紹了Python數(shù)據(jù)可視化探索實(shí)例分享，數(shù)據(jù)可視化是指用圖形或表格的方式來(lái)呈現(xiàn)數(shù)據(jù)，關(guān)于更多相關(guān)介紹需要的小伙伴可以參考下面文章的具體內(nèi)容
2022-05-05
集成開(kāi)發(fā)環(huán)境Pycharm的安裝及模板設(shè)置圖文教程
PyCharm是一種Python的集成開(kāi)發(fā)環(huán)境，帶有一整套可以幫助用戶在使用Python語(yǔ)言開(kāi)發(fā)時(shí)提高效率的工具，這篇文章主要介紹了集成開(kāi)發(fā)環(huán)境Pycharm的安裝及模板設(shè)置,需要的朋友可以參考下
2022-07-07
python開(kāi)發(fā)前景如何
在本篇文章中小編給大家整理了關(guān)于python開(kāi)發(fā)前景的知識(shí)點(diǎn)及相關(guān)內(nèi)容，有興趣的朋友們可以跟著學(xué)習(xí)參考下。
2020-06-06
解讀torch.nn.GRU的輸入及輸出示例
這篇文章主要介紹了解讀torch.nn.GRU的輸入及輸出示例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教
2023-01-01
使用matplotlib的pyplot模塊繪圖的實(shí)現(xiàn)示例
這篇文章主要介紹了使用matplotlib的pyplot模塊繪圖的實(shí)現(xiàn)示例，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧
2020-07-07
自己用python做的一款超炫酷音樂(lè)播放器
這篇文章主要介紹了自己用python做的一款超炫酷音樂(lè)播放器，本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下
2021-08-08
Python的Flask框架中集成CKeditor富文本編輯器的教程
在用Flask搭建網(wǎng)站時(shí)的后臺(tái)文章編輯器可以使用CKeditor,CKeditor所支持的文本樣式較多且開(kāi)源,這里我們就來(lái)看一下Python的Flask框架中集成CKeditor富文本編輯器的教程
2016-06-06
解決python中顯示圖片的plt.imshow plt.show()內(nèi)存泄漏問(wèn)題
這篇文章主要介紹了解決python中顯示圖片的plt.imshow plt.show()內(nèi)存泄漏問(wèn)題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
2020-04-04
使用Python的urllib和urllib2模塊制作爬蟲(chóng)的實(shí)例教程
這篇文章主要介紹了使用Python的urllib和urllib2模塊制作爬蟲(chóng)的實(shí)例教程,展現(xiàn)了這兩個(gè)常用爬蟲(chóng)制作模塊的基本用法,極度推薦!需要的朋友可以參考下
2016-01-01
python中pow函數(shù)用法及功能說(shuō)明
在本篇文章里小編給大家整理的是一篇關(guān)于python中pow函數(shù)用法及功能說(shuō)明內(nèi)容，有需要的朋友們可以學(xué)習(xí)下。
2020-12-12