A system for improving imaging during a surgical procedure includes an image sensor coupled to a controller. The image sensor is coupled to capture a video, including a first image frame with a smoke occlusion due to surgical smoke, of the surgical procedure. The first image frame includes a plurality of pixels each having an imaged color representing a view of the tissue affected by the smoke occlusion. The controller stores instructions that when executed causes the system to perform operations. The operations include determining an estimated true color of each of the plurality of pixels. The estimated true color is closer to an unoccluded color of the tissue than the imaged color. The operations also include generating, in response to determining the estimated true color, a desmoked first image frame with a reduced amount of the smoke occlusion relative to the first image frame.