freealise commited on
Commit
89b318d
1 Parent(s): a9b6fe2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -449,7 +449,7 @@ def optimize(v, d):
449
 
450
  dcolor = []
451
  for k, f in enumerate(frames):
452
- frame = cv2.imread(frames[k]).convert('RGB').astype(np.uint8)
453
 
454
  # convert to np.float32
455
  f = np.float32(frame.reshape((-1,3)))
 
449
 
450
  dcolor = []
451
  for k, f in enumerate(frames):
452
+ frame = cv2.cvtColor(cv2.imread(frames[k]).astype(np.uint8), cv2.COLOR_BGR2RGB)
453
 
454
  # convert to np.float32
455
  f = np.float32(frame.reshape((-1,3)))