Spaces:
Build error
Build error
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 |
|