mirror of https://github.com/davisking/dlib.git
49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
import pkgutil
|
|
import sys
|
|
|
|
def save_pickled_compatible(obj_to_pickle, file_name):
|
|
'''
|
|
Save an object to the specified file in a backward compatible
|
|
way for Pybind objects. See:
|
|
http://pybind11.readthedocs.io/en/stable/advanced/classes.html#pickling-support
|
|
and https://github.com/pybind/pybind11/issues/271
|
|
'''
|
|
try:
|
|
import cPickle as pickle # Use cPickle on Python 2.7
|
|
except ImportError:
|
|
import pickle
|
|
data = pickle.dumps(obj_to_pickle, 2)
|
|
with open(file_name, "wb") as handle:
|
|
handle.write(data)
|
|
|
|
def load_pickled_compatible(file_name):
|
|
'''
|
|
Loads a pickled object from the specified file
|
|
'''
|
|
try:
|
|
import cPickle as pickle # Use cPickle on Python 2.7
|
|
except ImportError:
|
|
import pickle
|
|
|
|
with open(file_name, "rb") as handle:
|
|
data = handle.read()
|
|
if not is_python3():
|
|
return pickle.loads(data)
|
|
else:
|
|
return pickle.loads(data, encoding="bytes")
|
|
|
|
def is_numpy_installed():
|
|
'''
|
|
Returns True if Numpy is installed otherwise False
|
|
'''
|
|
if pkgutil.find_loader("numpy"):
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def is_python3():
|
|
'''
|
|
Returns True if using Python 3 or above, otherwise False
|
|
'''
|
|
return sys.version_info >= (3, 0)
|