File size: 1,181 Bytes
2b534de | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 | from confs import *
import json,random,os
import numpy as np
class Mediapipe_Result_Cache:
"""
Convention: when a cache entry exists, it must not be None.
In other words, None results should not be cached; get/set guard against historical None values.
"""
# DIR = Path('/inspurfs/group/mayuexin/suncy/mediapipe_result/A')
DIR = Path('data/mediapipe_result')
def __init__(self):
pass
def get_path(self, img_path):
img_path = Path(img_path)
str_img_folder = str(img_path.parent)
assert '|' not in str_img_folder
str_img_folder = str_img_folder.replace('/', '|')
lmk_folder = self.DIR / str_img_folder
lmk_folder.mkdir(parents=1, exist_ok=True)
ret= lmk_folder / (img_path.name+'.npy')
return ret
def get(self, img_path):
path = self.get_path(img_path)
# print(f"[get] {path=}")
if path.exists():
ret = np.load(path)
assert ret is not None
return ret
def set(self, img_path, lmks):
assert lmks is not None
path = self.get_path(img_path)
np.save(path, lmks)
# print(f"{path=}")
|