test.py 231 B

123456789
  1. from workspace import trtresnet
  2. import cv2
  3. infer = trtresnet.TrtResnetInfer("workspace/resnet.engine")
  4. # image = cv2.imread("workspace/inference/car.jpg")
  5. result = infer.forward_path("workspace/inference/car.jpg")
  6. print(result)