trtResnet/test.py

from workspace import trtresnet
import cv2

infer = trtresnet.TrtResnetInfer("workspace/resnet.engine")

# image = cv2.imread("workspace/inference/car.jpg")
result = infer.forward_path("workspace/inference/car.jpg")


print(result)

image = cv2.imread("workspace/inference/car.jpg")

result = infer.forward(image)

print(result)