yeezhu / SPN.pytorch

PyTorch implementation of "Soft Proposal Networks for Weakly Supervised Object Localization", ICCV 2017.
http://yzhu.work/spn.html
MIT License
211 stars 37 forks source link

Python2 back compat #9

Closed vadimkantorov closed 6 years ago

vadimkantorov commented 6 years ago

On my machine python3 segfaults due some nasty import issues, but python2 still works.

Here are a couple minor fixes to make the code run on python2 if one wishes so:

# in demo/experiment/util.py
import sys
if sys.version_info.major > 2:
    from urllib.request import urlretrieve
else:
    from urllib import urlretrieve

# in demo/experiment/voc.py
import sys
if sys.version_info.major > 2:
    from urllib.parse import urlparse
else:
    from urlparse import urlparse
yeezhu commented 6 years ago

Thank you!