mirror of
https://github.com/netfun2000/dddd_trainer.git
synced 2026-08-12 22:11:36 +08:00
sml2h3
40 lines
1.7 KiB
Python
40 lines
1.7 KiB
Python
import os
|
|
from configs import Config
|
|
from loguru import logger
|
|
|
|
|
|
class ProjectManager:
|
|
|
|
def __init__(self):
|
|
self.base_path = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "projects")
|
|
|
|
def create_project(self, project_name: str, single: bool = False):
|
|
project_base_path = os.path.join(self.base_path, project_name)
|
|
logger.info("Creating Directory... ----> {}".format(project_base_path))
|
|
if not os.path.exists(project_base_path):
|
|
os.mkdir(project_base_path)
|
|
if not os.path.exists(project_base_path):
|
|
logger.error("Directory create failed! ----> {}".format(project_base_path))
|
|
return False
|
|
models_path = os.path.join(project_base_path, "models")
|
|
logger.info("Creating Directory... ----> {}".format(models_path))
|
|
os.mkdir(models_path)
|
|
|
|
cache_path = os.path.join(project_base_path, "cache")
|
|
logger.info("Creating Directory... ----> {}".format(cache_path))
|
|
os.mkdir(cache_path)
|
|
|
|
checkpoints_path = os.path.join(project_base_path, "checkpoints")
|
|
logger.info("Creating Directory... ----> {}".format(checkpoints_path))
|
|
os.mkdir(checkpoints_path)
|
|
|
|
config_path = os.path.join(os.path.join(project_base_path, "config.yaml"))
|
|
logger.info("Creating {} Config File... ----> {}".format("CNN" if single else "CRNN", config_path))
|
|
conf = Config(project_name)
|
|
conf.make_config(single=single)
|
|
|
|
logger.info("Create Project Success! ----> {}".format(project_name))
|
|
else:
|
|
logger.error("Directory already exists! ----> {}".format(project_base_path))
|
|
return False
|