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