Hello, I trained a u-net to remove jpeg artefacts from an image.
I exported the u-net to get the learner for inference.
learn.predict(IMG) works fine but I don’t want to use this function to predict one by one the images from a folder.
I tried to use learn.get_preds(ds_type=DatasetType.Test), but I don’t get why the predicted results are completely different from the results of the simple predict.
imgList = ImageImageList.from_folder(path / "Apple_vhs_test/", presort=True)
learn = load_learner("/models/", test=imgList [:1], tfm_y=False)
p = learn.get_preds(ds_type=DatasetType.Test)
Image(p[0][0]).show(figsize=(10,10))
fn = 'dataset/.../apple_VHS_0.jpg'
img_vhs = open_image(fn)
p,img_hr,b = learn.predict(img_vhs)
Is there anything I can do to make get_preds
behave as predict
?