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
|