Abstract: Certain aspects of the present disclosure provide techniques for machine learning-based deblurring. An input image is received, and a deblurred image is generated based on the input image using a neural network, comprising: generating a feature tensor by processing the input image using a first portion of the neural network, generating a motion mask by processing the feature tensor using a motion portion of the neural network, and generating the deblurred image by processing the feature tensor and the motion mask using a deblur portion of the neural network.