The multi-task library is appropriate for a model that has multiple sub-tasks. The Multi-task model in the Vitis AI Library has two sub-tasks: semantic segmentation and SSD detection. The following table listss the multi-task models supported by the Vitis AI Library.
| Number | Model Name | Framework |
|---|---|---|
| 1 | multi_task | Caffe |
| 2 | MT-resnet18_mixed_pt | PyTorch |