You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
14 lines
494 B
14 lines
494 B
from apps.service import style
|
|
|
|
def style_transfer(img_style_name, img_content_name, result_name, epoch, per_epoch, learn_rate):
|
|
img_style_path = 'apps/assets/' + img_style_name
|
|
img_content_path = 'apps/assets/' + img_content_name
|
|
result_path = 'apps/results/' + result_name
|
|
|
|
try:
|
|
style.style_transfer(img_style_path, img_content_path, result_path, int(epoch), int(per_epoch), float(learn_rate))
|
|
except Exception as e:
|
|
print(e)
|
|
return 2
|
|
|
|
return 1 |