Skip to content

Add support for VOC on detection references #6037

Open
@pmeier

Description

@pmeier

The segmentation references already support VOC and COCO

paths = {
"voc": (dir_path, torchvision.datasets.VOCSegmentation, 21),
"voc_aug": (dir_path, sbd, 21),
"coco": (dir_path, get_coco, 21),
}

Unfortunately, the detection references only support COCO

paths = {"coco": (data_path, get_coco, 91), "coco_kp": (data_path, get_coco_kp, 2)}

There was a first push for this in #1216 by @lpuglia. We can probably salvage some parts from the PR.

We are currently reworking the datasets and transforms, and in turn the reference scripts will change quite a bit. Thus, we should only go for this after they are stable-ish again.

cc @datumbox @YosuaMichael

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions