PeterL1n / BackgroundMattingV2

Real-Time High-Resolution Background Matting
MIT License
6.84k stars 953 forks source link

I can't find convGRU in decoder.py #181

Closed NingNanXin closed 2 years ago

NingNanXin commented 2 years ago

Hi, thanks for your good job! But when I read the article and the code. I can't find where to use the convGRU?

image image
  1. Does the model not use the convGRU?
  2. What role does sampling plays in refiner.py? I try to export tensorrt but the topk is out of range because tensorrt only supports topk with the number 3084. So I use the threshold type. What is the difference between the sampling and threshold?

Thanks for your help!

PeterL1n commented 2 years ago

You are in the wrong repo.

RVM paper repo is here: https://github.com/PeterL1n/RobustVideoMatting