Skip to content

DeepEdit

Andres Diaz-Pinto edited this page Jul 5, 2021 · 22 revisions

DeepEdit is an algorithm that combines the power of two models in one single architecture. It allows the user to perform inference, as a standard segmentation method (i.e. UNet), and also to interactively segment part of an image using clicks (Sakinis et al.). DeepEdit aims to facilitate the user experience and to facilitate the development of new active learning techniques.

Training schema:

The training process of a DeepEdit App involves a combination of simulated clicks and standard training. As shown in the next figure, the input of the network is a concatenation of three tensors: image, positive (foreground) and negative (background) points or clicks. This model has two types of training: For some iterations, tensors representing the foreground and background points are zeros and for other iterations, positive and negative clicks are simulated so the model can receive inputs for interactive segmentation. For the click simulation, users can take advantage of the already developed transforms and engines in MONAI.

DeepEdit Schema for Training

Inference schema:

DeepEdit Schema for Testing

Clone this wiki locally