luni/cvpr_2023
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix static cache

Browse Source
This commit is contained in:
luni
2023-07-01 10:44:32 +02:00
parent d3ff1c05e1
commit f9ac7fe3c3
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
data/dataset.py
+2 -2
View File
@@ -104,7 +104,7 @@ class StaticCache():
return data return data
def __getitem__(self, frame) -> dict: def __getitem__(self, frame: Frame) -> dict:
if (frame.id, frame.garment) not in self.store: if (frame.id, frame.garment) not in self.store:
item = self.read(frame.id, frame.garment) item = self.read(frame.id, frame.garment)
stored_data = { key: torch.from_numpy(value).to('cpu') for key, value in item.items()} stored_data = { key: torch.from_numpy(value).to('cpu') for key, value in item.items()}
@@ -164,7 +164,7 @@ class DatasetCloth3D(torch.utils.data.Dataset):
if self.cache is not None: if self.cache is not None:
if frame in self.cache.keys(): if frame in self.cache.keys():
data = { key: (value.to(self.device) if type(value) == torch.Tensor else value) for key, value in self.cache[frame].items()} data = { key: (value.to(self.device) if type(value) == torch.Tensor else value) for key, value in self.cache[frame].items()}
data = { **data, **self.static_cache[frame.id] } data = { **data, **self.static_cache[frame] }
return data return data
item = dict() item = dict()
data/utils.py
+3 -3
View File
@@ -2,9 +2,9 @@ import tables as tb
class Frame: class Frame:
def __init__(self, id: str, frame_nr: int, garment: str=''): def __init__(self, id: str, frame_nr: int, garment: str=''):
self.id = id self.id: str = id
self.frame_nr = frame_nr self.frame_nr: int = frame_nr
self.garment = garment self.garment: str = garment
def __hash__(self): def __hash__(self):
return hash((self.id, self.frame_nr, self.garment)) return hash((self.id, self.frame_nr, self.garment))