augment_combined Namespace Reference

Functions
	main (input_dir, output_dir, checkpoint, config_path="configs/default.yaml")
	load_vqvae_model (checkpoint_path, device)
	augment_images (model, input_dir, output_dir, device, apply_grain)

Variables
	parser = argparse.ArgumentParser()
	required
	type
	str
	default
	args = parser.parse_args()
	True
	help
	action
	logger = get_logger("augment_combined")
	device = torch.device(args.device if torch.cuda.is_available() else 'cpu')
	model = load_vqvae_model(args.vqvae_ckpt, device)

Function Documentation

augment_images()

augment_combined.augment_images	(		model,
			input_dir,
			output_dir,
			device,
			apply_grain )

Definition at line 24 of file augment_combined.py.

load_vqvae_model()

augment_combined.load_vqvae_model	(		checkpoint_path,
			device )

Definition at line 16 of file augment_combined.py.

main()

augment_combined.main	(		input_dir,
			output_dir,
			checkpoint,
			config_path = "configs/default.yaml" )

Definition at line 10 of file augment_combined.py.

Variable Documentation

action

augment_combined.action

Definition at line 56 of file augment_combined.py.

args

augment_combined.args = parser.parse_args()

Definition at line 36 of file augment_combined.py.

default

augment_combined.default

Definition at line 35 of file augment_combined.py.

device

augment_combined.device = torch.device(args.device if torch.cuda.is_available() else 'cpu')

Definition at line 63 of file augment_combined.py.

help

augment_combined.help

Definition at line 52 of file augment_combined.py.

logger

augment_combined.logger = get_logger("augment_combined")

Definition at line 59 of file augment_combined.py.

model

augment_combined.model = load_vqvae_model(args.vqvae_ckpt, device)

Definition at line 64 of file augment_combined.py.

parser

augment_combined.parser = argparse.ArgumentParser()

Definition at line 31 of file augment_combined.py.

required

augment_combined.required

Definition at line 32 of file augment_combined.py.

str

augment_combined.str

Definition at line 35 of file augment_combined.py.

True

augment_combined.True

Definition at line 52 of file augment_combined.py.

type

augment_combined.type

Definition at line 35 of file augment_combined.py.