search

tinyvision / SOLIDER-HumanParsing

MIT License
16 stars 6 forks source link

KeyError: 'state_dict' #2

Open sebastianopazo1 opened 11 months ago

sebastianopazo1 commented 11 months ago

Hi! I was trying to use simple_extractor.py and got this error: python simple_extractor.py --input-dir /home/seba/Documents/SOLIDER/SOLIDER-HumanParsing/input --output-dir /home/seba/Documents/SOLIDER/SOLIDER-HumanParsing/output Evaluating total class number 20 with ['Background', 'Hat', 'Hair', 'Glove', 'Sunglasses', 'Upper-clothes', 'Dress', 'Coat', 'Socks', 'Pants', 'Jumpsuits', 'Scarf', 'Skirt', 'Face', 'Left-arm', 'Right-arm', 'Left-leg', 'Right-leg', 'Left-shoe', 'Right-shoe'] Traceback (most recent call last): File "/home/seba/Documents/SOLIDER/SOLIDER-HumanParsing/simple_extractor.py", line 155, in main() File "/home/seba/Documents/SOLIDER/SOLIDER-HumanParsing/simple_extractor.py", line 106, in main state_dict = torch.load(args.model_restore)['state_dict'] KeyError: 'state_dict'

Thanks!

cedricnam commented 4 months ago

Hi! I was trying to use simple_extractor.py and got this error: python simple_extractor.py --input-dir /home/seba/Documents/SOLIDER/SOLIDER-HumanParsing/input --output-dir /home/seba/Documents/SOLIDER/SOLIDER-HumanParsing/output Evaluating total class number 20 with ['Background', 'Hat', 'Hair', 'Glove', 'Sunglasses', 'Upper-clothes', 'Dress', 'Coat', 'Socks', 'Pants', 'Jumpsuits', 'Scarf', 'Skirt', 'Face', 'Left-arm', 'Right-arm', 'Left-leg', 'Right-leg', 'Left-shoe', 'Right-shoe'] Traceback (most recent call last): File "/home/seba/Documents/SOLIDER/SOLIDER-HumanParsing/simple_extractor.py", line 155, in main() File "/home/seba/Documents/SOLIDER/SOLIDER-HumanParsing/simple_extractor.py", line 106, in main state_dict = torch.load(args.model_restore)['state_dict'] KeyError: 'state_dict'

Thanks!

I got same failed