zllrunning / face-parsing.PyTorch

Using modified BiSeNet for face parsing in PyTorch
MIT License
2.32k stars 456 forks source link

Where is the class label documentation? #55

Open weiweiwang opened 6 months ago

weiweiwang commented 6 months ago

@zllrunning Sorry to disturb, could you please provide the 19 class labels documentation in your readme?

Thanks very much!

renhaa commented 5 months ago

@weiweiwang @zllrunning
Look at https://github.com/zllrunning/face-parsing.PyTorch/blob/d2e684cf1588b46145635e8fe7bcc29544e5537e/test.py#L16 and compare with this list

   ["background",'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'] 

Should be correct, no?