CVxTz/medical_image_segmentation

Prediction on Custom Dataset

Closed this issue · 1 comments

@CVxTz

Hello Youness,
How can I run a prediction on raw .jpg custom datasets? Thanks in advance.

CVxTz commented

you can change the prediction .py files so it looks for jpg images here :

val_data = list(zip(sorted(glob('../input/DRIVE/test/images/*.**tif**')), => here change the extention
                      sorted(glob('../input/DRIVE/test/2nd_manual/*.**gif**')),
                    sorted(glob('../input/DRIVE/test/mask/*.**gif**'))))

If you dont have the ground truth you might have to change other parts of the code.