admin/LC_NeoRefacer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use local temp dir during face extraction

Browse Source
This commit is contained in:
Felipe Daragon
2025-04-26 13:28:11 +01:00
parent 99c91d20b2
commit ef87bffe7d
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app.py
View File

@@ -48,7 +48,7 @@ num_faces = args.max_num_faces
def create_dummy_image(): def create_dummy_image():
dummy = Image.new('RGB', (1, 1), color=(255, 255, 255)) dummy = Image.new('RGB', (1, 1), color=(255, 255, 255))
temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png") temp_file = tempfile.NamedTemporaryFile(delete=False, dir="./tmp", suffix=".png")
dummy.save(temp_file.name) dummy.save(temp_file.name)
return temp_file.name return temp_file.name

2
refacer.py
View File

@@ -342,7 +342,7 @@ class Refacer:
cropped = frame[y1:y2, x1:x2] cropped = frame[y1:y2, x1:x2]
pil_img = Image.fromarray(cv2.cvtColor(cropped, cv2.COLOR_BGR2RGB)) pil_img = Image.fromarray(cv2.cvtColor(cropped, cv2.COLOR_BGR2RGB))
temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png") temp_file = tempfile.NamedTemporaryFile(delete=False, dir="./tmp", suffix=".png")
pil_img.save(temp_file.name) pil_img.save(temp_file.name)
cropped_faces.append(temp_file.name) cropped_faces.append(temp_file.name)