swap_face.py 931 B

12345678910111213141516171819202122232425262728
  1. import os
  2. import time
  3. import requests
  4. class swap_face():
  5. def __init__(self, imgurl):
  6. self.imgurl = imgurl
  7. def run(self):
  8. name_hash = str(time.time()).replace('.','')
  9. src_img = 'FaceSwap/src_img/'+name_hash+'.jpg'
  10. sv_path = '../html/swap_save'+name_hash+'.avi'
  11. try:
  12. im = Image.open(requests.get(self.imgurl, stream=True).raw)
  13. im= im.convert("RGB")
  14. im.save(src_img)
  15. except:
  16. return '圖片錯誤'
  17. os.system('python3 FaceSwap/main_video.py --src_img '+src_img+' --video_path nina_noGesture_adj.mp4 --correct_color --save_path '+sv_path)
  18. while True:
  19. print('waiting...')
  20. if os.path.exists(sv_path):
  21. break
  22. time.sleep(5)
  23. print('waiting...')
  24. return "人物生成完成 at : www.choozmo.com:8168/swap_save/"+name_hash+".avi"