ceccocats / tkDNN

Deep neural network library and toolkit to do high performace inference on NVIDIA jetson platforms
GNU General Public License v2.0
720 stars 209 forks source link

[TensorRT-8] Add independent padding and depth NN #278

Closed perseusdg closed 2 years ago

perseusdg commented 2 years ago

This pr adds a padding layer with three main padding type - reflection(tested),constant(untested) and zero(untested) using plugins,using TensorRT's IPaddingLayer and ISliceLayer can be explored for a future release and also adds a monocular depth neural network (monodepth2) with help from @mive93 and @ceccocats but copy rights and licensing might be an issue ref : https://github.com/nianticlabs/monodepth2/issues/316 and https://github.com/nianticlabs/monodepth2/issues/384 ,it might involve adding nianticlabs copyright to the license file (I am not sure about this ) .

This pr also has general bug fixes for windows and slight performance improvements for neural networks that use an upsample layer.

mive93 commented 2 years ago

Merging soon depth