rename yolonas to post_process_yolonas

This commit is contained in:
MarcA711 2024-05-17 16:34:10 +00:00
parent a92bd60742
commit aded9c91a3

View File

@ -22,7 +22,7 @@ class DetectionApi(ABC):
def detect_raw(self, tensor_input):
pass
def yolonas(self, output):
def post_process_yolonas(self, output):
"""
@param output: output of inference
expected shape: [np.array(1, N, 4), np.array(1, N, 80)]