Bering.training.TrainerEdge.predict

TrainerEdge.predict(batch_data, image)[source]

Predict the edge labels from the input data

Parameters:

batch_data – Input data
image – image as the input for the image encoder