/datasetutils

Making it easier to work with (some) public datasets

Primary LanguagePythonMIT LicenseMIT

1) BSDS500 dataset

https://www2.eecs.berkeley.edu/Research/Projects/CS/vision/grouping/resources.html

Usage

from bsds500 import BSDS500

bsds = BSDS500(
    path_to_bsds=None, # we will download the dataset
    images_to_gray=False, 
    target_size=(256,256), 
    masks_to_binary=True
)

fileids,contours,segments,images = bsds.get_train()

print(len(fileids), contours.shape, segments.shape, images.shape, sep='\n')

#1087
#(1087, 256, 256, 1)
#(1087, 256, 256, 1)
#(1087, 256, 256, 3)