运行jetbot 脚本如下:
from jetbot import ObjectDetector
model = ObjectDetector('ssd_mobilenet_v2_coco.engine')
报错如下:
AttributeErrorTraceback (most recent call last)
in
1 from jetbot import ObjectDetector
----> 2 model = ObjectDetector('ssd_mobilenet_v2_coco.engine')
/usr/local/lib/python3.6/dist-packages/jetbot-0.4.1-py3.6.egg/jetbot/object_detection.py in init(self, engine_path, preprocess_fn)
27 load_plugins()
28 self.trt_model = TRTModel(engine_path, input_names=[TRT_INPUT_NAME],
---> 29 output_names=[TRT_OUTPUT_NAME, TRT_OUTPUT_NAME + '_1'])
30 self.preprocess_fn = preprocess_fn
31
/usr/local/lib/python3.6/dist-packages/jetbot-0.4.1-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'