Faster Tensorflow inference step with tf.function.

PeterL1n / BackgroundMattingV2-TensorFlow

TensorFlow implementation of Real-Time High-Resolution Background Matting

MIT License

78 stars 13 forks source link

Faster Tensorflow inference step with tf.function. #1

Closed dathudeptrai closed 3 years ago

dathudeptrai commented 3 years ago

Hi, thanks for your TensorFlow supported model. I just correct the right way to inference the TensorFlow model with tf.function. It should speedup around 2-3 times.

PeterL1n commented 3 years ago

Merged. Thanks for pointing that out.