PhysAugNet 1.0.1
VQ-VQE powered augmentation for metal defect segmentation
Loading...
Searching...
No Matches
gen_vqvae Namespace Reference

Functions

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

Variables

 parser = argparse.ArgumentParser()
 required
 type
 str
 default
 args = parser.parse_args()

Function Documentation

◆ main() [1/2]

gen_vqvae.main ( )

Definition at line 19 of file gen_vqvae.py.

◆ main() [2/2]

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

Definition at line 6 of file gen_vqvae.py.

◆ parse_args()

gen_vqvae.parse_args ( )

Definition at line 9 of file gen_vqvae.py.

Variable Documentation

◆ args

gen_vqvae.args = parser.parse_args()

Definition at line 19 of file gen_vqvae.py.

◆ default

gen_vqvae.default

Definition at line 18 of file gen_vqvae.py.

◆ parser

gen_vqvae.parser = argparse.ArgumentParser()

Definition at line 14 of file gen_vqvae.py.

◆ required

gen_vqvae.required

Definition at line 15 of file gen_vqvae.py.

◆ str

gen_vqvae.str

Definition at line 18 of file gen_vqvae.py.

◆ type

gen_vqvae.type

Definition at line 18 of file gen_vqvae.py.