如何将自己训练的yolo模型转换为tensorrt相关的模型(.trt/.engine)

如何将自己训练的yolo模型转为tensorrt相关的模型

• 【1

【1】.pt模型转.wts模型

注意 ：在完成.pt模型向.wts模型的转换过程中，首要需要明确的问题是自身训练所得的模型在存储时，究竟是仅保留了权重参数的.pt模型，还是同时保存了权重参数与结构信息的.pt模型
①针对仅保存权重参数的.pt模型
当模型存储时仅包含权重参数，此时可借助.cfg文件协同完成模型转换工作，具体转换途径如下：
1）从github平台克隆一个yolov3.git项目并进行解压

    git clone -b archive https://github.com/ultralytics/yolov3.git
    
    
      
    

2）从github平台克隆一个名为tensorrtx的仓库并进行解压操作

    git clone https://github.com/wang-xinyu/tensorrtx.git
    
    
      
    

3）将自主训练完成的yolov3-tiny模型复制至解压后的yolov3/weights目录内，同时将tensorrtx/yolov3-tiny文件夹中的gen_wts.py文件一并拷贝至yolov3对应的文件夹中
4）随