pytorch-yolov3训练、剪枝、转onnx/tensorrt 加速

简单记录下pytorch yolov3从训练到剪枝，onnx转trt的过程。有问题随时留言交流

一、pytorch-yolov3训练与剪枝
推荐博客2篇
（1）https://blog.csdn.net/dQCFKyQDXYm3F8rB0/article/details/95041551
（2）https://mp.weixin.qq.com/s?__biz=MzIwMTE1NjQxMQ==&mid=2247489169&idx=2&sn=f8b1c76a6657b8065d039df704df5913&chksm=96f364c5a184edd3bd3754854c494da4912b6fc90735ee1350ab031862aed5466881f41a68ab&scene=21#wechat_redirect（原作者github:https://github.com/coldlarry/YOLOv3-complete-pruning, fork后进行了步骤记录见step.txt. github: https://github.com/2017ND/yolov3_prune/）
二. weights——onnx——trt
感谢参照github:https://github.com/jkjung-avt/tensorrt_demos
按照readme进行操作，填写对应的cfg和weights文件。
注意修改：utils/yolov3.py 中line76 category_num=num_classes(检测种类）
line407 output_shapes中[(1, 255, h//32, w//32)]中255均改为(num_classes+5)*3