INNER CODE UNIT · Python
__init__
patrickjohncyh/fashion-clip · fashion_clip/fashion_clip.py:44
def __init__(self, name: str, image_source_path: str, image_source_type: str, catalog: List[dict] = None):
image_source_type = image_source_type.upper()
assert image_source_type in ['LOCAL', 'S3', 'URL']
if name in _CATALOGS:
print('Loading dataset {}...'.format(name))
catalog_path = _download(_CATALOGS[name], _CACHE_DIR)
with open(catalog_path, 'rb') as f:
catalog = json.load(f)
elif catalog is None:
raise RuntimeError(f"Catalog {name} not found; available catalogs = {list(_CATALOGS.keys())}")
self.name = name
self.catalog = np.array(catalog)
# extract FCLIP related info
self.ids = np.array([str(_['id']) for _ in catalog])
self.images = np.array([_['image'] for _ in catalog])
self.images_path = np.array([os.path.join(image_source_path, _['image']) for _ in catalog])