mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2026-03-23 23:00:26 -07:00
975dda4b56
* docs(M03): seed M03 plan and toolcalls Made-with: Cursor * ci: add repository guardrail, update CONTRIBUTING workflow Made-with: Cursor * M03: Test architecture (smoke/quality/nightly) - Move tests to test/smoke/, scaffold test/quality and test/nightly - Add pytest.ini with smoke/quality/nightly markers - Split CI: run_smoke_tests (PR), run_quality_tests (push main), run_nightly_tests (schedule) - Remove run_tests.yaml - Add prevent_upstream_push.sh pre-push hook template - Update CONTRIBUTING.md with hook install and test tier docs - Add repo and base-branch guardrails to workflows Made-with: Cursor * ci: remove obsolete warns_merge_master workflow (Serena uses main) Made-with: Cursor * fix: add base_url to pytest.ini for pytest-base-url plugin Made-with: Cursor * docs(M03): ledger, run1, audit, summary Made-with: Cursor * docs(M03): ledger commit 4ce5cde9 Made-with: Cursor
29 lines
1.1 KiB
Python
29 lines
1.1 KiB
Python
import os
|
|
from test.conftest import test_files_path, test_outputs_path
|
|
|
|
import numpy as np
|
|
import pytest
|
|
from PIL import Image
|
|
|
|
|
|
@pytest.mark.usefixtures("initialize")
|
|
@pytest.mark.parametrize("restorer_name", ["gfpgan", "codeformer"])
|
|
def test_face_restorers(restorer_name):
|
|
from modules import shared
|
|
|
|
if restorer_name == "gfpgan":
|
|
from modules import gfpgan_model
|
|
gfpgan_model.setup_model(shared.cmd_opts.gfpgan_models_path)
|
|
restorer = gfpgan_model.gfpgan_fix_faces
|
|
elif restorer_name == "codeformer":
|
|
from modules import codeformer_model
|
|
codeformer_model.setup_model(shared.cmd_opts.codeformer_models_path)
|
|
restorer = codeformer_model.codeformer.restore
|
|
else:
|
|
raise NotImplementedError("...")
|
|
img = Image.open(os.path.join(test_files_path, "two-faces.jpg"))
|
|
np_img = np.array(img, dtype=np.uint8)
|
|
fixed_image = restorer(np_img)
|
|
assert fixed_image.shape == np_img.shape
|
|
assert not np.allclose(fixed_image, np_img) # should have visibly changed
|
|
Image.fromarray(fixed_image).save(os.path.join(test_outputs_path, f"{restorer_name}.png"))
|