NVIDIA / ai-assisted-annotation-client

Client side integration example source code and libraries for AI-Assisted Annotation SDK
Other
308 stars 64 forks source link

[Feature request] Integrate predictions of "external" models into the annotation server #68

Closed MicPie closed 3 years ago

MicPie commented 4 years ago

Thank you for setting up the AI assisted annotation client!

I was searching through the docs but I was not able to find a way to use predictions from models that are not loaded onto the annotation server. (For bringing my own models I only found the option to upload them in the TensorFlow checkpoint or TensorRT format.)

Are there other options to incorporate predictions of "external" models into the annotation server?

Thank you very much & kind regards Michael

SachidanandAlle commented 4 years ago

You can do so... at-least PT (PyTorch) models.. https://docs.nvidia.com/clara/tlt-mi/clara-train-sdk-v3.0/aiaa/byom/byoi.html