训练单通道ir图报错！！！，要改哪些地方？

minivision-ai / Silent-Face-Anti-Spoofing

静默活体检测（Silent-Face-Anti-Spoofing）

Apache License 2.0

1.36k stars 453 forks source link

Open Aruen24 opened 2 years ago

Aruen24 commented 2 years ago

RuntimeError: Given groups=1, weight of size [32, 1, 3, 3], expected input[128, 224, 224, 3] to have 1 channels, but got 224 channels instead

zhuyingSeu commented 2 years ago

change the cv2.imread mode (src/data_io/dataset_folder.py, cv2.imread(image_path, 0)).