Comment out model channel testing.

4ee687b8 · Robin Tissot · 76c8fb17 · 4ee687b8
Commit 4ee687b8 authored Nov 18, 2020 by Robin Tissot
Hide whitespace changes
Inline Side-by-side

Showing with 10 additions and 9 deletions

app/apps/core/models.py app/apps/core/models.py +10 -9

No files found.
--- a/app/apps/core/models.py
+++ b/app/apps/core/models.py
@@ -614,15 +614,16 @@ class DocumentPart(OrderedModel):
        # TODO: check model_type [None, 'recognition', 'segmentation']
        #    &  seg_type [None, 'bbox', 'baselines']
-        if model_.one_channel_mode == '1':
+        # will be fixed sometime in the future
-            # TODO: need to binarize, probably not live...
+        # if model_.one_channel_mode == '1':
-            if not self.bw_image:
+        #     # TODO: need to binarize, probably not live...
-                self.binarize()
+        #     if not self.bw_image:
-            im = Image.open(self.bw_image.file.name)
+        #         self.binarize()
-        elif model_.one_channel_mode == 'L':
+        #     im = Image.open(self.bw_image.file.name)
-            im = Image.open(self.image.file.name).convert('L')
+        # elif model_.one_channel_mode == 'L':
-        else:
+        #     im = Image.open(self.image.file.name).convert('L')
-            im = Image.open(self.image.file.name)
+        # else:
+        #     im = Image.open(self.image.file.name)
        options = {
            'device': getattr(settings, 'KRAKEN_TRAINING_DEVICE', 'cpu'),