AssertionError while running evaluate.py on ratinal data set

print(patches_imgs_test.shape) (214200, 64, 64, 1)

predicted images size : (214200, 64, 64, 1)

if average_mode == True:
    pred_imgs = **recompone_overlap**(pred_patches, new_height, new_width, stride_height, stride_width)# predictions
    orig_imgs = my_PreProc(test_imgs_orig[0:pred_imgs.shape[0],:,:,:])    #originals
    gtruth_masks = masks_test  #ground truth masks

else:
    pred_imgs = recompone(pred_patches,13,12)       # predictions
    orig_imgs = recompone(patches_imgs_test,13,12)  # originals
    gtruth_masks = recompone(patches_masks_test,13,12)**

**Traceback (most recent call last):

  File "C:\Users\BIOIMA~1\AppData\Local\Temp/ipykernel_25740/3778051638.py", line 2, in <module>
    pred_imgs = **recompone_overlap**(pred_patches, new_height, new_width, stride_height, stride_width)# predictions

  File "C:\Users\Bioimaging\shrirang_IPW\conference\data\DRIVE\BCDU-Net-master\Retina Blood Vessel Segmentation\extract_patches.py", line 266, in recompone_overlap
    assert (preds.shape[1]==1 or preds.shape[1]==3)  #check the channel is 1 or 3

AssertionError**

I think here the error is occuring because of this >> recompone_overlap

rezazad68 / BCDU-Net

AssertionError while running evaluate.py on ratinal data set #44