Is this mean filter really correct?

[SimZhou]

https://github.com/turpaultn/DCASE2019_task4/blob/4d42e1fcc2cc0097979ac0f5935f0fd548945e0a/baseline/evaluation_measures.py#L214 In my understanding, pred_strong should be an array of 2 dims: (BatchSize, SeqLength), the median filter should do the median along the sequence, not the batch.

So I guess the filter should be (1, cfg.median_window) ?