kumuji / trt_yolo_ros

yolov3 ros node using tensorrt acceleration
12 stars 4 forks source link

customized yolov4-tiny #9

Open Mafumaful opened 2 years ago

Mafumaful commented 2 years ago

when I tried to run the trt with yolov4-tiny: python3: yolo_layer.cu:71: nvinfer1::YoloLayerPlugin::YoloLayerPlugin(const void*, size_t): Assertion `d == a + length' failed. this is my device thanks for your code!!