jordon/DeepEncode
Archived
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Removed unused variables from DeepEncode.py

Browse source
This commit is contained in:
Jordon Brooks 2023-08-13 15:55:52 +01:00
parent f43ef69f0d
commit 185e3fac9a
1 changed files with 1 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

5
DeepEncode.py
View file

@ -12,7 +12,6 @@ import cv2
from video_compression_model import VideoCompressionModel from video_compression_model import VideoCompressionModel
# Constants # Constants
CHUNK_SIZE = 24 # Adjust based on available memory and video resolution
COMPRESSED_VIDEO_FILE = 'compressed_video.avi' COMPRESSED_VIDEO_FILE = 'compressed_video.avi'
MAX_FRAMES = 0 # Limit the number of frames processed MAX_FRAMES = 0 # Limit the number of frames processed
@ -37,9 +36,8 @@ def predict_frame(uncompressed_frame):
#display_frame = np.clip(cv2.cvtColor(uncompressed_frame, cv2.COLOR_BGR2RGB) * 255.0, 0, 255).astype(np.uint8) #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)
cv2.waitKey(1)
combined_feature, _ = preprocess_frame(uncompressed_frame) combined_feature, frame = preprocess_frame(uncompressed_frame)
compressed_frame = MODEL.predict(np.expand_dims(combined_feature, axis=0))[0] compressed_frame = MODEL.predict(np.expand_dims(combined_feature, axis=0))[0]
@ -72,7 +70,6 @@ for i in range(total_frames):
compressed_frame = np.clip(compressed_frame * 255.0, 0, 255).astype(np.uint8) compressed_frame = np.clip(compressed_frame * 255.0, 0, 255).astype(np.uint8)
compressed_frame = cv2.cvtColor(compressed_frame, cv2.COLOR_RGB2BGR) compressed_frame = cv2.cvtColor(compressed_frame, cv2.COLOR_RGB2BGR)
out.write(compressed_frame) out.write(compressed_frame)
#cv2.imshow("output", compressed_frame)
out.release() out.release()
print("Compression completed.") print("Compression completed.")