Use topK by attaching it to the model

General
1

Hi,

I am trying to attach a topK filtering to an object detection model.
I tried with simple topK from torch but the parser in Dataflow Compiler gives me an error.
What is the best way to implement it so that it can be compiled for the hailo8L?

2

Hi @Pantos
Can you explain your use a bit more? Generally topK is used in classification models (though some detection models that do not use NMS use topK). Is the model you have a standard object detection model (like yolov8 etc)?