PhysAugNet 1.0.1
VQ-VQE powered augmentation for metal defect segmentation
Loading...
Searching...
No Matches
vqvae.VQVAE Member List

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
decodervqvae.VQVAE
encodervqvae.VQVAE
forward(self, x)vqvae.VQVAE
vqvqvae.VQVAE