File size: 1,253 Bytes
87ec96b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
from PIL import Image, ImageChops
import requests
import os
import torch
import functools
import numpy as np
from modules.utils.paths import *
TEST_IMAGE_URL = "https://github.com/microsoft/onnxjs-demo/raw/master/src/assets/EmotionSampleImages/sad_baby.jpg"
TEST_IMAGE_PATH = os.path.join(PROJECT_ROOT_DIR, "tests", "test.png")
TEST_EXPRESSION_OUTPUT_PATH = os.path.join(PROJECT_ROOT_DIR, "tests", "edited_expression.png")
TEST_EXPRESSION_AAA = 100
def download_image(url, path):
if os.path.exists(path):
return
response = requests.get(url, stream=True)
if response.status_code == 200:
with open(path, 'wb') as file:
for chunk in response.iter_content(1024):
file.write(chunk)
print(f"Image successfully downloaded to {path}")
else:
raise Exception(f"Failed to download image. Status code: {response.status_code}")
def are_images_different(image1_path: str, image2_path: str):
image1 = Image.open(image1_path)
image2 = Image.open(image2_path)
diff = ImageChops.difference(image1, image2)
if diff.getbbox() is None:
return False
else:
return True
@functools.lru_cache
def is_cuda_available():
return torch.cuda.is_available()
|