search

lewes6369 / TensorRT-Yolov3

TensorRT for Yolov3
MIT License
489 stars 165 forks source link

build engine failed on jetson tx2 #16

Open xudongyang080715 opened 5 years ago

xudongyang080715 commented 5 years ago

I use your caffemodel and prototxt. when I used this command :./install/runYolov3 --caffemodel=./yolov3_608.caffemodel --prototxt=./yolov3_608.prototxt --input=./dog.jpg --W=608 --H=608 --class=80 the output is : C=3; H=608; W=608; caffemodel=./yolov3_608.caffemodel; calib=; class=80; input=./dog.jpg; mode=fp32; nms=0.450000; outputs=yolo-det; prototxt=./yolov3_608.prototxt; ####### end args####### init plugin proto: ./yolov3_608.prototxt caffemodel: ./yolov3_608.caffemodel Begin parsing model... End parsing model... Begin building engine... runYolov3: ../builder/tacticOptimizer.cpp:2204: void nvinfer1::builder::{anonymous}::getFormatRequirements(nvinfer1::builder::{anonymous}::LeafCNode*, nvinfer1::builder::TacticSupply&): Assertion `!n->candidateRequirements.empty() && "no supported formats"' failed. Aborted (core dumped)

lewes6369 commented 5 years ago

I did not meet this problem. What is the code of tacticOptimizer.cpp ? Can you show me the TensorRT version? Thanks.

xudongyang080715 commented 5 years ago

My TensorRT version is 4.0.1 which belongs to jetpack 3.3.

xudongyang080715 commented 5 years ago

And I can't find file tacticOptimizer.cpp.

lewes6369 commented 5 years ago

@xudongyang080715 Can you try to update your tensorRT to 5.0.2.6 or 4.0.1.6 ? I only tested in these two version. The assert seems some supported issues in tensorrt .

uname0x96 commented 5 years ago

@xudongyang080715 you just change yolov3_608.prototxt to yolov3_608_trt.prototxt