daijifeng001 / MNC

Instance-aware Semantic Segmentation via Multi-task Network Cascades

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

cpu_mask_voting v.s. gpu_mask_voting

wk910930 opened this issue · comments

Dear @oh233 ,

From the function names, I thought the functions cpu_mask_voting and gpu_mask_voting are doing the same things but on different devices (i.e. CPU and GPU). However, the their implementations are a little bit different.

Just to be sure, what is the difference between these two functions?

Thanks!
Kun

I've been doing inference with a trained model and, as far as I can tell right now, replacing one for another doesn't work for me right away.