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

new dataloader

Browse Source
This commit is contained in:
luni
2023-04-01 19:07:53 +00:00
parent ba24dca22c
commit 1d6a089dd4
2 changed files with 207 additions and 461 deletions
Download Patch File Download Diff File Expand all files Collapse all files
data/utils.py
+3 -3
View File
@@ -1,16 +1,16 @@
import tables as tb
class Frame:
def __init__(self, id, frame_nr, garment):
def __init__(self, id, frame_nr, garment=''):
self.id = id
self.frame_nr = frame_nr
self.garment = garment
def __hash__(self):
return hash((self.id, self.frame_nr))
return hash((self.id, self.frame_nr, self.garment))
def __eq__(self, other):
return (self.id, self.frame_nr) == (other.id, other.frame_nr)
return (self.id, self.frame_nr, self.garment) == (other.id, other.frame_nr, self.garment)
def save_to_h5file(h5file, id, frame_nr, garment, data, name, type):