spmallick / learnopencv

Learn OpenCV : C++ and Python Examples
https://www.learnopencv.com/
21.36k stars 11.62k forks source link

how to deploy PyTorch-Multi-Label-Image-Classification model in triton server #941

Open liulan93 opened 3 months ago

liulan93 commented 3 months ago

I have trained a multi label(my own data with three properties, three labels will be given for every inference) model using code of PyTorch-Multi-Label-Image-Classification, and then convert the model to onnx format. Now I just want to deploy the model with Triton Inference Server, how should I do? and if deployed, how should I do to place the labels of properties in the model and do inference with it? Thanks.