search

SonyCSLParis / pesto

Self-supervised learning for fast pitch estimation
GNU Lesser General Public License v3.0
168 stars 15 forks source link

solve bug when exporting in png #1

Closed aRI0U closed 10 months ago

aRI0U commented 1 year ago 
$ pesto AClassicEducation_NightOwl_STEM_08.RESYN.wav -e csv png
Traceback (most recent call last):TEM_08.RESYN.wav
  File "/home/alain/.local/bin/pesto", line 8, in <module>
    sys.exit(pesto())
  File "/home/alain/.local/lib/python3.10/site-packages/pesto/main.py", line 8, in pesto
    predict_from_files(**vars(args))
  File "/home/alain/.local/lib/python3.10/site-packages/torch/utils/_contextlib.py", line 115, in decorate_context
    return func(*args, **kwargs)
  File "/home/alain/.local/lib/python3.10/site-packages/pesto/predict.py", line 118, in predict_from_files
    export(fmt, output_file, *predictions)
  File "/home/alain/.local/lib/python3.10/site-packages/pesto/export.py", line 16, in export
    export_png(output_file, timesteps, pitch, confidence, activations)
  File "/home/alain/.local/lib/python3.10/site-packages/pesto/export.py", line 34, in export_png
    bps = activations.size(1) // 128
IndexError: Dimension out of range (expected to be in range of [-1, 0], but got 1)