Test only 27 images

[HounaidaM]

@onurboyar

Thank you for your code, it is beneficial to me.

I have my private dataset, and my U-net code runs correctly (train and test) on the full dataset.
I want to test the model on only 27 images. When I execute evaluate function, I meet this error :

**InvalidArgumentError: 2 root error(s) found.
(0) Invalid argument: slice index 1 of dimension 2 out of bounds.
[[{{node strided_slice_1}}]]
[[IteratorGetNext]]
[[IteratorGetNext/_4]]
(1) Invalid argument: slice index 1 of dimension 2 out of bounds.
[[{{node strided_slice_1}}]]
[[IteratorGetNext]]
0 successful operations.
0 derived errors ignored. [Op:__inference_test_function_12429]

Function call stack:
test_function -> test_function**

Can you please help me?
thank you