25 lines
667 B
Python
25 lines
667 B
Python
import pickle
|
|
import os.path
|
|
|
|
class Importer:
|
|
def __init__(self, config):
|
|
self.path = config["importPath"]
|
|
|
|
def import_raw_data(self):
|
|
print("Loading previous results")
|
|
|
|
layers = self.load_if_present(self.path + "_layers")
|
|
contours = self.load_if_present(self.path + "_contours")
|
|
masks = self.load_if_present(self.path + "_masks")
|
|
|
|
return layers, contours, masks
|
|
|
|
def load_if_present(self, path):
|
|
var = None
|
|
if os.path.isfile(path):
|
|
with open(path, "rb") as file:
|
|
var = pickle.load(file)
|
|
else:
|
|
print(path, "file not found")
|
|
return var
|