class COVID19DN():
""" COVID19DN Class
Core COVID-19 Tensorflow DenseNet Classifier wrapper class using Tensroflow 2.
"""
def __init__(self):
""" Initializes the class. """
self.Helpers = Helpers("Core")
self.Model = Model()
self.Helpers.logger.info(
"COVID19DN Tensorflow initialization complete.")
def do_data(self):
""" Sorts the training data. """
self.Data = Data()
self.Data.process_data(
self.Data.paths_n_labels())
def do_train(self):
""" Creates & trains the model. """
self.Model.do_model(self.Data)
self.Model.do_train()
self.Model.do_evaluate()
def do_load_model(self):
""" Loads the model """
self.Model.load_model_and_weights()
def do_classify(self):
""" Loads model and classifies test data """
self.do_load_model()
self.Model.test_classifier()
def do_server(self):
""" Loads the API server """
self.do_load_model()
self.Server = Server(self.Model)
self.Server.start()
def do_http_classify(self):
""" Loads model and classifies test data """
self.Model.test_http_classifier()
class COVID19DN():
""" COVID19DN Class
Core COVID-19 Tensorflow DenseNet Classifier wrapper class using Tensroflow 2.
"""
def __init__(self):
""" Initializes the class. """
self.Helpers = Helpers("Core")
# Initiates the iotJumpWay connection class
self.iotJumpWayDevice = iotJumpWay({
"host": self.Helpers.confs["iotJumpWay"]["host"],
"port": self.Helpers.confs["iotJumpWay"]["port"],
"lid": self.Helpers.confs["iotJumpWay"]["loc"],
"zid": self.Helpers.confs["iotJumpWay"]["zne"],
"did": self.Helpers.confs["iotJumpWay"]["id"],
"dn": self.Helpers.confs["iotJumpWay"]["name"],
"un": self.Helpers.confs["iotJumpWay"]["mqtt"]["username"],
"pw": self.Helpers.confs["iotJumpWay"]["mqtt"]["password"]
})
self.iotJumpWayDevice.connect()
self.Model = Model(self.iotJumpWayDevice)
self.Helpers.logger.info(
"COVID19DN Tensorflow initialization complete.")
def life(self):
""" Sends vital statistics to HIAS """
cpu = psutil.cpu_percent()
mem = psutil.virtual_memory()[2]
hdd = psutil.disk_usage('/').percent
tmp = psutil.sensors_temperatures()['coretemp'][0].current
r = requests.get('http://ipinfo.io/json?token=' + self.Helpers.confs["iotJumpWay"]["key"])
data = r.json()
location = data["loc"].split(',')
self.Helpers.logger.info("COVID19DN Life (TEMPERATURE): " + str(tmp) + "\u00b0")
self.Helpers.logger.info("COVID19DN Life (CPU): " + str(cpu) + "%")
self.Helpers.logger.info("COVID19DN Life (Memory): " + str(mem) + "%")
self.Helpers.logger.info("COVID19DN Life (HDD): " + str(hdd) + "%")
self.Helpers.logger.info("COVID19DN Life (LAT): " + str(location[0]))
self.Helpers.logger.info("COVID19DN Life (LNG): " + str(location[1]))
# Send iotJumpWay notification
self.iotJumpWayDevice.devicePub("Life", {
"CPU": cpu,
"Memory": mem,
"Diskspace": hdd,
"Temperature": tmp,
"Latitude": location[0],
"Longitude": location[1]
})
threading.Timer(60.0, self.life).start()
def threading(self):
""" Creates required module threads. """
# Life thread
Thread(target = self.life, args = ()).start()
threading.Timer(60.0, self.life).start()
def do_data(self):
""" Sorts the training data. """
self.Data = Data()
self.Data.process_data(
self.Data.paths_n_labels())
def do_train(self):
""" Creates & trains the model. """
self.Model.do_model(self.Data)
self.Model.do_train()
self.Model.do_evaluate()
def do_load_model(self):
""" Loads the model """
self.Model.load_model_and_weights()
def do_classify(self):
""" Loads model and classifies test data """
self.do_load_model()
self.Model.test_classifier()
def do_server(self):
""" Loads the API server """
self.do_load_model()
self.Server = Server(self.Model)
self.Server.start()
def do_http_classify(self):
""" Loads model and classifies test data """
self.Model.test_http_classifier()
