"""
Trainer to learn depth information on unlabeled data (raw images/videos)

Allows pluggable depth networks for differing performance (including fast-depth)
"""

import tensorflow.keras as keras


class SFMLearner(keras.Model):

    def __init__(depth_model, pose_model):
        pass

    def train_step(self, data):
        pass


def make_sfm_learner_pose_net(input_shape=(224, 224, 3)):
    pass