trying to get basic model working

utils/data_utils.py

@@ -8,7 +8,7 @@ from math import cos, sin
 
 
 sys.path.append(os.path.dirname(os.path.abspath(__file__)) + '/..')
-from superSMPL.IO import readPC2Frame, readOBJ
+# from superSMPL.IO import readPC2Frame, readOBJ
 
 
 class Frame:
@@ -17,6 +17,12 @@ class Frame:
         self.frame_nr = frame_nr
         self.garment = garment
 
+    def __hash__(self):
+        return hash((self.id, self.frame_nr))
+
+    def __eq__(self, other):
+        return (self.id, self.frame_nr) == (other.id, other.frame_nr)
+
 
 def mask_faces(F, mask):
     F_mask = np.take(mask, F.reshape(-1), axis=0)