fixed convertion issues

app.py

@@ -7,7 +7,7 @@ import numpy as np
 import torchvision.transforms.functional as tf
 from PIL import Image
 
-args = OmegaConf.load(os.path.join(f"./config/test.yaml"))
+args = OmegaConf.load(os.path.join(f"./config/demo.yaml"))
 
 log("Model loading")
 phnet = Inference(**args)

converter.py

@@ -14,7 +14,7 @@ from PIL import Image
 import cv2 as cv
 from onnxruntime import InferenceSession
 
-args = OmegaConf.load(os.path.join(f"./config/test.yaml"))
+args = OmegaConf.load(os.path.join(f"./config/demo.yaml"))
 
 log("Model loading")
 phnet = Inference(**args)

live_demo.py

@@ -45,9 +45,6 @@ async def main(queue):
         # live_matting_step(inp, back, stylematte))
         # res = await live_matting_step(inp, back, stylematte)
         log(f"{i} await")
-        # Display the resulting frame
-
-        # blurred_frame = cv.blur(frame, (10, 10))
         end = time.time()
         log(f"frames: {frame_count}, time: {end - begin}, fps: {frame_count/(end - begin) }")
 

requirements.txt

@@ -36,5 +36,5 @@ torchvision==0.12.0
 tornado==6.2
 tqdm==4.64.1
 transformers==4.28.1
-onnx==1.14.1
-onnxruntime==1.16.0
+# onnx==1.14.1
+# onnxruntime==1.16.0