Support skipping glide

Author: yqzhishen

configs/continuous.yaml

@@ -7,6 +7,7 @@ raw_data_dir: []
 binary_data_dir: null
 binarization_args:
   num_workers: 0
+  skip_glide: true  # skip data with glide
   merge_rest: true  # merge continuous rest notes
   merge_slur: true  # merge slurs with the similar pitch
   round_midi: false  # round midi value

configs/discrete.yaml

@@ -8,6 +8,7 @@ binary_data_dir: data/some_ds_quant_spk4_aug8/binary
 binarization_args:
   num_workers: 0
   shuffle: true
+  skip_glide: true  # skip data with glide
   merge_rest: true  # merge continuous rest notes
   merge_slur: true  # merge slurs with the similar pitch
 use_bound_loss: true

preprocessing/me_binarizer.py

@@ -39,6 +39,7 @@ class MIDIExtractionBinarizer(BaseBinarizer):
     def __init__(self, config: dict):
         super().__init__(config, data_attrs=MIDI_EXTRACTION_ITEM_ATTRIBUTES)
         self.lr = LengthRegulator().to(self.device)
+        self.skip_glide = self.binarization_args['skip_glide']
         self.merge_rest = self.binarization_args['merge_rest']
         self.merge_slur = self.binarization_args['merge_slur']
         self.slur_tolerance = self.binarization_args.get('slur_tolerance')
@@ -60,6 +61,11 @@ def load_meta_data(self, raw_data_dir: pathlib.Path, ds_id):
                     ds = json.load(f)
                     if isinstance(ds, list):
                         ds = ds[0]
+                if self.skip_glide and ds.get('note_glide') is not None and any(
+                        g != 'none' for g in ds['note_glide'].split()
+                ):
+                    print(f'Item {ds_id}:{item_name} contains glide notes. Skipping.')
+                    continue
                 # normalize
                 note_seq = [
                     librosa.midi_to_note(