UNET Training on Multi-Class Segmentation from Satellite Imagery (DSTL)

Accelerated Computing Intelligent Video Analytics TAO Toolkit
8

The 3.21.11 release supports

["resnet", "vgg", "vanilla_unet", "efficientnet_b0",
  "vanilla_unet_dynamic"]

You can set vanilla_unet_dynamic.

1 Like