search

Xilinx / Vitis-AI-Tutorials

MIT License
381 stars 145 forks source link

Using DPU as a feature extractor #17

Open Afef00 opened 3 years ago

Afef00 commented 3 years ago

Hello, I was wondering is it possible to use the DPU as a feature extractor? how can I retrieve and copy the final feature vectors (the last feature map) ? I tried this in the main cc file: 

feature = dpuGetOutputTensorInHWCFP32(taskResnet50, OUTPUT_NODE, FCResult, channel);
printf("features = %f\n\r", feature);

But I only obtained one value.

Any help please?