search

DefTruth / RVM-Inference

🔥Robust Video Matting C++ inference toolkit with ONNXRuntime、MNN、NCNN and TNN, via lite.ai.toolkit.
GNU General Public License v3.0
116 stars 27 forks source link

cuda运行方式出来的是错误结果？ #35

Open CoderJackLiu opened 2 years ago

CoderJackLiu commented 2 years ago

同样的代码我使用CPU和GPU方式去运行modle出来了两种结果，CPU是正常的，但是GPU的Alpha数据是空的，前景是错误的，当然我们只关注Alpha的结果。 结果如下： image image

这是什么原因造成的呢？