利用COCO格式 加载自己的数据集

import torchvision.datasets
from PIL import ImageDraw

# 下面的路径可以替换为自己标注的数据集
coco_dataset = torchvision.datasets.CocoDetection(root=r"D:\PycharmProjects\learn_pytorch\data\val2017",
                                                  annFile=r"D:\PycharmProjects\learn_pytorch\data\annotations\instances_val2017.json")
print(coco_dataset[0])    #可以在这一步加断点， 然后debug，再点expression按钮，查看返回的是什么？ 这里返回的是元组（img,segm）

img, info = coco_dataset[0]
# img.show()

img_handler = ImageDraw.ImageDraw(img)

for annotation in info:
    x_min, y_min, width, height = annotation["bbox"]
    img_handler.rectangle(((x_min, y_min),(x_min + width, y_min + height)))

img.show()