File size: 642 Bytes
27ca8b3 |
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 |
from datasets.video import (
MinecraftVideoDataset,
MinecraftVideoPoseDataset
)
from algorithms.worldmem import WorldMemMinecraft
from algorithms.worldmem import PosePrediction
from .exp_base import BaseLightningExperiment
class VideoPredictionExperiment(BaseLightningExperiment):
"""
A video prediction experiment
"""
compatible_algorithms = dict(
df_video_worldmemminecraft=WorldMemMinecraft,
pose_prediction=PosePrediction
)
compatible_datasets = dict(
# video datasets
video_minecraft=MinecraftVideoDataset,
video_minecraft_pose=MinecraftVideoPoseDataset
)
|