2DUnet - 3.5 English

Vitis AI Library User Guide (UG1354)

Document ID
Release Date
3.5 English

UNet is the most commonly used and simplest segmentation model. It is simple, efficient, easy to understand, easy to build, and can be trained from small datasets. The original intention of UNet is to solve the problem of medical image segmentation. In terms of solving the task of cell-level segmentation, it won multiple firsts in the ISBI cell tracking competition in 2015. After that, UNet has been widely used in various directions of semantic segmentation (such as satellite image segmentation, industrial defect detection, etc.) due to its outstanding segmentation effect.

The following table lists the 2DUnet model supported by the Vitis AI Library.

Table 1. 2DUnet Model
No Model Name Framework
1 unet2d_tf2 Tensorflow2