could not broadcast input array from shape (96,96,3) into shape (96,96)

[m1081706]

I use my own dataset.
The data image size is 96*96.
When I use this dataset, I saw this problem.
I use "python main.py --dataset dataset --output_width=96 --output_height=96 --input_width=96 --input_height=96 --train=False --batch_size 64 --num_test 50"run it.
How can I solve it?
Thank you