Limit scan in generate_meta_info_pairdata.py to image types

muslll / neosr

neosr is a framework for training real-world single-image super-resolution networks.

https://github.com/muslll/neosr

Apache License 2.0

136 stars 28 forks source link

Limit scan in generate_meta_info_pairdata.py to image types #43

Closed Mikewando closed 6 months ago

Mikewando commented 6 months ago

Sometimes non-image files can sneak into dataset folders (like desktop.ini on windows) which can disrupt training. Therefore this change limits the scan when generating a paired dataset to only consider known image type extensions.

Mikewando commented 6 months ago

re-opens https://github.com/muslll/neosr/pull/26 against dev