Update database.py

Fixes #3.
thomeou · May 12, 2022 · 3c03140 · 3c03140
1 parent cf59e4b
commit 3c03140
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dataset/database.py b/dataset/database.py
@@ -112,7 +112,7 @@ def get_segment_idxes(self, n_frames: int, downsample_ratio: int, pointer: int):
                           pointer + n_crop_frames - self.chunk_len // downsample_ratio + 1,
                           self.chunk_hop_len // downsample_ratio).tolist()
         # Include the leftover of the cropped data
-        if (n_crop_frames - self.chunk_len // downsample_ratio) % self.chunk_hop_len // downsample_ratio != 0:
+        if (n_crop_frames - self.chunk_len // downsample_ratio) % (self.chunk_hop_len // downsample_ratio) != 0:
             idxes.append(pointer + n_crop_frames - self.chunk_len // downsample_ratio)
         pointer += n_crop_frames
 
@@ -307,4 +307,4 @@ def load_classwise_gt(self, gt_meta_fn, n_frames):
     print(db_data['doa_targets'].shape)
     print(len(db_data['feature_chunk_idxes']))
     print(len(db_data['gt_chunk_idxes']))
-    print(len(db_data['filename_list']))
+    print(len(db_data['filename_list']))