s444463/projekt_widzenie
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

processing fix

Browse Source
This commit is contained in:
Kamil Guttmann 2023-01-29 17:48:12 +01:00
parent 0a64ba8b19
commit d3afeb0e3b
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
process_video.py
View File

@ -16,15 +16,17 @@ def segment_video(video, fps=5):
result = [] result = []
i=0 i=0
num = 0
while True: while True:
ret, frame = video.read() ret, frame = video.read()
if ret == False: if ret == False:
break break
if i % n == 0: if i % n == 0:
result.append(frame) result.append(frame)
num += 1
i += 1 i += 1
return result, i return result, num
def save_frames(frames, dir): def save_frames(frames, dir):
for i, frame in enumerate(frames): for i, frame in enumerate(frames):
@ -41,7 +43,7 @@ def classify(img, model):
def read_saved_frames(dir, n): def read_saved_frames(dir, n):
result = [] result = []
for i in range(n+1): for i in range(n):
img = tf.keras.utils.load_img(f"{dir}/frame{i}.jpg", target_size = [224, 224]) img = tf.keras.utils.load_img(f"{dir}/frame{i}.jpg", target_size = [224, 224])
result.append(img) result.append(img)
return result return result
@ -52,6 +54,7 @@ if __name__ == "__main__":
model = tf.keras.models.load_model('model_pred/VGG16_sign_char_detection_model') model = tf.keras.models.load_model('model_pred/VGG16_sign_char_detection_model')
frames, num = segment_video(video, 20) frames, num = segment_video(video, 20)
print(num)
save_frames(frames, "frames") save_frames(frames, "frames")
frames = read_saved_frames("frames", num) frames = read_saved_frames("frames", num)
result = [] result = []