Typos and recursive dataset structure in utils.bench

InterDigitalInc / CompressAI

A PyTorch library and evaluation platform for end-to-end compression research

https://interdigitalinc.github.io/CompressAI/

BSD 3-Clause Clear License

1.19k stars 232 forks source link

Closed lbhm closed 3 years ago

lbhm commented 3 years ago

Directory structures like:

root
  - train
    - img01
    - ...
  - val
    - img01
    - ...

are currently not supported by the utils.bench module. I added an os.walk() to recursively build a file list.

Misc changes:

jbegaint commented 3 years ago

Hey, thanks for the PR. We'll probably want to propose some minor changes. I'll review it soon.

jbegaint commented 3 years ago

Thank you for the PR! We'll push the changes soon.