import matplotlib.pyplot as plt
from DataProcessor import DataProcessor
plt.rcParams['image.cmap'] = 'gist_earth'
plugin_config = "./config/config.json"
type_of_data = "multi"
# os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
dataprocessor_v2 = DataProcessor(plugin_config,
base_model_name="v1",
model_name="v2",
image_dir="v2")
dataprocessor_v2.execute()
net_v2, trainer_v2, operators_v2 = dataprocessor_v2.get_model(type_of_data)
dataprocessor_v2.validate(trainer_v2,
operators_v2,
display_step=2,
restore=True)
number_of_models = dataprocessor_v2.get_total_numberof_model_count(trainer_v2)
dataprocessor_v2.evalfscore_v12(trainer_v2, operators_v2, number_of_models)
