search

GoGoDuck912 / Self-Correction-Human-Parsing

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

Add support to devices without CUDA #50

Open samuelesabella opened 3 years ago

heyJatin commented 2 years ago

Thanks a lot for these changes. It saved me a lot of time!

belonel commented 1 year ago

@samuelesabella Thanks a lot for your contribution! I tried to run your branch in google collab and faced the import error. I haven't been able to figure out how to fix this. Do you have any suggestions?

Traceback (most recent call last): File "/content/Self-Correction-Human-Parsing/simple_extractor.py", line 24, in import networks File "/content/Self-Correction-Human-Parsing/networks/init.py", line 3, in from .AugmentCE2P import resnet101 File "/content/Self-Correction-Human-Parsing/networks/AugmentCE2P.py", line 21, in from ..modules import InPlaceABNSync ImportError: attempted relative import beyond top-level package

EgorOs commented 1 year ago

Thank you very much @samuelesabella ! @belonel I managed to load checkpoint in Colab, here is an example https://colab.research.google.com/drive/1Rdm30i9ZuJEfTYMnFb-PRXWLkE6O_NVJ?usp=sharing