kiokpam commited on
Commit
09f9169
·
1 Parent(s): e0c6250
Files changed (1) hide show
  1. app.py +1 -2
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, 20)
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