azaion/ai-training
1
0
Fork 0
mirror of https://github.com/azaion/ai-training.git synced 2026-04-23 01:56:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix tensor rt engine

Browse Source
This commit is contained in:
zxsanny
2025-03-28 14:50:43 +02:00
committed by Alex Bezdieniezhnykh
parent 5b89a21b36
commit 06a23525a6
16 changed files with 272 additions and 94 deletions
Download Patch File Download Diff File Expand all files Collapse all files
inference/inference.py
+2 -3
View File
@@ -1,8 +1,7 @@
import cv2
import numpy as np
from onnx_engine import InferenceEngine
from dto import AnnotationClass, Annotation, Detection
from inference.dto import Annotation, Detection, AnnotationClass
from inference.onnx_engine import InferenceEngine
class Inference: