added annot

data_loader.py

@@ -36,6 +36,17 @@ class DataLoader():
             images.append(image)
             annotations.append(anno)
         return images,annotations
+
+    def get_all_annotations(self):
+        '''
+        Function to get all annotations
+        '''
+        annotations = []
+        for pidx in self.pidxs:
+            print(pidx)
+            image,anno = self.pin_dataset.get_annotation(pin_id=pidx)
+            annotations.append(anno)
+        return annotations
 '''
 if __name__ == "__main__":
     data_loader = DataLoader()