paths.py

"""
This file contains the Path class that creates, deletes, or
modifies paths. Paths augmented may be related to saving
trained models or saving images.

This file is Copyright (c) 2022 Steven Tin Sui Luo.
"""

import os

from parameters import Params

params = Params()

class Path:
    """This class prepares the directories for saving models
    and training logs depending on the path names specified in
    the Params() class.
    """
    def __init__(self):
        self.model_path = params.MODEL_PATH
        self.log_path = params.LOG_PATH
        self.model_log_path = params.MODEL_NAME

    def create_model_path(self):
        """This method creates a subdirectory for trained models."""
        if self.model_path not in os.listdir('.'):
            os.makedirs(self.model_path)

    def create_log_path(self):
        """This method creates a subdirectory for saving training logs."""
        if self.log_path not in os.listdir('.'):
            os.makedirs(self.log_path)

    def create_model_log_path(self):
        """This method creates a subdirectory in <logs> to save training logs
        of a specific model.
        """
        if self.model_log_path not in os.listdir(self.model_path):
            os.makedirs(os.path.join(self.model_path, self.model_log_path))