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(image)

print(result)