Skip to content
Snippets Groups Projects

Compare revisions

Changes are shown as if the source revision was being merged into the target revision. Learn more about comparing revisions.

Source

Select target project
No results found

Target

Select target project
  • atr/dan
1 result
Show changes
Hide whitespace changes
Inline Side-by-side
tests/test_prediction.py
View file @ bb413c0f
......@@ -5,6 +5,7 @@ import shutil
import pytest
from dan.ocr.predict.attention import Level
from dan.ocr.predict.prediction import DAN
from dan.ocr.predict.prediction import run as run_prediction
from dan.utils import parse_tokens
......@@ -69,7 +70,7 @@ def test_predict(
),
(
"0a56e8b3-95cd-4fa5-a17b-5b0ff9e6ea84",
["word"],
[Level.Word],
1.0,
{
"text": "ⓈBellisson ⒻGeorges Ⓑ91 ⓁP ⒸM ⓀCh ⓄPlombier ⓅPatron?12241",
......@@ -100,7 +101,7 @@ def test_predict(
),
(
"0a56e8b3-95cd-4fa5-a17b-5b0ff9e6ea84",
["word"],
[Level.Word],
3.5,
{
"text": "ⓈBellisson ⒻGeorges Ⓑ91 ⓁP ⒸM ⓀCh ⓄPlombier ⓅPatron?12241",
......@@ -131,7 +132,7 @@ def test_predict(
),
(
"0a56e8b3-95cd-4fa5-a17b-5b0ff9e6ea84",
["line"],
[Level.Line],
1.0,
{
"text": "ⓈBellisson ⒻGeorges Ⓑ91 ⓁP ⒸM ⓀCh ⓄPlombier ⓅPatron?12241",
......@@ -158,7 +159,7 @@ def test_predict(
),
(
"0a56e8b3-95cd-4fa5-a17b-5b0ff9e6ea84",
["line"],
[Level.Line],
3.5,
{
"text": "ⓈBellisson ⒻGeorges Ⓑ91 ⓁP ⒸM ⓀCh ⓄPlombier ⓅPatron?12241",
......@@ -191,7 +192,7 @@ def test_predict(
),
(
"0dfe8bcd-ed0b-453e-bf19-cc697012296e",
["char", "word", "line"],
[Level.Char, Level.Word, Level.Line],
1.0,
{
"text": "ⓈTemplié ⒻMarcelle Ⓑ93 ⓁS Ⓚch ⓄE dactylo Ⓟ18376",
......@@ -336,7 +337,7 @@ def test_run_prediction(
),
(
["0a56e8b3-95cd-4fa5-a17b-5b0ff9e6ea84"],
["word"],
[Level.Word],
1.0,
[
{
......@@ -372,7 +373,7 @@ def test_run_prediction(
"0a56e8b3-95cd-4fa5-a17b-5b0ff9e6ea84",
"0a56e8b3-95cd-4fa5-a17b-5b0ff9e6ea84",
],
["word"],
[Level.Word],
1.0,
[
{
......@@ -431,7 +432,7 @@ def test_run_prediction(
),
(
["0a56e8b3-95cd-4fa5-a17b-5b0ff9e6ea84"],
["word"],
[Level.Word],
1.0,
[
{
......