Spaces:
Running
on
Zero
Running
on
Zero
aki-0421
commited on
F: fix
Browse files- pipeline.py +1 -4
pipeline.py
CHANGED
@@ -17,9 +17,6 @@ from vtdm.util import tensor2vid
|
|
17 |
|
18 |
remover = Remover(jit=False)
|
19 |
|
20 |
-
def cv2_to_pil(cv_image: np.ndarray) -> Image.Image:
|
21 |
-
return Image.fromarray(cv2.cvtColor(cv_image, cv2.COLOR_BGR2RGB))
|
22 |
-
|
23 |
|
24 |
def pil_to_cv2(pil_image: Image.Image) -> np.ndarray:
|
25 |
cv_image = np.array(pil_image)
|
@@ -118,7 +115,7 @@ class MultiViewGenerator:
|
|
118 |
result = []
|
119 |
|
120 |
for i, frame in enumerate(out_list):
|
121 |
-
input_image =
|
122 |
output_image = remover.process(input_image, type='rgba')
|
123 |
result.append(output_image)
|
124 |
|
|
|
17 |
|
18 |
remover = Remover(jit=False)
|
19 |
|
|
|
|
|
|
|
20 |
|
21 |
def pil_to_cv2(pil_image: Image.Image) -> np.ndarray:
|
22 |
cv_image = np.array(pil_image)
|
|
|
115 |
result = []
|
116 |
|
117 |
for i, frame in enumerate(out_list):
|
118 |
+
input_image = Image.fromarray(frame)
|
119 |
output_image = remover.process(input_image, type='rgba')
|
120 |
result.append(output_image)
|
121 |
|