Get_preds_cyclegan — get_preds

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').