data_gen/utils/path_converter.py

import os


class PathConverter():
    def __init__(self):
        self.prefixs = {
            "vid": "/video/",
            "gt": "/gt_imgs/",
            "head": "/head_imgs/", 
            "torso": "/torso_imgs/", 
            "person": "/person_imgs/", 
            "torso_with_bg": "/torso_with_bg_imgs/", 
            "single_bg": "/bg_img/",
            "bg": "/bg_imgs/",
            "segmaps": "/segmaps/",
            "inpaint_torso": "/inpaint_torso_imgs/",
            "com": "/com_imgs/",
            "inpaint_torso_with_com_bg": "/inpaint_torso_with_com_bg_imgs/",
        }
        
    def to(self, path: str, old_pattern: str, new_pattern: str):
        return path.replace(self.prefixs[old_pattern], self.prefixs[new_pattern], 1)

pc = PathConverter()