Omnibus commited on
Commit
1ee735f
·
1 Parent(s): 6fa5ca9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -115,8 +115,8 @@ def rem_bg(vid,frame_count):
115
  for i in range(int(frame_count1)):
116
  capture.set(cv2.CAP_PROP_POS_FRAMES, i)
117
  ret, frame_f = capture.read(i)
118
- frame_ff = cv2.cvtColor(frame_f, cv2.COLOR_BGR2RGB)
119
- out = rm(frame_ff)
120
  print (i)
121
  cv2.imwrite(f'{uid}-rembg/{i}.png',out)
122
  fbox2.append(f'{uid}-rembg/{i}.png')
 
115
  for i in range(int(frame_count1)):
116
  capture.set(cv2.CAP_PROP_POS_FRAMES, i)
117
  ret, frame_f = capture.read(i)
118
+ #frame_ff = cv2.cvtColor(frame_f, cv2.COLOR_BGR2RGB)
119
+ out = rm(frame_f)
120
  print (i)
121
  cv2.imwrite(f'{uid}-rembg/{i}.png',out)
122
  fbox2.append(f'{uid}-rembg/{i}.png')