get_preds_cyclegan: Get_preds_cyclegan

In fastai: Interface to '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').

fastai documentation built on June 22, 2024, 11:15 a.m.