save_trt.py

[eyebies]

Thanks for sharing the code !Would it be possible to upload the save_trt file as well ?
thanks!

[Phan Hoang]

Hi @eyebies ,
Currently, I don't have this script anymore. But you can convert this DB model to TensorRT format by using trtexec

Note, if used F.interpolate function, you must change the mode to "bilinear" instead of the current default mode "nearest" to avoid wrong output mask from Trt model (even though the model converts successfully)!