lib/data_factory/common/ds_formatter.py

import os
import os.path as osp
import numpy as np
import numpy.random as npr
import torch
from PIL import Image
import copy
import gc
import itertools

def singleton(class_):
    instances = {}
    def getinstance(*args, **kwargs):
        if class_ not in instances:
            instances[class_] = class_(*args, **kwargs)
        return instances[class_]
    return getinstance

@singleton
class get_formatter(object):
    def __init__(self):
        self.formatter = {}

    def register(self, formatf):
        self.formatter[formatf.__name__] = formatf

    def __call__(self, cfg):
        if cfg is None:
            return None
        t = cfg.type
        return self.formatter[t](**cfg.args)

def register():
    def wrapper(class_):
        get_formatter().register(class_)
        return class_
    return wrapper