Update app.py

app.py

@@ -24,18 +24,19 @@ main.py
 """
 
 
-def mainTest(inputpath, outpath):
-    watermark = deep_nude_process(inputpath)
-    watermark1 = cv2.cvtColor(watermark, cv2.COLOR_RGBA2BGRA)
-    cv2.imwrite(outpath, watermark1)
+def mainTest(bgra, outpath):
+    watermark = deep_nude_process(bgra)
+    #watermark1 = cv2.cvtColor(watermark, cv2.COLOR_RGBA2BGRA)
+    #cv2.imwrite(outpath, watermark1)
     return watermark
     #
 
 
 def deep_nude_process(item):
     # print('Processing {}'.format(item))
-    dress = cv2.imread(item)
-    dress = cv2.cvtColor(dress, cv2.COLOR_BGRA2RGBA)
+    #dress = cv2.imread(item)
+    #dress = cv2.cvtColor(dress, cv2.COLOR_BGRA2RGBA)
+    dress = item
     h = dress.shape[0]
     w = dress.shape[1]
     dress = cv2.resize(dress, (512, 512), interpolation=cv2.INTER_CUBIC)
@@ -47,13 +48,13 @@ def deep_nude_process(item):
 def inference(img):
     global index
     bgra = cv2.cvtColor(img, cv2.COLOR_RGBA2BGRA)
-    inputpath = "input_" + str(index) + ".jpg"
-    cv2.imwrite(inputpath, bgra)
+    #inputpath = "input_" + str(index) + ".jpg"
+    #cv2.imwrite(inputpath, bgra)
 
     outputpath = "out_" + str(index) + ".jpg"
     index += 1
     print(time.strftime("开始!!!!!!!!! %Y-%m-%d %H:%M:%S", time.localtime()))
-    output = mainTest(inputpath, outputpath)
+    output = mainTest(bgra, outputpath)
     print(time.strftime("结束!!!!!!!!! %Y-%m-%d %H:%M:%S", time.localtime()))
     return output