File size: 246 Bytes
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
        )