def get_tra_ori(): img_input=Input(shape=(None, None, 1)) theta = Lambda(orientation)(img_input) model = Model(inputs=[img_input,], outputs=[theta,]) return model