|
PhysAugNet 1.0.1
VQ-VQE powered augmentation for metal defect segmentation
|
This is the complete list of members for vqvae.VQVAE, including all inherited members.
| __init__(self, in_channels=3, embedding_dim=64, num_embeddings=512) | vqvae.VQVAE | |
| decoder | vqvae.VQVAE | |
| encoder | vqvae.VQVAE | |
| forward(self, x) | vqvae.VQVAE | |
| vq | vqvae.VQVAE |