Fixes OMR-Research#13

v-dvorak · Sep 6, 2024 · 6aac9dc · 6aac9dc
1 parent ff46218
commit 6aac9dc
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/MeasureDetector/create_tf_record_from_joint_dataset.py b/MeasureDetector/create_tf_record_from_joint_dataset.py
@@ -99,6 +99,10 @@ def main(dataset_directory: str, annotations_filename: str, output_path: str, la
                 current_engraving, number_of_staves = sampling_categories[index % len(sampling_categories)]
                 all_items_in_category = dataset[current_engraving][number_of_staves]
 
+                if len(all_items_in_category) == 0:
+                    index += 1
+                    continue
+
                 encoding_succeeded = False
                 tf_example = None
                 while not encoding_succeeded: