search

PengyiZhang / SlimYOLOv3

This page is for the SlimYOLOv3: Narrower, Faster and Better for UAV Real-Time Applications
1.14k stars 320 forks source link

Converting pruned model to tensorRT #52

Open dsgh2 opened 5 years ago

dsgh2 commented 5 years ago

Hi,

I am trying to convert normal pruned yolo v3(not spp) to tensorRT model. But there are errors like below,

ERROR: conv_85: at least three non-batch dimensions are required for input ERROR: Parameter check failed at: C:\p4sw\sw\gpgpu\MachineLearning\DIT\release\5.1\builder\Network.cpp::nvinfer1::Network::addScale::167, condition: inputTensor.getDimensions().nbDims >= 3

Have someone who tried to this?

Thanks in advance.