from .oxford_pet import OxfordPetDataset, SimpleOxfordPetDataset