There is a potential issue in utils/visualise at line 43 where test_slice is defined:
test_slice = random.randint(0, val_data["image"].shape[4])
This will create an IndexError if test_slice == val_data["image"].shape[4].
The fix is to adjust randint:
test_slice = random.randint(0, val_data["image"].shape[4]-1)
There is a potential issue in
utils/visualiseat line 43 wheretest_sliceis defined:This will create an
IndexErroriftest_slice == val_data["image"].shape[4].The fix is to adjust
randint: