initial commit

app.py

@@ -0,0 +1,71 @@
+import gradio as gr
+import cv2
+import numpy as np
+import insightface
+from insightface.app import FaceAnalysis
+import datetime
+import os
+from PIL import Image
+
+
+def faceswapper(user_image, result_image, username="test"):
+    output_folder = 'outputs'
+
+    # Convert PIL images to NumPy arrays for processing
+    guest_img = np.array(user_image)
+    result_img = np.array(result_image)
+
+    # Convert RGB (PIL) to BGR (OpenCV)
+    guest_img = guest_img[:, :, ::-1]
+    result_img = result_img[:, :, ::-1]
+
+
+    # Initialize the FaceAnalysis app
+    app = FaceAnalysis(name='buffalo_l')
+    app.prepare(ctx_id=0, det_size=(640, 640))
+
+    # Initialize the face swapper model
+    swapper = insightface.model_zoo.get_model('inswapper_128.onnx', download=False, download_zip=False)
+
+    # Detect face in the guest image
+    guest_faces = app.get(guest_img)
+    guest_face = guest_faces[0]
+
+    # Detect faces in the result image
+    faces = app.get(result_img)
+
+    # Perform face swapping
+    for face in faces:
+        result_img = swapper.get(result_img, face, guest_face, paste_back=True)
+
+    # Save the result in the specified output folder
+    if not os.path.exists(output_folder):
+        os.makedirs(output_folder)
+    current_time = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
+    output_path = os.path.join(output_folder, f'{username}_swapped_face_{current_time}.jpg')
+    cv2.imwrite(output_path, result_img)
+
+    # Convert the final image from BGR to RGB before returning
+    result_img = cv2.cvtColor(result_img, cv2.COLOR_BGR2RGB)
+
+
+    # Convert back to PIL image
+    result_img_pil = Image.fromarray(result_img)
+    original_size = result_image.size
+    result_img_pil = result_img_pil.resize(original_size, Image.Resampling.LANCZOS)
+
+    return result_img_pil
+
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column():
+            name = gr.Textbox(label="이름(파일저장용)")
+            with gr.Row():
+                user_image_input = gr.Image(type="pil", label="유저사진(얼굴추출)", width=300, height=300)
+                result_image_input = gr.Image(type="pil", label="결과물 사진", width=300, height=300)
+    swap_btn = gr.Button("Swap Faces")
+    with gr.Column():
+            output_image = gr.Image(label="합성 후 사진")
+    swap_btn.click(fn=faceswapper, inputs=[user_image_input, result_image_input, name], outputs=output_image)
+
+demo.launch(debug=True)

requirements.txt

