midasmodel module

class selective_blur.blurring.midasmodel.MidasModel(model: str = 'DPT_Hybrid')

Bases: object

Class to load Intel MiDaS model and make predictions

predict_adjust(image: ndarray) → ndarray

Parameters:

image (np.ndarray)

Returns:

depth map of the image

Return type:

np.ndarray