Kid-Liet / Reg-GAN

149 stars 21 forks source link

about how to process my own dataset for this project #8

Closed 445193519 closed 2 years ago

445193519 commented 2 years ago

Can you please describe the features in the dataset processing process, I am training with my own dataset, but the dataset format is not correct. I've converted the png image to npy format, but I can't understand why the original dataset is two-dimensional

Kid-Liet commented 2 years ago

We use a single channel dataset by default. That is, the dimension of the original data is (W, H). If your default original data format is (1, W, H) or (3, W, H), you need to comment out 33 lines in utils.py.

image