guardiancc commited on
Commit
c5b4561
·
verified ·
1 Parent(s): 0e0a3f2

Update engine.py

Browse files
Files changed (1) hide show
  1. engine.py +2 -3
engine.py CHANGED
@@ -132,9 +132,8 @@ class Engine:
132
 
133
  _, frame = await self.transform_frame(processed_data, params)
134
  bgr_frame = cv2.cvtColor(np.array(frame), cv2.COLOR_BGR2RGB)
135
- new_frame = cv2.cvtColor(bgr_frame, cv2.COLOR_BGR2RGB)
136
- _, _, output = enhancer.enhance(new_frame, has_aligned=False, only_center_face=False, paste_back=True)
137
- video_writer.write(output)
138
 
139
  video_writer.release()
140
  cap.release()
 
132
 
133
  _, frame = await self.transform_frame(processed_data, params)
134
  bgr_frame = cv2.cvtColor(np.array(frame), cv2.COLOR_BGR2RGB)
135
+ new_frame = cv2.cvtColor(bgr_frame, cv2.COLOR_BGR2RGB)
136
+ video_writer.write(new_frame)
 
137
 
138
  video_writer.release()
139
  cap.release()