Spaces:
Sleeping
Sleeping
| """Web demo (Gradio): upload a photo, press Rectify, get the flat sheet and the status. | |
| pip install gradio && python3 app.py | |
| """ | |
| import tempfile | |
| import uuid | |
| from pathlib import Path | |
| import cv2 | |
| import gradio as gr | |
| from ecglib import illum, nets | |
| from rectify import OUTPUT_PX_PER_MM, THREADS, rectify_photo | |
| MAX_SIDE = 2048 | |
| MODELS = nets.load_models(threads=THREADS) | |
| OUT = Path(tempfile.mkdtemp()) | |
| def status_text(report, shrunk): | |
| lines = ["**Status: {}** — {}".format(report.get("status"), report.get("reason"))] | |
| holdout = report.get("selfcheck_holdout_mm", {}).get("median_mm") | |
| if holdout is not None: | |
| lines.append("Self-check median: {} mm".format(holdout)) | |
| if report.get("photo_px_per_mm"): | |
| lines.append("Photo scale: {:.2f} px/mm".format(report["photo_px_per_mm"])) | |
| if report.get("sheet_size_mm"): | |
| lines.append("Sheet: {:.0f} x {:.0f} mm".format(*report["sheet_size_mm"])) | |
| if shrunk: | |
| lines.append("Photo resized to {} px on the longer side".format(MAX_SIDE)) | |
| lines.append("Time: {} s".format(report.get("times", {}).get("total"))) | |
| return "\n\n".join(lines) | |
| def run(path, fix_illumination): | |
| if path is None: | |
| return None, None, "Upload a photo first." | |
| photo = cv2.imread(path, cv2.IMREAD_COLOR) | |
| if photo is None: | |
| return None, None, "Cannot read the image." | |
| scale = MAX_SIDE / max(photo.shape[:2]) | |
| if scale < 1: | |
| photo = cv2.resize(photo, None, fx=scale, fy=scale, interpolation=cv2.INTER_AREA) | |
| try: | |
| flat, check, report = rectify_photo(MODELS, photo) | |
| except Exception as error: | |
| return None, None, "**Status: fail** — {!r}".format(error) | |
| stem = OUT / uuid.uuid4().hex | |
| flat_path = check_path = None | |
| if flat is not None: | |
| if fix_illumination: | |
| flat = illum.flatten(flat, OUTPUT_PX_PER_MM, paper="white") | |
| flat_path = str(stem) + "-flat.jpg" | |
| cv2.imwrite(flat_path, flat, [cv2.IMWRITE_JPEG_QUALITY, 92]) | |
| if check is not None: | |
| check_path = str(stem) + "-check.jpg" | |
| cv2.imwrite(check_path, check, [cv2.IMWRITE_JPEG_QUALITY, 85]) | |
| return flat_path, check_path, status_text(report, scale < 1) | |
| with gr.Blocks(title="ECG Rectifier") as demo: | |
| gr.Markdown("# ECG Rectifier\nA phone photo of a paper ECG (bent, crumpled, folded) -> a flat sheet " | |
| "with a square grid at 10 px/mm. CPU only; one photo takes about a minute.") | |
| with gr.Row(): | |
| with gr.Column(): | |
| photo = gr.Image(type="filepath", label="Photo") | |
| fix = gr.Checkbox(value=True, label="Fix illumination") | |
| button = gr.Button("Rectify", variant="primary") | |
| gr.Examples([["examples/folded.jpg"], ["examples/crumpled.jpg"]], inputs=[photo]) | |
| with gr.Column(): | |
| flat = gr.Image(type="filepath", label="Flat sheet") | |
| status = gr.Markdown() | |
| check = gr.Image(type="filepath", label="Mask (blue), working area (yellow), crop (green), grid nodes used (red)") | |
| button.click(run, inputs=[photo, fix], outputs=[flat, check, status]) | |
| if __name__ == "__main__": | |
| demo.launch() | |