search

lizhengwei1992 / Semantic_Human_Matting

Semantic Human Matting
527 stars 145 forks source link

is possible execute knn-matting.py with GPU ? it takes too much time #50

Open LoveU3tHousand2 opened 3 years ago

LoveU3tHousand2 commented 3 years ago

Can I get the GPU version of knn-matting ?
my: Ubuntu 18.04 RTX3070 only one card cuda 11.2.1 for nvidia driver 460.32.03 opencv 4.8.3 pytorch. 1.7