train_dataset = dset.CocoDetection(root = path2data,
annFile = path2json)在这个函数之后，我想使用train_dataset得到这三个输出，但是它不起作用。
train_image, train_label, train_segm = train_dataset[6]我目前得到两个输出，作为train_dataset image和traget的结果。
我需要train_image、train_label、train_segm来执行将来要使用的操作。
1条答案
按热度按时间ljo96ir51#
不要以为你能一步到位。
如www.example.com中的getitem代码https://pytorch.org/vision/stable/_modules/torchvision/datasets/coco.html#CocoDetection.__getitem__示，加载器只提供图像和目标（注解ID）。
然后，您必须使用沿着COCO annotations to masks - For instance segmentation model training的语句将annotationId转换为线段