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