-
Notifications
You must be signed in to change notification settings - Fork 7.1k
Open
Description
The segmentation references already support VOC and COCO
vision/references/segmentation/train.py
Lines 20 to 24 in 08c8f0e
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
vision/references/detection/train.py
Line 37 in a1232c2
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.
yassineAlouini and lpuglia