search

turpaultn / DCASE2019_task4

Baseline of dcase 2019 task 4
59 stars 27 forks source link

Is this mean filter really correct? #18

Closed SimZhou closed 2 years ago

SimZhou commented 2 years ago

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) ?

turpaultn commented 2 years ago

Sorry for the delay in my answer. The array is of 2 dims: (SeqLength, NbClasses), that is why the median filter is like this :)