Good day, I am trying to troubleshoot a JetBot issue, this is a Waveshare jetbot kit, installed using the latest Nvidia Image and then WaveShare’s installer.
When I attempt to run the object detection script
from jetbot import ObjectDetector
model = ObjectDetector('ssd_mobilenet_v2_coco.engine')
I get the following output.
---------------------------------------------------------------------------
AttributeError Traceback (most recent call last)
<ipython-input-1-78c468b59bd4> in <module>
1 from jetbot import ObjectDetector
2
----> 3 model = ObjectDetector('ssd_mobilenet_v2_coco.engine')
/usr/local/lib/python3.6/dist-packages/jetbot-0.3.0-py3.6.egg/jetbot/object_detection.py in __init__(self, engine_path, preprocess_fn)
26 load_plugins()
27 self.trt_model = TRTModel(engine_path, input_names=[TRT_INPUT_NAME],
---> 28 output_names=[TRT_OUTPUT_NAME, TRT_OUTPUT_NAME + '_1'])
29 self.preprocess_fn = preprocess_fn
30
/usr/local/lib/python3.6/dist-packages/jetbot-0.3.0-py3.6.egg/jetbot/tensorrt_model.py in __init__(self, engine_path, input_names, output_names, final_shapes)
57 with open(engine_path, 'rb') as f:
58 self.engine = self.runtime.deserialize_cuda_engine(f.read())
---> 59 self.context = self.engine.create_execution_context()
60
61 if input_names is None:
AttributeError: 'NoneType' object has no attribute 'create_execution_context'
Some have suggested this is due to TensorRT version. The version in this jetpack is 6.0 - some say I need 5. I have tried re-compiling TRT and a few other things to no avail.
Any advice appreciated, thanks.