Omnibus commited on
Commit
1c81419
·
1 Parent(s): eb1adc6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -50,10 +50,10 @@ def capture_function(vid=None):
50
  for i in range(int(frame_count1)):
51
  capture.set(cv2.CAP_PROP_POS_FRAMES, i)
52
  ret, frame_f = capture.read(i)
53
- frame_ff = cv2.cvtColor(frame_f, cv2.COLOR_BGR2RGB)
54
- cv2.imwrite(f'{uid}-frames/{i}.png',frame_ff)
55
  fbox.append(f'{uid}-frames/{i}.png')
56
- return fbox,fbox
57
 
58
  def im_2_vid(images,fps):
59
  #width, height = Image.open(images[0]).size
 
50
  for i in range(int(frame_count1)):
51
  capture.set(cv2.CAP_PROP_POS_FRAMES, i)
52
  ret, frame_f = capture.read(i)
53
+ #frame_ff = cv2.cvtColor(frame_f, cv2.COLOR_BGR2RGB)
54
+ cv2.imwrite(f'{uid}-frames/{i}.png',frame_f)
55
  fbox.append(f'{uid}-frames/{i}.png')
56
+ yield fbox,fbox
57
 
58
  def im_2_vid(images,fps):
59
  #width, height = Image.open(images[0]).size