fix load pickle for python2 and python3

mjirik · Apr 18, 2018 · 36cdb14 · 36cdb14
1 parent 64f3508
commit 36cdb14
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/io3d/misc.py b/io3d/misc.py
@@ -83,7 +83,10 @@ def obj_from_file(filename='annotation.yaml', filetype='auto'):
         else:
             import _pickle as pickle
         # import sPickle as pickle
-        obj = pickle.loads(fcontent, encoding="latin1")
+        if sys.version_info.major == 2:
+            obj = pickle.loads(fcontent)
+        else:
+            obj = pickle.loads(fcontent, encoding="latin1")
     else:
         logger.error('Unknown filetype ' + filetype)
     return obj