File size: 896 Bytes
59630ba | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | from datasets.video import (
MinecraftSimpleVideoDataset,
)
from algorithms.vae import ImageVAEPreprocessor
from .base_exp import BaseLightningExperiment
from .data_modules import ValDataModule
class VideoLatentPreprocessingExperiment(BaseLightningExperiment):
"""
Experiment for preprocessing videos to latents using a pretrained ImageVAE model.
"""
compatible_algorithms = dict(
image_vae_preprocessor=ImageVAEPreprocessor,
)
compatible_datasets = dict(
video_minecraft=MinecraftSimpleVideoDataset,
)
data_module_cls = ValDataModule
def training(self) -> None:
raise NotImplementedError(
"Training not implemented for video preprocessing experiments"
)
def testing(self) -> None:
raise NotImplementedError(
"Testing not implemented for video preprocessing experiments"
)
|