added frame num to __get_item__

ds1.py

@@ -39,6 +39,7 @@ class PatchDataset(Dataset):
         self.length = 0
         self.xy_positions = []
         self.patches=[]
+        self.f_nums=[]
         with open(ffile, 'rb') as f:
             for fNr in range(1,nFrames+1):
                 BytesToSkip = fHead + fNr*NrPixels*NrPixels*2
@@ -65,6 +66,7 @@ class PatchDataset(Dataset):
                     sub_img = thisFrame[start_y:end_y,start_x:end_x]
                     self.patches.append(normalize_patch(sub_img))
                     self.xy_positions.append([start_y,start_x])
+                    self.f_nums.append(fNr)
                 self.length += len(self.patches)
         f.close()
 
@@ -72,5 +74,5 @@ class PatchDataset(Dataset):
         return self.length
 
     def __getitem__(self, index):
-        return self.patches[index], self.xy_positions[index]
+        return self.patches[index], self.xy_positions[index], self.f_nums[index]