leon 5 달 전
부모
커밋
099e5228a7
1개의 변경된 파일9개의 추가작업 그리고 0개의 파일을 삭제
  1. 9 0
      test.py

+ 9 - 0
test.py

@@ -0,0 +1,9 @@
+from workspace import trtresnet
+import cv2
+
+infer = trtresnet.TrtResnetInfer("resnet.engine")
+
+image = cv2.imread("workspace/inference/car.jpg")
+result = infer.forward(image)
+
+print(result)