Update handler.py
Browse files- 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)
|