A prediction function that takes the Learner object `learn` with the trained model, the `test_path` folder with the images to perform

get_preds_cyclegan(
  learn,
  test_path,
  pred_path,
  bs = 4,
  num_workers = 4,
  suffix = "tif"
)

Arguments

learn

learner/model

test_path

testdat path

pred_path

predict data path

bs

batch size

num_workers

number of workers

suffix

suffix

Details

batch inference on, and the output folder `pred_path` where the predictions will be saved, with a batch size `bs`, `num_workers`, and suffix of the prediction images `suffix` (default='png').