需要将网络模型转成tensorrt格式，下采样部分用的是swintransformer，转换为tensorrt的时候会遇到算子报错的问题，所以在github上找到了一个修改好的swintransformer。但是需要安装pytorch-quantization，但是安装这个量化模块有点复杂，网上搜索了许多要换源什么的，最后指定了一下版本就安装成功了。

dir = Path('E:\\daima\\yolov5\\yolov5-6.0\\yolov5-6.0\\data')# 数据集路径，这里需要写到VOC数据集的上一级路径 在这一行代码中数据集还是第1点中的原始数据集将其放在data目录下（data目录一般里面都是数据集的.yaml文件）然后在data目录下会生成images和labels文件将这两个文件复制进datasets中的VOC2007

使用VScode的时候在运行train_network.py时，会出现找不到同文件夹下不同文件的错误，报错模式通常是模块错误，但是下载报错的模块显然不合理。首选项——>设置——>搜索Code-run——>扩展中找“Run Code…"——>找到下面图里这个选项并打勾——>重新启动。接下来的报错内容就很好解决了（下图是我打勾后运行出现的错误）