Use local temp dir during face extraction
This commit is contained in:
2
app.py
2
app.py
@@ -48,7 +48,7 @@ num_faces = args.max_num_faces
|
||||
|
||||
def create_dummy_image():
|
||||
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)
|
||||
return temp_file.name
|
||||
|
||||
|
||||
@@ -342,7 +342,7 @@ class Refacer:
|
||||
cropped = frame[y1:y2, x1:x2]
|
||||
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)
|
||||
cropped_faces.append(temp_file.name)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user