search

GoGoDuck912 / Self-Correction-Human-Parsing

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

Error while running simple_extractor.py #53

Closed soumyajit123dev closed 3 years ago

soumyajit123dev commented 3 years ago

I ran the following command

python simple_extractor.py --dataset 'lip' --input-dir 'images' --model-restore 'pretrained/exp-schp-201908261155-lip.pth' --output-dir 'output'

input image is kept inside images folder pretrained model is kept inside pretrained folder

I am getting some errors. I do not know how to resolve these errors. Please see the errors below

Screenshot from 2021-02-19 12-23-19 Screenshot from 2021-02-19 12-45-54 :

Let me know what is required to solve the errors

jutanke commented 3 years ago

@soumyajit123dev resetting pytorch to an older version solved the issue on my end: conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.2 -c pytorch

GoGoDuck912 commented 3 years ago

Please try to run the CoLab notebook following https://colab.research.google.com/drive/1JOwOPaChoc9GzyBi5FUEYTSaP2qxJl10?usp=sharing. If this works for you, basically the issue is related to the environment configuration based on your circumstances.