SAMHI/resources/SimpleClick/isegm/data/datasets/berkeley.py

from .grabcut import GrabCutDataset


class BerkeleyDataset(GrabCutDataset):
    def __init__(self, dataset_path, **kwargs):
        super().__init__(dataset_path, images_dir_name='images', masks_dir_name='masks', **kwargs)