From 5b60deadfd06daa4f50c2c80decadcddf3fe1503 Mon Sep 17 00:00:00 2001 From: Robin Tissot <tissotrobin@gmail.com> Date: Thu, 13 Aug 2020 11:19:43 +0200 Subject: [PATCH] Segmentation training resize and train on top of default model. --- app/apps/core/tasks.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/app/apps/core/tasks.py b/app/apps/core/tasks.py index bc7a008a..0480f7de 100644 --- a/app/apps/core/tasks.py +++ b/app/apps/core/tasks.py @@ -146,8 +146,7 @@ def segtrain(task, model_pk, document_pk, part_pks, user_pk=None): load = model.file.path upload_to = model.file.field.upload_to(model, model.name + '.mlmodel') except ValueError: # model is empty - # load = settings.KRAKEN_DEFAULT_SEGMENTATION_MODEL - load = None + load = settings.KRAKEN_DEFAULT_SEGMENTATION_MODEL upload_to = model.file.field.upload_to(model, model.name + '.mlmodel') model.file = upload_to @@ -185,6 +184,7 @@ def segtrain(task, model_pk, document_pk, part_pks, user_pk=None): evaluation_data=evaluation_data, threads=LOAD_THREADS, augment=True, + resize='both', load_hyper_parameters=True) if not os.path.exists(os.path.split(modelpath)[0]): -- GitLab