zllrunning / face-parsing.PyTorch

Using modified BiSeNet for face parsing in PyTorch
MIT License
2.33k stars 457 forks source link

How to get probability prediction ? #34

Open martinhoang11 opened 3 years ago

martinhoang11 commented 3 years ago

It's nice github repo, Could you please tell me how can i get probability prediction at the and with parsing labels ? Thank you