21 lines
402 B
Python
21 lines
402 B
Python
"""
|
|
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
|