index by word, consultant and/or repetition

Author: tefenet

examples/test_basics.py

@@ -1,6 +1,6 @@
 import sldatasets as sld
 
-dataset = sld.get("asllvd")
+dataset = sld.get("asllvd", word='Twenty', consultant='liz')
 print("dataset summary:")
 dataset.summary()
 for video, description in dataset:
@@ -9,14 +9,20 @@
     frame, frame_annot = video[3]
     print('frame shape: ', frame.shape)
     print('estimated body parts (x,y,score): ')
+    # this method pretty print a Dictionary that returns the method frame_annot.get()
     frame_annot.pretty()
+
     face = frame_annot.h.get_face_box(640, 480)
     print('face box: ', face)
-    print(frame_annot.h.get_upper_body_box(640, 480))
-# dataset = sld.get("lsa64", version='pre')
-# dataset.summary()
-# for video, description in dataset:
-#     print(description)
-#     frame, frame_annot = video[3]
-#     print(frame.shape)
-#     print(frame_annot)
+    print('upper body box: ', frame_annot.h.get_upper_body_box(640, 480))
+
+print('\n testing lsa64...')
+dataset = sld.get("lsa64", version='cut', word=25,
+                  consultant=10, repetition=5)
+dataset.summary()
+for video, description in dataset:
+    print(description)
+    frame, frame_annot = video[3]
+    print('frame shape: ', frame.shape)
+    print('estimated body parts (x,y,score): ')
+    frame_annot.pretty()

sldatasets/datasethandler.py

@@ -46,11 +46,15 @@ def redux(self, files, **kwargs):
 
         def file_filter(self, f, clas, consultant, repetition):
             d_specs = self.video_info(f)
-            return ((int(self.word_class(d_specs)) == int(clas) if clas else True) and
-                    (int(self.consultant_of(d_specs)) == consultant if consultant else True) and
-                    (int(self.rept(d_specs)) == repetition if repetition else True))
+            try:
+                clas=clas.lower()
+                consultant=consultant.lower()            
+            finally:
+                return ((self.word_class(d_specs) == clas if clas else True) and
+                        (self.consultant_of(d_specs) == consultant if consultant else True) and
+                        (self.rept(d_specs) == repetition if repetition else True))
         l = list(filter(lambda f: file_filter(
-            self, f, kwargs.get('index'), kwargs.get('consultant'), kwargs.get('repetition')), l))
+            self, f, kwargs.get('word'), kwargs.get('consultant'), kwargs.get('repetition')), l))
         return l
 
     def specs_from(self, filename):
@@ -111,9 +115,9 @@ def get_my_file_ext(self):
     def video_info(self,filename):
         splited = filename.split('_')
         info_dict={}
-        info_dict['class']= splited[0]
-        info_dict['consultant']= splited[1]        
-        info_dict['repetition'] = splited[2].split('.')[0]
+        info_dict['class']= int(splited[0])
+        info_dict['consultant']= int(splited[1])
+        info_dict['repetition'] = int(splited[2].split('.')[0])
         return info_dict
 
 
@@ -133,9 +137,9 @@ def get_pos_url(self):
     def video_info(self, filename):
         splited = filename.split('_')
         info_dict = {}
-        info_dict['class']= splited[0]
-        info_dict['consultant']= splited[1]        
-        info_dict['repetition'] = splited[2]
+        info_dict['class']= int(splited[0])
+        info_dict['consultant']= int(splited[1])
+        info_dict['repetition'] = int(splited[2])
         info_dict['hand'] = splited[3].split('.')[0]
         return info_dict
 
@@ -197,10 +201,10 @@ def video_info(self, filename):
         return self.dframe.loc[:, 'Consultant':'Passive Arm'][(self.dframe.Filename == filename)].to_dict('records')[0]
 
     def consultant_of(self, d):
-        return d['Consultant']
+        return d['Consultant'].lower()
 
     def word_class(self, d):
-        return d['Main_Gloss']
+        return d['Main_Gloss'].lower()
 
     def rept(self, d):
         return None

sldatasets/video_dataset.py

@@ -25,8 +25,6 @@ def send(self, ignored_arg):
             for f in frames:
                 l.append((f, anotation[anot_index]))
             return (l, specs)
-        except ModuleNotFoundError:
-            print(anotation[anot_index].__class__)
 
     def throw(self, type=None, value=None, traceback=None):
         raise StopIteration