ptrblck:

ImageFolder


the problem is, the file name in test directory are numbers. I got this DataSet from Kaggle, and it is not easy to separate test data set into two category. I think the first solution you offered is more suitable in this case