get_preds_cyclegan: Get_preds_cyclegan

Description Usage Arguments Details

View source: R/cycleGAN_models.R

Description

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

Usage

1
2
3
4
5
6
7
8
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').


fastai documentation built on Oct. 25, 2021, 5:08 p.m.