如何使用pytorch加載并讀取COCO數(shù)據(jù)集

這篇文章主要介紹“如何使用pytorch加載并讀取COCO數(shù)據(jù)集”，在日常操作中，相信很多人在如何使用pytorch加載并讀取COCO數(shù)據(jù)集問題上存在疑惑，小編查閱了各式資料，整理出簡單好用的操作方法，希望對大家解答”如何使用pytorch加載并讀取COCO數(shù)據(jù)集”的疑惑有所幫助！接下來，請跟著小編一起來學(xué)習(xí)吧！

環(huán)境配置

看pytorch入門教程

基礎(chǔ)知識：元祖、字典、數(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)證集和對應(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分開讀取
image, info = coco_dataset[0]
# ImageDraw 畫圖工具
image_handler = ImageDraw.ImageDraw(image)
for annotation in info:
    # bbox為檢測框的位置坐標(biāo)
    x_min, y_min, width, height = annotation['bbox']
    # ((), ())分別為左上角的坐標(biāo)對和右上角的坐標(biāo)對，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ù)集”的學(xué)習(xí)就結(jié)束了，希望能夠解決大家的疑惑。理論與實(shí)踐的搭配能更好的幫助大家學(xué)習(xí)，快去試試吧！若想繼續(xù)學(xué)習(xí)更多相關(guān)知識，請繼續(xù)關(guān)注億速云網(wǎng)站，小編會繼續(xù)努力為大家?guī)砀鄬?shí)用的文章！