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

Changed video and model paths to new location

Browse source
This commit is contained in:
Jordon Brooks 2023-07-24 16:51:56 +01:00
parent 58fcf819ee
commit 80c5f2216d
2 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

4
DeepEncode.py
View file

@ -7,10 +7,10 @@ from video_compression_model import VideoCompressionModel
NUM_CHANNELS = 3
# Step 2: Load the trained model
model = tf.keras.models.load_model('ai_rate_control_model.keras', custom_objects={'VideoCompressionModel': VideoCompressionModel})
model = tf.keras.models.load_model('models/model.keras', custom_objects={'VideoCompressionModel': VideoCompressionModel})
# Step 3: Load the uncompressed video
UNCOMPRESSED_VIDEO_FILE = 'test_video.mkv'
UNCOMPRESSED_VIDEO_FILE = 'test_data/test_video.mkv'
def load_frames_from_video(video_file, num_frames = 0):
print("Extracting video frames...")