search

faedtodd / Tensorrt-Yolov3-tiny

darknet2onnx2tensorrt
26 stars 10 forks source link

Yolov3-tiny .cfg file #2

Open HilaRobotican opened 5 years ago

HilaRobotican commented 5 years ago

Hi, can you please share your yolov3-tiny .cfg file? When I run onnx_to_tensorrt.py it fails to reshape on this line: trt_outputs = [output.reshape(shape) for output, shape in zip(trt_outputs, output_shapes)]

Thanks!

barzan-hayati commented 5 years ago

@HilaRobotican I used tiny-yolov3.cfg in AlexeyAB repo and trained it, but it has some problems in training: it could not find object in images. You could also use tiny-yolov3.cfg in Pjreddie repo.

Error in converting onnx to trt,

[TensorRT] ERROR: (Unnamed Layer* 38) [Concatenation]: all concat input tensors must have the same dimensions except on the concatenation axis #4