config.py

import argparse
from multiprocessing import cpu_count

class Config:
    def __init__(self):
        self.device = "cpu"
        self.is_half = False
        self.n_cpu = cpu_count()
        (self.python_cmd, self.colab, self.noparallel, self.noautoopen, self.api) = self.arg_parse()
        self.listen_port = 7860
        self.x_pad, self.x_query, self.x_center, self.x_max = self.device_config()

    @staticmethod
    def arg_parse() -> tuple:
        parser = argparse.ArgumentParser()
        parser.add_argument("--pycmd", type=str, default="python")
        parser.add_argument("--colab", action="store_true")
        parser.add_argument("--noparallel", action="store_true")
        parser.add_argument("--noautoopen", action="store_true")
        parser.add_argument("--api", action="store_true")
        cmd_opts = parser.parse_args()

        return (cmd_opts.pycmd, cmd_opts.colab, cmd_opts.noparallel, cmd_opts.noautoopen, cmd_opts.api)

    def device_config(self) -> tuple:
        x_pad = 1
        x_query = 6
        x_center = 38
        x_max = 41

        return x_pad, x_query, x_center, x_max