2cdd41c 1615d09
1
2
3
4
5
6
7
8
9
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 )