Commit a7e2575d authored by dmMaze's avatar dmMaze
Browse files

fix auto downloading, close #490

parent e79705a6

launch.py

+2 −3
Original line number Diff line number Diff line
@@ -168,9 +168,6 @@ def main(): 
        config.module.load_model_on_demand = True

        config.module.empty_runcache = False



    from modules.prepare_local_files import prepare_local_files_forall

    prepare_local_files_forall()



    if sys.platform == 'win32':

        import ctypes

        myappid = u'BalloonsTranslator' # arbitrary string
@@ -198,6 +195,8 @@ def main(): 
    setup_logging(shared.LOGGING_PATH)



    load_modules()

    from modules.prepare_local_files import prepare_local_files_forall

    prepare_local_files_forall()



    app_args = sys.argv

    if args.headless:

modules/ocr/ocr_mit.py

+1 −1
Original line number Diff line number Diff line
@@ -81,7 +81,7 @@ class OCRMIT48px(MITModels): 


    params = deepcopy(mit_params)

    download_file_list = [{

        'url': 'https://huggingface.co/zyddnys/manga-image-translator/resolve/main/',

        'url': 'https://github.com/zyddnys/manga-image-translator/releases/download/beta-0.3/',

        'files': [OCR48PXMODEL_PATH, 'data/alphabet-all-v7.txt'],

        'sha256_pre_calculated': ['29daa46d080818bb4ab239a518a88338cbccff8f901bef8c9db191a7cb97671d', None],

        'concatenate_url_filename': 2,

modules/textdetector/base.py

+0 −7
Original line number Diff line number Diff line
@@ -18,13 +18,6 @@ class TextDetectorBase(BaseModule): 
    _postprocess_hooks = OrderedDict()

    _preprocess_hooks = OrderedDict()



    download_file_list = [{

        'url': 'https://github.com/zyddnys/manga-image-translator/releases/download/beta-0.3/',

        'files': ['data/models/comictextdetector.pt', 'data/models/comictextdetector.pt.onnx'],

        'sha256_pre_calculated': ['1f90fa60aeeb1eb82e2ac1167a66bf139a8a61b8780acd351ead55268540cccb', '1a86ace74961413cbd650002e7bb4dcec4980ffa21b2f19b86933372071d718f'],

        'concatenate_url_filename': 2,

    }]



    def __init__(self, **params) -> None:

        super().__init__(**params)

        self.name = ''

modules/textdetector/detector_ctd.py

+6 −0
Original line number Diff line number Diff line
@@ -30,6 +30,12 @@ class ComicTextDetector(TextDetectorBase): 
        'description': 'ComicTextDetector'

    }

    _load_model_keys = {'model'}

    download_file_list = [{

        'url': 'https://github.com/zyddnys/manga-image-translator/releases/download/beta-0.3/',

        'files': ['data/models/comictextdetector.pt', 'data/models/comictextdetector.pt.onnx'],

        'sha256_pre_calculated': ['1f90fa60aeeb1eb82e2ac1167a66bf139a8a61b8780acd351ead55268540cccb', '1a86ace74961413cbd650002e7bb4dcec4980ffa21b2f19b86933372071d718f'],

        'concatenate_url_filename': 2,

    }]



    device = DEFAULT_DEVICE

    detect_size = 1024