lifeisawar41

I also face this problem ,I tried to change the defination of the class Image_dataset_folder, but I don’t know how to fix it

I don't know if the data_transform in 120 rows in pytorch_run.py can be useful for the class Image_dataset_folder

Oh! thank you , the program seems to work. I have tried this before, haha, maybe it's because I commented out the code that converts label data to grayscale