fix
Browse files
app.py
CHANGED
@@ -84,11 +84,10 @@ def get_top_k(probs, k=5, label_map=CLASSES):
|
|
84 |
return tuple(zip(top_labels, top_probs))
|
85 |
|
86 |
def inferenece(video):
|
87 |
-
video = frames_from_video_file(video,
|
88 |
clips = tf.split(video, video.shape[1], axis=1)
|
89 |
states = init_states
|
90 |
for clip in clips:
|
91 |
-
# Input shape: [1, 1, 172, 172, 3]
|
92 |
outputs = runner(**states, image=clip)
|
93 |
logits = outputs.pop('logits')[0]
|
94 |
states = outputs
|
|
|
84 |
return tuple(zip(top_labels, top_probs))
|
85 |
|
86 |
def inferenece(video):
|
87 |
+
video = frames_from_video_file(video, 13)
|
88 |
clips = tf.split(video, video.shape[1], axis=1)
|
89 |
states = init_states
|
90 |
for clip in clips:
|
|
|
91 |
outputs = runner(**states, image=clip)
|
92 |
logits = outputs.pop('logits')[0]
|
93 |
states = outputs
|