face-parsing.PyTorch icon indicating copy to clipboard operation
face-parsing.PyTorch copied to clipboard

train classes mis-leading..

Open sycophant-stone opened this issue 3 years ago • 4 comments

Since there are 17 classes1 in the prepare stags, while there are 19 classes in pretrained model from test.py.2 So,my concern is what is the differences between those two models?

sycophant-stone avatar Sep 28 '20 08:09 sycophant-stone

they said that it contains last background part, but I test result found the class name make more confused parsing_torch

and this is my label order: ['skin', 'l_brow', 'r_brow', 'l_eye', 'r_eye', 'eye_g', 'l_ear', 'r_ear', 'ear_r', 'nose', 'mouth', 'u_lip', 'l_lip', 'neck', 'neck_l', 'cloth', 'hair', 'hat', 'bg']

showkeyjar avatar Oct 10 '21 02:10 showkeyjar

https://github.com/switchablenorms/CelebAMask-HQ/tree/master/face_parsing#well-trained-model

Ir1d avatar Nov 25 '21 03:11 Ir1d

Background is first class. I tested this model (the one that is mentioned in README.md) and found this labeling:

['bg', 'skin', 'l_brow', 'r_brow', 'l_eye', 'r_eye', 'eye_g', 'l_ear', 'r_ear', 'ear_r', 'nose', 'mouth', 'u_lip', 'l_lip', 'neck', 'neck_l', 'cloth', 'hair', 'hat']

https://github.com/switchablenorms/CelebAMask-HQ/tree/master/face_parsing#well-trained-model

This label is not suitable for this model

fortunechen avatar Oct 31 '23 07:10 fortunechen