02-18-2021 02:59 AM - edited 02-18-2021 03:21 AM
I am trying to run tiny_yolov3_416 with DpuTask::create on VAI 1.3, but I don't know how to implement DpuTask functions for calling multiple subgraphs.
After compiling, there are two sugraphs of tiny_yolov3_416 and I am not sure how to implement them.
The following example does not work:
auto task = vitis::ai::DpuTask::create(kernel_name) // kernel_name = "/usr/share/...../tiny_yolov3_416.xmodel"
task->run(0u); // run only subgraph
auto output_tensor = task->getOutputTensor(1u); // get output tensors from subgraph
I think I need to get output tensors from sugraph and set them as input tensors of subgraph and run subgraph but I don't know how to do it.
Any help would be appreciated.
02-22-2021 03:50 PM
Are you saying that you have 2 DPU subgraphs? I would not expect that for TinyYolov3.
Can you share the compiler output.
If you are converting from Darknet to Caffe, there is a layer change you need to make to compatible with the DPU.
02-23-2021 12:57 AM