get_preds_cyclegan {fastai}

Get_preds_cyclegan

Description

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

Usage

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