dataset load error "FileNotFoundError"

[kwanwoo02]

Hi Thanks for revealing the source code.
I'm trying to use sketch2color.

However, an error occurs while loading the dataset. Can you tell me what the structure of the dataset is?

[kwanwoo02]

in my case

├── dataset
│   ├── train
│          ├── 1.png
│          ├── ···
│          └── 1000.png
│   └── val
│          ├── 1.png
│          ├── ···
│          └── 1000.png

ex. 1.png