| |
| import io |
| import json |
| import uuid |
| import requests |
| import base64 |
| from PIL import Image |
|
|
| URL = 'http://127.0.0.1:5000/faceswap' |
|
|
| SOURCE_IMAGE = '../data/src.jpg' |
| TARGET_IMAGE = '../data/target.jpg' |
| SOURCE_INDEXES = '-1' |
| TARGET_INDEXES = '-1' |
| BACKGROUND_ENHANCE = True |
| FACE_RESTORE = True |
| FACE_UPSAMPLE = True |
| UPSCALE = 1 |
| CODEFORMER_FIDELITY = 0.5 |
| OUTPUT_FORMAT = 'JPEG' |
|
|
|
|
| def encode_image_to_base64(image_path): |
| with open(image_path, 'rb') as image_file: |
| return base64.b64encode(image_file.read()).decode('utf-8') |
|
|
|
|
| def save_result_image(resp_json): |
| img = Image.open(io.BytesIO(base64.b64decode(resp_json['output']['image']))) |
| output_file = f'{uuid.uuid4()}.jpg' |
|
|
| with open(output_file, 'wb') as f: |
| print(f'Saving image: {output_file}') |
| img.save(f, format='JPEG') |
|
|
|
|
| if __name__ == '__main__': |
| payload = { |
| "source_image": encode_image_to_base64(SOURCE_IMAGE), |
| "target_image": encode_image_to_base64(TARGET_IMAGE), |
| "source_indexes": SOURCE_INDEXES, |
| "target_indexes": TARGET_INDEXES, |
| "background_enhance": BACKGROUND_ENHANCE, |
| "face_restore": FACE_RESTORE, |
| "face_upsample": FACE_UPSAMPLE, |
| "upscale": UPSCALE, |
| "codeformer_fidelity": CODEFORMER_FIDELITY, |
| "output_format": OUTPUT_FORMAT |
| } |
|
|
| r = requests.post( |
| URL, |
| json=payload |
| ) |
|
|
| print(f'HTTP status code: {r.status_code}') |
|
|
| resp_json = r.json() |
|
|
| if r.status_code == 200: |
| img = Image.open(io.BytesIO(base64.b64decode(resp_json['image']))) |
| output_file = f'{uuid.uuid4()}.jpg' |
|
|
| with open(output_file, 'wb') as f: |
| print(f'Saving image: {output_file}') |
| img.save(f, format='JPEG') |
| else: |
| print(json.dumps(resp_json, indent=4, default=str)) |
|
|
