aki-0421 commited on
Commit
cb40006
·
unverified ·
1 Parent(s): e5cd8c6
Files changed (1) hide show
  1. 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 = cv2_to_pil(frame)
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