TOPK operator for RKNN export

PINTO0309 / PINTO_model_zoo

A repository for storing models that have been inter-converted between various frameworks. Supported frameworks are TensorFlow, PyTorch, ONNX, OpenVINO, TFJS, TFTRT, TensorFlowLite (Float32/16/INT8), EdgeTPU, CoreML.

https://qiita.com/PINTO

MIT License

3.43k stars 560 forks source link

TOPK operator for RKNN export #390

Closed ax-anoop closed 5 months ago

ax-anoop commented 5 months ago

Issue Type

Support

OS

Ubuntu

OS architecture

x86_64, aarch64

Programming Language

Python

Framework

ONNX

Model name and Weights/Checkpoints URL

movenet multi

Description

Is there anyway to swap out the topK operator ?

I am trying to export to RKNN; however, this operator doesn't seem to be supported.

Relevant Log Output

No response

URL or source code for simple inference testing code

No response

PINTO0309 commented 5 months ago

I am not interested in posts that have nothing to do with this repository.