main() problem.

[zhangbaijin]

Found 0 images belonging to 0 classes.
Traceback (most recent call last):
File "/home//PConv-Keras-master/main.py", line 180, in
test_data = next(test_generator)
File "/home//PConv-Keras-master/main.py", line 122, in flow_from_directory
for _ in range(ori.shape[0])], axis=0
File "/home/zxf/anaconda3/envs/tensorflow/lib/python3.6/site-packages/numpy/core/shape_base.py", line 412, in stack
raise ValueError('need at least one array to stack')
ValueError: need at least one array to stack

excuse me . I follow the comand python main.py --name celeba --train ./data/celeba/train --validation ./data/celeba/val --test ./data/celeba/test --vgg_path './data/logs/pytorch_to_keras_vgg16.h5'
but there is no found files about mask and image, why?? i put masks dataset under ./data/masks/train/

[jiunyen-ching]

Are your images placed in a sub-folder in each train/test/val folder?
I assume your images are now placed like this:
Train: "./data/train/.png"
Test: "./data/test/.png"
Val: "./data/val/*.png"

where they should be placed like this
Train: "./data/train/subdir_1/.png"
Test: "./data/test/subdir_1/.png"
Val: "./data/val/subdir_1/*.png"

I guess number of subdirs in each train/test/val depends on your number of classes