Update app.py

app.py

@@ -26,15 +26,16 @@ main.py
 
 def mainTest(inputpath, outpath):
     watermark = deep_nude_process(inputpath)
-    #cv2.imwrite(outpath, watermark)
+    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 = (item)
+    dress = cv2.imread(item)
+    dress = cv2.cvtColor(dress, cv2.COLOR_BGRA2RGBA)
     h = dress.shape[0]
     w = dress.shape[1]
     dress = cv2.resize(dress, (512, 512), interpolation=cv2.INTER_CUBIC)
@@ -45,13 +46,15 @@ def deep_nude_process(item):
 
 def inference(img):
     global index
-    # inputpath = "input/" + str(index) + ".jpg"
+    bgra = cv2.cvtColor(img, cv2.COLOR_RGBA2BGRA)
+    inputpath = "input_" + str(index) + ".jpg"
+    cv2.imwrite(inputpath, bgra)
+
     outputpath = "out_" + str(index) + ".jpg"
-    # cv2.imwrite(inputpath, img)
     index += 1
-    print(time.strftime("开始时间!!!!!!!!! %Y-%m-%d %H:%M:%S", time.localtime()))
-    output = mainTest(img, outputpath)
-    print(time.strftime("完成时间!!!!!!!!! %Y-%m-%d %H:%M:%S", time.localtime()))
+    print(time.strftime("开始!!!!!!!!! %Y-%m-%d %H:%M:%S", time.localtime()))
+    output = mainTest(inputpath, outputpath)
+    print(time.strftime("结束!!!!!!!!! %Y-%m-%d %H:%M:%S", time.localtime()))
     return output