search

talebolano / TensorRT-Yolov3

TensorRT for Yolov3
MIT License
49 stars 16 forks source link

yolov3-tiny #3

Open liteonandy opened 4 years ago

liteonandy commented 4 years ago

Hello, My yolov3-tiny always reports an error: ####### input args####### C=3; H=416; W=416; caffemodel=./yolov3-tiny.caffemodel; calib=; cam=0; class=80; classname=coco.name; display=1; evallist=; input=0; inputstream=cam; mode=fp32; nms=0.450000; outputs=yolo-det; prototxt=./yolov3-tiny-trt.prototxt; savefile=result; saveimg=0; videofile=sample.mp4; ####### end args####### init plugin proto: ./yolov3-tiny-trt.prototxt caffemodel: ./yolov3-tiny.caffemodel Begin parsing model... ERROR: layer21-route: all concat input tensors must have the same dimensions except on the concatenation axis runYolov3: ./parserHelper.h:97: nvinfer1::DimsCHW parserhelper::getCHW(const nvinfer1::Dims&): Assertion `d.nbDims >= 3' failed. Aborted (core dumped)

Can you give me your yolov3-tiny?

huiofficial commented 3 years ago

I hit with the same problem when I parse a SSD model. Did you have idea now?