Open HilaRobotican opened 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,
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!