import os
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_v3 = DataProcessor(plugin_config,
base_model_name="v1",
model_name="v3",
image_dir="v3",
is_final=True)
dataprocessor_v3.execute()
net_v3, trainer_v3, operators_v3 = dataprocessor_v3.get_model(type_of_data)
dataprocessor_v3.validate(trainer_v3,
operators_v3,
display_step=2,
restore=True)
number_of_models = dataprocessor_v3.get_total_numberof_model_count(trainer_v3)
dataprocessor_v3.evalfscore_v16(trainer_v3, operators_v3, number_of_models)
