rashid996958 commited on
Commit
f69473d
·
1 Parent(s): 1319d42

add example images

Browse files
app.py CHANGED
@@ -37,11 +37,9 @@ def inference(A,B):
37
  Xs = graph.get_tensor_by_name('decoder_1/g:0')
38
  X = graph.get_tensor_by_name('X:0')
39
  Y = graph.get_tensor_by_name('Y:0')
40
- print("1")
41
  A_img, A_img_ = load_image(A)
42
  B_img, B_img_ = load_image(B)
43
  Xs_ = sess.run(Xs, feed_dict={X: A_img_, Y: B_img_})
44
- print("2")
45
  output = deprocess(Xs_)[0]
46
  output = np.array(np.array(output)*255,dtype=np.uint8)
47
  # output = cv2.cvtColor(output, cv2.COLOR_RGB2BGR)
@@ -66,11 +64,21 @@ def makeupTransfer(arr1,arr2):
66
  makeup = "input/ref.png"
67
  inference(no_makeup, makeup)
68
  return cv2.cvtColor(cv2.imread("output/scaled_2x.png"), cv2.COLOR_BGR2RGB)
 
 
 
 
 
 
 
 
69
 
70
  app = gr.Interface(fn=makeupTransfer,
71
  inputs=[gr.Image(label="Reference Image",type='numpy'),
72
  gr.Image(label="Makeup Image",type='numpy')],
73
  outputs=gr.Image(label="Makeup Transfer Image",type='numpy'),
74
- title="MakeUp Transfer APP")
 
 
75
 
76
- app.launch(share=True)
 
37
  Xs = graph.get_tensor_by_name('decoder_1/g:0')
38
  X = graph.get_tensor_by_name('X:0')
39
  Y = graph.get_tensor_by_name('Y:0')
 
40
  A_img, A_img_ = load_image(A)
41
  B_img, B_img_ = load_image(B)
42
  Xs_ = sess.run(Xs, feed_dict={X: A_img_, Y: B_img_})
 
43
  output = deprocess(Xs_)[0]
44
  output = np.array(np.array(output)*255,dtype=np.uint8)
45
  # output = cv2.cvtColor(output, cv2.COLOR_RGB2BGR)
 
64
  makeup = "input/ref.png"
65
  inference(no_makeup, makeup)
66
  return cv2.cvtColor(cv2.imread("output/scaled_2x.png"), cv2.COLOR_BGR2RGB)
67
+ examples = [
68
+ [,
69
+ 'faces/no_makeup/xfsy_0521.png',
70
+ 'faces/makeup/vFG756.png'],
71
+ [,
72
+ 'faces/no_makeup/xfsy_0068.png',
73
+ 'faces/makeup/XMY-136.png']
74
+ ]
75
 
76
  app = gr.Interface(fn=makeupTransfer,
77
  inputs=[gr.Image(label="Reference Image",type='numpy'),
78
  gr.Image(label="Makeup Image",type='numpy')],
79
  outputs=gr.Image(label="Makeup Transfer Image",type='numpy'),
80
+ title="MakeUp Transfer APP",
81
+ examples=examples
82
+ )
83
 
84
+ app.launch()
faces/makeup/XMY-014.png ADDED
faces/makeup/XMY-074.png ADDED
faces/makeup/XMY-136.png ADDED
faces/makeup/XMY-266.png ADDED
faces/makeup/vFG112.png ADDED
faces/makeup/vFG137.png ADDED
faces/makeup/vFG56.png ADDED
faces/makeup/vFG756.png ADDED
faces/makeup/vRX916.png ADDED
faces/no_makeup/XYUH-006.png ADDED
faces/no_makeup/vSYYZ306.png ADDED
faces/no_makeup/vSYYZ429.png ADDED
faces/no_makeup/vSYYZ639.png ADDED
faces/no_makeup/xfsy_0055.png ADDED
faces/no_makeup/xfsy_0068.png ADDED
faces/no_makeup/xfsy_0226.png ADDED
faces/no_makeup/xfsy_0401.png ADDED
faces/no_makeup/xfsy_0405.png ADDED
faces/no_makeup/xfsy_0503.png ADDED
faces/no_makeup/xfsy_0521.png ADDED