EscVM / OIDv4_ToolKit

Download and visualize single or multiple classes from the huge Open Images v4 dataset
GNU General Public License v3.0
809 stars 635 forks source link

Instance Segmentation #93

Closed faresbs closed 3 years ago

faresbs commented 3 years ago

Hello,

Is there a way to download the images from the Open Images Dataset v6 dataset with their Object Segmentation annotations?

I appreciate your help in advance.

Sadam1195 commented 3 years ago

Did you find any solution for that? @faresbs

whitenightwu commented 3 years ago

Did you find any solution for that? @faresbs

u can use 'fiftyone'. like

import fiftyone as fo
import fiftyone.zoo as foz
dataset = foz.load_zoo_dataset(
    "open-images-v6",
    splits=("train", "validation", "test"),
    label_types=["segmentations"],
    classes = ['Apple', 'Grape', 'Strawberry', 'Lemon'],
    #max_samples=3000,
    seed=51,
    shuffle=True
)