Add load dict function

Rye74 · Apr 7, 2020 · 229fdcb · 229fdcb
1 parent 7d8797d
commit 229fdcb
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/utils/image_utils.py b/utils/image_utils.py
@@ -18,6 +18,10 @@ def is_png_file(filename):
 def is_pkl_file(filename):
     return any(filename.endswith(extension) for extension in [".pkl"])
 
+def load_dict(filename_):
+    with open(filename_, 'rb') as f:
+        ret_dict = pickle.load(f)
+    return ret_dict
 
 def load_pkl(filename_):
     with open(filename_, 'rb') as f: