Browse Source

add save src face

ming 3 years ago
parent
commit
659f2186a3
1 changed files with 4 additions and 0 deletions
  1. 4 0
      api/main.py

+ 4 - 0
api/main.py

@@ -182,6 +182,10 @@ def gen_avatar(name_hash, imgurl):
     c._config['sync_request_timeout'] = None
     remote_svc = c.root
     my_answer = remote_svc.call_avatar(name_hash,imgurl) # method call
+
+    im = Image.open(requests.get(imgurl, stream=True).raw)
+    im= im.convert("RGB")
+    im.save('../../face_swap/tmp_face/'+name_hash+'.jpg')
    
     shutil.copy(tmp_avatar_dir+name_hash+'.mp4',avatar_dest+name_hash+'.mp4')
     os.remove(tmp_avatar_dir+name_hash+'.mp4')