mirror of https://github.com/JosephKJ/OWOD.git
17 lines
532 B
Python
17 lines
532 B
Python
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
|
|
from fvcore.common.file_io import PathManager
|
|
|
|
from detectron2.data import MetadataCatalog
|
|
|
|
from densepose import DensePoseTransformData
|
|
|
|
|
|
def load_for_dataset(dataset_name):
|
|
path = MetadataCatalog.get(dataset_name).densepose_transform_src
|
|
densepose_transform_data_fpath = PathManager.get_local_path(path)
|
|
return DensePoseTransformData.load(densepose_transform_data_fpath)
|
|
|
|
|
|
def load_from_cfg(cfg):
|
|
return load_for_dataset(cfg.DATASETS.TEST[0])
|