Hintergrundverarbeitung, Bildaufbereitung, Sammelexport entfernt
All checks were successful
Build & Deploy / deploy (push) Successful in 1m24s
All checks were successful
Build & Deploy / deploy (push) Successful in 1m24s
Auslesen im Hintergrund: /api/scan antwortet, sobald die Karten freigestellt und gespeichert sind, und stößt die Extraktion als Task an. Ein Zwanzigerstapel blockierte vorher den Upload für die ganze Dauer aller Modellaufrufe. Die Oberfläche zeigt "wird gelesen" und lädt nach, solange etwas offen ist. Zwei Folgen davon sind mitbehandelt: Beim Schreiben der Ergebnisse steht COALESCE, damit ein Handeintrag während des Lesens nicht überschrieben wird, und offene Karten werden beim Start nachgeholt, statt dauerhaft in der Warteschlange zu hängen. Bildaufbereitung des Zuschnitts: - Der Einzug zieht die erkannten Ecken um 1,5 % zur Mitte, damit kein Untergrund im Zuschnitt bleibt. - Die Beleuchtung wird ausgeglichen (Division durch eine weichgezeichnete Fassung), damit das Papier weiß wird statt grau. Dunkle Karten bleiben unangetastet - bei ihnen ist das Dunkle das Papier, kein Schatten. - Das Modell meldet die nötige Drehung im Schema; das gespeicherte Bild wird entsprechend gedreht. Geometrisch ist die Lage nicht bestimmbar. Sammelexport entfernt: Der Button "Alle als vCard" ist weg, mit ihm der Endpunkt /vcf sowie vcard.build_many und db.execute_many, die dadurch keinen Aufrufer mehr hatten. Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
This commit is contained in:
89
tests/test_bildaufbereitung.py
Normal file
89
tests/test_bildaufbereitung.py
Normal file
@@ -0,0 +1,89 @@
|
||||
"""Zuschnitt aufhellen, Untergrund abschneiden, richtig herum drehen."""
|
||||
import io
|
||||
|
||||
import numpy as np
|
||||
from PIL import Image
|
||||
|
||||
from app import imaging, segment
|
||||
from tests.helpers import photo_with_cards
|
||||
|
||||
|
||||
def _jpeg(width: int, height: int, color=(255, 255, 255)) -> bytes:
|
||||
buffer = io.BytesIO()
|
||||
Image.new("RGB", (width, height), color).save(buffer, format="JPEG")
|
||||
return buffer.getvalue()
|
||||
|
||||
|
||||
def _size(raw: bytes) -> tuple:
|
||||
return Image.open(io.BytesIO(raw)).size
|
||||
|
||||
|
||||
def test_drehung_um_null_grad_laesst_die_datei_unangetastet():
|
||||
raw = _jpeg(40, 20)
|
||||
|
||||
assert imaging.rotate_jpeg(raw, 0) is raw
|
||||
assert imaging.rotate_jpeg(raw, 360) is raw
|
||||
|
||||
|
||||
def test_drehung_um_180_grad_behaelt_das_format():
|
||||
assert _size(imaging.rotate_jpeg(_jpeg(40, 20), 180)) == (40, 20)
|
||||
|
||||
|
||||
def test_drehung_um_90_grad_tauscht_die_seiten():
|
||||
assert _size(imaging.rotate_jpeg(_jpeg(40, 20), 90)) == (20, 40)
|
||||
assert _size(imaging.rotate_jpeg(_jpeg(40, 20), 270)) == (20, 40)
|
||||
|
||||
|
||||
def test_drehrichtung_ist_im_uhrzeigersinn():
|
||||
# Oben links schwarz: nach 90 Grad im Uhrzeigersinn liegt es oben rechts.
|
||||
image = np.full((20, 40, 3), 255, dtype=np.uint8)
|
||||
image[:10, :20] = 0
|
||||
buffer = io.BytesIO()
|
||||
Image.fromarray(image).save(buffer, format="PNG")
|
||||
gedreht = np.asarray(
|
||||
Image.open(io.BytesIO(imaging.rotate_jpeg(buffer.getvalue(), 90)))
|
||||
)
|
||||
|
||||
assert gedreht[:10, -10:].mean() < 60 # oben rechts jetzt dunkel
|
||||
assert gedreht[:10, :10].mean() > 195 # oben links jetzt hell
|
||||
|
||||
|
||||
def test_helle_karte_wird_aufgehellt():
|
||||
# Karte mit Helligkeitsverlauf, wie ihn schraeges Licht erzeugt.
|
||||
verlauf = np.linspace(150, 225, 400, dtype=np.float32)
|
||||
karte = np.repeat(verlauf[None, :, None], 250, axis=0).repeat(3, axis=2)
|
||||
karte = karte.astype(np.uint8)
|
||||
|
||||
aufgehellt = segment.flatten(karte)
|
||||
|
||||
assert aufgehellt.mean() > karte.mean() + 20
|
||||
# Der Verlauf ist danach deutlich flacher als vorher.
|
||||
vorher = float(karte[:, -1].mean() - karte[:, 0].mean())
|
||||
nachher = float(aufgehellt[:, -1].mean() - aufgehellt[:, 0].mean())
|
||||
assert nachher < vorher / 2
|
||||
|
||||
|
||||
def test_dunkle_karte_bleibt_dunkel():
|
||||
dunkel = np.full((250, 400, 3), 40, dtype=np.uint8)
|
||||
|
||||
assert np.array_equal(segment.flatten(dunkel), dunkel)
|
||||
|
||||
|
||||
def test_zuschnitt_laesst_den_untergrund_draussen():
|
||||
# Der Untergrund im Testfoto ist dunkel. Bleibt ein Streifen davon im
|
||||
# Zuschnitt, waeren die Randspalten deutlich dunkler als die Mitte.
|
||||
crops, _ = segment.segment(photo_with_cards(count=4))
|
||||
|
||||
for crop in crops:
|
||||
rand = np.concatenate([crop[:, :3], crop[:, -3:]], axis=1)
|
||||
assert rand.mean() > 170
|
||||
|
||||
|
||||
def test_einzug_zieht_die_ecken_zur_mitte():
|
||||
quad = np.array([[0, 0], [100, 0], [100, 100], [0, 100]], dtype="float32")
|
||||
|
||||
kleiner = segment._shrink(quad, factor=0.1)
|
||||
|
||||
assert kleiner[0][0] > quad[0][0]
|
||||
assert kleiner[2][0] < quad[2][0]
|
||||
assert np.allclose(kleiner.mean(axis=0), quad.mean(axis=0))
|
||||
Reference in New Issue
Block a user