From 93ef52e66f9644f4d164fb98f20e5c8dd19b464f Mon Sep 17 00:00:00 2001 From: Jordon Brooks Date: Sun, 13 Aug 2023 17:46:48 +0100 Subject: [PATCH] Removed uncompressed frame view and fixed predictor --- DeepEncode.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/DeepEncode.py b/DeepEncode.py index ba0827c..6ae73d9 100644 --- a/DeepEncode.py +++ b/DeepEncode.py @@ -35,11 +35,11 @@ def load_frame_from_video(video_file, frame_num): def predict_frame(uncompressed_frame): #display_frame = np.clip(cv2.cvtColor(uncompressed_frame, cv2.COLOR_BGR2RGB) * 255.0, 0, 255).astype(np.uint8) - cv2.imshow("uncomp", uncompressed_frame) + #cv2.imshow("uncomp", uncompressed_frame) - combined_feature, frame = preprocess_frame(uncompressed_frame) + frame = preprocess_frame(uncompressed_frame) - compressed_frame = MODEL.predict(np.expand_dims(combined_feature, axis=0))[0] + compressed_frame = MODEL.predict([np.expand_dims(frame, axis=0)])[0] display_frame = np.clip(cv2.cvtColor(compressed_frame, cv2.COLOR_BGR2RGB) * 255.0, 0, 255).astype(np.uint8) @@ -67,9 +67,17 @@ for i in range(total_frames): uncompressed_frame = load_frame_from_video(UNCOMPRESSED_VIDEO_FILE, frame_num=i) compressed_frame = predict_frame(uncompressed_frame) + compressed_frame = cv2.resize(compressed_frame, (width, height)) + compressed_frame = np.clip(compressed_frame * 255.0, 0, 255).astype(np.uint8) + compressed_frame = cv2.cvtColor(compressed_frame, cv2.COLOR_RGB2BGR) + out.write(compressed_frame) + #if i % 10 == 0: # Print progress every 10 frames + # print(f"Processed {i} / {total_frames} frames") + out.release() print("Compression completed.") +