augment_combined.py File Reference

Go to the source code of this file.

Namespaces
namespace	augment_combined

Functions
	augment_combined.load_vqvae_model (checkpoint_path, device)
	augment_combined.augment_images (model, input_dir, output_dir, device, apply_grain)

Variables
	augment_combined.True
	augment_combined.help
	augment_combined.action
	augment_combined.logger = get_logger("augment_combined")
	augment_combined.device = torch.device(args.device if torch.cuda.is_available() else 'cpu')
	augment_combined.model = load_vqvae_model(args.vqvae_ckpt, device)