Anti-hero commited on
Commit
6035dd2
1 Parent(s): 0263d84

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +0 -2
handler.py CHANGED
@@ -1,7 +1,6 @@
1
  import cv2
2
  import numpy as np
3
  import tensorflow as tf
4
- import time
5
  import random
6
  from typing import Any, Dict, List
7
 
@@ -22,7 +21,6 @@ class EndpointHandler():
22
  return top_labels[0]
23
 
24
  def perform_action_recognition(self, frame, k=1):
25
-
26
  outputs = self.model.signatures['serving_default'](image=frame)
27
  probs = tf.nn.softmax(outputs['classifier_head_1'])
28
  return self.get_top_k(probs[0], k=k)
 
1
  import cv2
2
  import numpy as np
3
  import tensorflow as tf
 
4
  import random
5
  from typing import Any, Dict, List
6
 
 
21
  return top_labels[0]
22
 
23
  def perform_action_recognition(self, frame, k=1):
 
24
  outputs = self.model.signatures['serving_default'](image=frame)
25
  probs = tf.nn.softmax(outputs['classifier_head_1'])
26
  return self.get_top_k(probs[0], k=k)