search

dusty-nv / jetson-inference

Hello AI World guide to deploying deep-learning inference networks and deep vision primitives with TensorRT and NVIDIA Jetson.
https://developer.nvidia.com/embedded/twodaystoademo
MIT License
7.89k stars 2.99k forks source link

Change network in background removal #1786

Open Hogushake opened 10 months ago

Hogushake commented 10 months ago

hello. I am trying to do background removal using backgroundnet in jetson inference. There is a command line to change the network. Can I use the network I want? (ex.. u2netp)

dusty-nv commented 10 months ago

hi @Hogushake, if the model is in ONNX format and expects the same inputs/outputs, theoretically yes you could try it. You may have to adapt the pre/post-processing in c/backgroundNet.cpp if that model expects different.

Hogushake commented 10 months ago

@dusty-nv Thanks for the quick reply. I'll try it and let you know!