GoGoDuck912 / Self-Correction-Human-Parsing

An out-of-box human parsing representation extractor.
MIT License
1.06k stars 237 forks source link

AssertionError when evaluating with batch size > 1 #57

Open AlkaSaliss opened 3 years ago

AlkaSaliss commented 3 years ago

Hello Thanks for the awesome code I've tried running the evaluation script using batch size > 1, however there's one assertion that failed in that case , it is line : https://github.com/PeikeLi/Self-Correction-Human-Parsing/blob/eb84c432cc697f494d99662a05f2335eb2f26095/evaluate.py#L202

I printed the content of parsing_preds variable and it seems to contain only len(testloader) elements instead of len(lip_test_dataset). I think something must be broken in the pipeline