gnes.encoder.image.cvae module

class gnes.encoder.image.cvae.CVAEEncoder(model_dir: str, latent_dim: int = 300, select_method: str = 'MEAN', l2_normalize: bool = False, use_gpu: bool = True, *args, **kwargs)[source]

Bases: gnes.encoder.base.BaseImageEncoder

batch_size = 64
encode(img: List[np.ndarray], *args, **kwargs) → numpy.ndarray[source]
post_init()[source]

Declare class attributes/members that can not be serialized in standard way

train(*args, **kwargs)

Train the model, need to be overrided