search

mailrocketsystems / JetsonYolov5

GNU General Public License v3.0
23 stars 23 forks source link

[TensorRT] ERROR: INVALID_ARGUMENT: getPluginCreator could not find plugin YoloLayer_TRT version 1 #7

Closed VietNguyenHoang812 closed 11 months ago

VietNguyenHoang812 commented 11 months ago

I have followed the tutorial and run successfully with "./yolo_det -d custom_model.engine ../images" but when I tried with "python3 app.py", this error occurs: [TensorRT] ERROR: INVALID_ARGUMENT: getPluginCreator could not find plugin YoloLayer_TRT version 1 [TensorRT] ERROR: safeDeserializationUtils.cpp (323) - Serialization Error in load: 0 (Cannot deserialize plugin since corresponding IPluginCreator not found in Plugin Registry) [TensorRT] ERROR: INVALID_STATE: std::exception [TensorRT] ERROR: INVALID_CONFIG: Deserialize the cuda engine failed. Traceback (most recent call last): File "app.py", line 7, in model = YoloTRT(library="yolov5/build/libmyplugins.so", engine="yolov5/build/license_plate_detection.engine", conf=0.5, yolo_ver="v5") File "/home/jetson/Documents/JetsonYolov5_backup/yoloDet.py", line 39, in init self.batch_size = self.engine.max_batch_size AttributeError: 'NoneType' object has no attribute 'max_batch_size'

VietNguyenHoang812 commented 11 months ago

My bad, I modified the code and forgot to change it back