@@ -0,0 +1,117 @@
+aiofiles==23.2.1
+albumentations==1.3.1
+altair==5.2.0
+annotated-types==0.6.0
+anyio==4.2.0
+appnope @ file:///home/conda/feedstock_root/build_artifacts/appnope_1649077682618/work
+asttokens @ file:///home/conda/feedstock_root/build_artifacts/asttokens_1698341106958/work
+attrs==23.2.0
+certifi==2023.11.17
+charset-normalizer==3.3.2
+click==8.1.7
+colorama==0.4.6
+coloredlogs==15.0.1
+comm @ file:///home/conda/feedstock_root/build_artifacts/comm_1704278392174/work
+contourpy==1.2.0
+cycler==0.12.1
+Cython==3.0.7
+debugpy @ file:///Users/runner/miniforge3/conda-bld/debugpy_1695534448458/work
+decorator @ file:///home/conda/feedstock_root/build_artifacts/decorator_1641555617451/work
+easydict==1.11
+exceptiongroup @ file:///home/conda/feedstock_root/build_artifacts/exceptiongroup_1700579780973/work
+executing @ file:///home/conda/feedstock_root/build_artifacts/executing_1698579936712/work
+fastapi==0.108.0
+ffmpy==0.3.1
+filelock==3.13.1
+flatbuffers==23.5.26
+fonttools==4.47.0
+fsspec==2023.12.2
+gradio==4.13.0
+gradio_client==0.8.0
+h11==0.14.0
+httpcore==1.0.2
+httpx==0.26.0
+huggingface-hub==0.20.2
+humanfriendly==10.0
+idna==3.6
+imageio==2.33.1
+importlib-metadata @ file:///home/conda/feedstock_root/build_artifacts/importlib-metadata_1703269254275/work
+importlib-resources==6.1.1
+insightface==0.7.3
+ipykernel @ file:///Users/runner/miniforge3/conda-bld/ipykernel_1703631823913/work
+ipython @ file:///home/conda/feedstock_root/build_artifacts/ipython_1703263236586/work
+jedi @ file:///home/conda/feedstock_root/build_artifacts/jedi_1696326070614/work
+Jinja2==3.1.2
+joblib==1.3.2
+jsonschema==4.20.0
+jsonschema-specifications==2023.12.1
+jupyter_client @ file:///home/conda/feedstock_root/build_artifacts/jupyter_client_1699283905679/work
+jupyter_core @ file:///Users/runner/miniforge3/conda-bld/jupyter_core_1704321426011/work
+kiwisolver==1.4.5
+lazy_loader==0.3
+markdown-it-py==3.0.0
+MarkupSafe==2.1.3
+matplotlib==3.8.2
+matplotlib-inline @ file:///home/conda/feedstock_root/build_artifacts/matplotlib-inline_1660814786464/work
+mdurl==0.1.2
+mpmath==1.3.0
+nest-asyncio @ file:///home/conda/feedstock_root/build_artifacts/nest-asyncio_1697083700168/work
+networkx==3.2.1
+numpy==1.26.3
+onnx==1.15.0
+onnxruntime==1.16.3
+opencv-python-headless==4.9.0.80
+orjson==3.9.10
+packaging @ file:///home/conda/feedstock_root/build_artifacts/packaging_1696202382185/work
+pandas==2.1.4
+parso @ file:///home/conda/feedstock_root/build_artifacts/parso_1638334955874/work
+pexpect @ file:///home/conda/feedstock_root/build_artifacts/pexpect_1667297516076/work
+pickleshare @ file:///home/conda/feedstock_root/build_artifacts/pickleshare_1602536217715/work
+pillow==10.2.0
+platformdirs @ file:///home/conda/feedstock_root/build_artifacts/platformdirs_1701708255999/work
+prettytable==3.9.0
+prompt-toolkit @ file:///home/conda/feedstock_root/build_artifacts/prompt-toolkit_1702399386289/work
+protobuf==4.25.1
+psutil @ file:///Users/runner/miniforge3/conda-bld/psutil_1702833169958/work
+ptyprocess @ file:///home/conda/feedstock_root/build_artifacts/ptyprocess_1609419310487/work/dist/ptyprocess-0.7.0-py2.py3-none-any.whl
+pure-eval @ file:///home/conda/feedstock_root/build_artifacts/pure_eval_1642875951954/work
+pydantic==2.5.3
+pydantic_core==2.14.6
+pydub==0.25.1
+Pygments @ file:///home/conda/feedstock_root/build_artifacts/pygments_1700607939962/work
+pyparsing==3.1.1
+python-dateutil @ file:///home/conda/feedstock_root/build_artifacts/python-dateutil_1626286286081/work
+python-multipart==0.0.6
+pytz==2023.3.post1
+PyYAML==6.0.1
+pyzmq @ file:///Users/runner/miniforge3/conda-bld/pyzmq_1701783254483/work
+qudida==0.0.4
+referencing==0.32.1
+requests==2.31.0
+rich==13.7.0
+rpds-py==0.16.2
+scikit-image==0.22.0
+scikit-learn==1.3.2
+scipy==1.11.4
+semantic-version==2.10.0
+shellingham==1.5.4
+six @ file:///home/conda/feedstock_root/build_artifacts/six_1620240208055/work
+sniffio==1.3.0
+stack-data @ file:///home/conda/feedstock_root/build_artifacts/stack_data_1669632077133/work
+starlette==0.32.0.post1
+sympy==1.12
+threadpoolctl==3.2.0
+tifffile==2023.12.9
+tomlkit==0.12.0
+toolz==0.12.0
+tornado @ file:///Users/runner/miniforge3/conda-bld/tornado_1695373570914/work
+tqdm==4.66.1
+traitlets @ file:///home/conda/feedstock_root/build_artifacts/traitlets_1704212992681/work
+typer==0.9.0
+typing_extensions @ file:///home/conda/feedstock_root/build_artifacts/typing_extensions_1702176139754/work
+tzdata==2023.4
+urllib3==2.1.0
+uvicorn==0.25.0
+wcwidth @ file:///home/conda/feedstock_root/build_artifacts/wcwidth_1700607916581/work
+websockets==11.0.3
+zipp @ file:///home/conda/feedstock_root/build_artifacts/zipp_1695255097490/work