notebook.txt

!pip install insightface
-------------------------------


from huggingface_hub import loggin

loggin(token="")
-----------------------------------------------


from huggingface_hub import hf_hub_download

repo_id = "AshanGimhana/NewFWModelV1"  
model_filename = "inswapperFTV1Model.onnx"

--------------------------------------------------


model_path = hf_hub_download(repo_id=repo_id, filename=model_filename)

----------------------------------------------------------------------------


model = "copy path and paste"

----------------------------------------------------------------------------

from matplotlib import pyplot as plt
import cv2
import insightface
import numpy as np
-----------------------------------------------------------------------------

providers = ["CPUExecutionProvider"]
-----------------------------------------------------------------------------

target_frame = cv2.imread('Enter reference image path')
src_frame = cv2.imread("Enter user image path")
fig = plt.figure()
ax1 = fig.add_subplot(1,2,1)
ax1.imshow(cv2.cvtColor(src_frame, cv2.COLOR_BGR2RGB))
ax2 = fig.add_subplot(1,2,2)
ax2.imshow(cv2.cvtColor(target_frame, cv2.COLOR_BGR2RGB))
plt.show()
------------------------------------------------------------------------------

FACE_ANALYSER = insightface.app.FaceAnalysis(
                    name="buffalo_l",
                    root=".", providers=providers,allowed_modules=["landmark_3d_68", "landmark_2d_106","detection","recognition"]
                )
FACE_ANALYSER.prepare(
                ctx_id=0,
                det_size=(640, 640),
            )
----------------------------------------------------------------------------------------------------------------------------------------

src_faces = FACE_ANALYSER.get(src_frame)
target_faces = FACE_ANALYSER.get(target_frame)
fig = plt.figure()
ax1 = fig.add_subplot(1,2,1)
rimg = FACE_ANALYSER.draw_on(src_frame,src_faces)
ax1.imshow(cv2.cvtColor(rimg, cv2.COLOR_BGR2RGB))
ax1 = fig.add_subplot(1,2,2)
rimg = FACE_ANALYSER.draw_on(target_frame,target_faces)
ax1.imshow(cv2.cvtColor(rimg, cv2.COLOR_BGR2RGB))
plt.show()

----------------------------------------------------------------------------------------------------------------------------------------

model_swap_insightface = insightface.model_zoo.get_model(model, providers=providers)

----------------------------------------------------------------------------------------------------

img_fake = model_swap_insightface.get(img = target_frame, target_face=FACE_ANALYSER.get(target_frame)[0], source_face=src_faces[0], paste_back=True)

-------------------------------------------------------------------------------------------------------------------------------------------------------

plt.imshow(cv2.cvtColor(img_fake, cv2.COLOR_BGR2RGB))

------------------------------------------------------------------------