yolov5 4.0版使用IR文件推理错误 invalid load key, '<'.
使用自定义训练yolov5 4.0后，把生成的last.pt转为IR文件 last.xml ；键入推理命令：
python detect.py --weights D:\pythonProject\yolov5-4_re\runs\train\exp8\weights\last.xml --source D:\pythonProject\yolov5-4_re\Hum_PetData\images\futi_vido
之后报错，报错如下
```python
(yolov5_v4) D:\pythonProject\yolov5-4_re>python detect.py --weights D:\pythonProject\yolov5-4_re\runs\train\exp8\weights\last.xml --source D:\pythonProject\yolov5-4_re\Hum_PetData\images\futi_vido
Namespace(agnostic_nms=False, augment=False, classes=None, conf_thres=0.25, device='', exist_ok=False, img_size=640, iou_thres=0.45, name='exp', project='runs/detect', save_conf=False, save_txt=False, source
='D:\\pythonProject\\yolov5-4_re\\Hum_PetData\\images\\futi_vido', update=False, view_img=False, weights=['D:\\pythonProject\\yolov5-4_re\\runs\\train\\exp8\\weights\\last.xml'])
Using torch 1.11.0 CUDA:0 (NVIDIA GeForce RTX 3060 Laptop GPU, 6143.5MB)
Traceback (most recent call last):
File "detect.py", line 172, in <module>
detect()
File "detect.py", line 33, in detect
model = attempt_load(weights, map_location=device) # load FP32 model
File "D:\pythonProject\yolov5-4_re\models\experimental.py", line 118, in attempt_load
model.append(torch.load(w, map_location=map_location)['model'].float().fuse().eval()) # load FP32 model
File "D:\Anaconda\envs\yolov5_v4\lib\site-packages\torch\serialization.py", line 713, in load
return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
File "D:\Anaconda\envs\yolov5_v4\lib\site-packages\torch\serialization.py", line 920, in _legacy_load
magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: invalid load key, '<'.
last.pt文件读取的内容如下:貌似读取到第一个引号就报错了‘<’
<?xml version="1.0" ?>
<net name="D:/pythonProject/yolov5-4_re/runs/train/exp8/weights/last" version="10">
<layers>
<layer id="0" name="images" type="Parameter" version="opset1">
<data shape="1, 3, 640, 640" element_type="f32"/>
<output>
<port id="0" precision="FP32" names="images">
<dim>1</dim>
<dim>3</dim>
<dim>640</dim>
<dim>640</dim>
</port>
</output>
</layer>
<layer id="1" name="Slice_4/Concat380013317" type="Const" version="opset1">
<data element_type="i64" shape="4" offset="0" size="32"/>
<output>
<port id="0" precision="I64">
<dim>4</dim>
</port>
</output>
</layer>
<layer id="2" name="Slice_4/Concat3784380113356" type="Const" version="opset1">
<data element_type="i64" shape="4" offset="32" size="32"/>
<output>
<port id="0" precision="I64">
<dim>4</dim>
</port>
</output>
</layer>
<layer id="3" name="Slice_4/Strides380213248" type="Const" version="opset1">
<data element_type="i64" shape="4" offset="64" size="32"/>
<output>
<port id="0" precision="I64">
<dim>4</dim>
</port>
</output>
</layer>
...
操作环境、软件版本等信息
操作环境：win10+Using torch 1.11.0 CUDA:0 (NVIDIA GeForce RTX 3060 Laptop GPU, 6143.5MB)
环境列表：
# packages in environment at D:\Anaconda\envs\yolov5_v4:
#
# Name Version Build Channel
addict 2.4.0 pypi_0 pypi
beautifulsoup4 4.10.0 pypi_0 pypi
blas 1.0 mkl https://mirrors.ustc.edu.cn/anaconda/pkgs/free
brotlipy 0.7.0 py37h2bbff1b_1003 https://mirrors.ustc.edu.cn/anaconda/pkgs/main
bs4 0.0.1 pypi_0 pypi
ca-certificates 2022.3.29 haa95532_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main
cachetools 5.0.0 pypi_0 pypi
certifi 2021.10.8 py37haa95532_2 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
cffi 1.15.0 py37h2bbff1b_1 https://mirrors.ustc.edu.cn/anaconda/pkgs/main
charset-normalizer 2.0.12 pypi_0 pypi
click 8.1.2 pypi_0 pypi
colorama 0.4.4 pypi_0 pypi
coremltools 5.2.0 pypi_0 pypi
cryptography 36.0.0 py37h21b164f_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main
cudatoolkit 11.3.1 h59b6b97_2 https://mirrors.ustc.edu.cn/anaconda/pkgs/main
cycler 0.11.0 pypi_0 pypi
cython 0.29.28 pypi_0 pypi
docker-pycreds 0.4.0 pypi_0 pypi
easydict 1.9 pypi_0 pypi
fast-ctc-decode 0.3.0 pypi_0 pypi
filelock 3.6.0 pypi_0 pypi
flask 2.1.1 pypi_0 pypi
fonttools 4.30.0 pypi_0 pypi
gitdb 4.0.9 pypi_0 pypi
gitpython 3.1.27 pypi_0 pypi
google-auth 2.6.0 pypi_0 pypi
google-auth-oauthlib 0.4.6 pypi_0 pypi
html5lib 1.1 pypi_0 pypi
huggingface-hub 0.4.0 pypi_0 pypi
idna 3.3 pyhd3eb1b0_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main
imagecodecs 2021.11.20 pypi_0 pypi
imageio 2.15.0 pypi_0 pypi
imgaug 0.4.0 pypi_0 pypi
importlib-metadata 4.11.3 pypi_0 pypi
intel-openmp 2021.4.0 haa95532_3556 https://mirrors.ustc.edu.cn/anaconda/pkgs/main
itsdangerous 2.1.2 pypi_0 pypi
jinja2 3.1.1 pypi_0 pypi
joblib 1.1.0 pypi_0 pypi
jsonpatch 1.32 pypi_0 pypi
jsonpointer 2.2 pypi_0 pypi
jstyleson 0.0.2 pypi_0 pypi
kiwisolver 1.4.0 pypi_0 pypi
labelimg 1.8.6 pypi_0 pypi
libuv 1.40.0 he774522_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main
lmdb 1.3.0 pypi_0 pypi
lxml 4.8.0 pypi_0 pypi
markdown 3.3.6 pypi_0 pypi
markupsafe 2.1.1 pypi_0 pypi
matplotlib 3.5.1 pypi_0 pypi
mkl 2021.4.0 haa95532_640 https://mirrors.ustc.edu.cn/anaconda/pkgs/main
mkl-service 2.4.0 py37h2bbff1b_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main
mkl_fft 1.3.1 py37h277e83a_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main
mkl_random 1.2.2 py37hf11a4ad_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main
mpmath 1.2.1 pypi_0 pypi
nibabel 3.2.2 pypi_0 pypi
ninja 1.10.2.3 pypi_0 pypi
nltk 3.7 pypi_0 pypi
numpy 1.19.0 pypi_0 pypi
numpy-base 1.21.5 py37hc2deb75_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main
oauthlib 3.2.0 pypi_0 pypi
onnx 1.11.0 pypi_0 pypi
onnx-simplifier 0.3.7 pypi_0 pypi
onnxoptimizer 0.2.6 pypi_0 pypi
onnxruntime 1.10.0 pypi_0 pypi
onnxruntime-gpu 1.11.0 pypi_0 pypi
opencv-python 4.5.5.64 pypi_0 pypi
openssl 1.0.1k 1 https://mirrors.ustc.edu.cn/anaconda/pkgs/free
openvino 2022.1.0 pypi_0 pypi
openvino-dev 2022.1.0 pypi_0 pypi
openvino-telemetry 2022.1.1 pypi_0 pypi
packaging 21.3 pypi_0 pypi
pandas 1.1.5 pypi_0 pypi
parasail 1.2.4 pypi_0 pypi
pathtools 0.1.2 pypi_0 pypi
pillow 9.1.0 pypi_0 pypi
pip 21.2.4 py37haa95532_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
progress 1.6 pypi_0 pypi
promise 2.3 pypi_0 pypi
protobuf 3.19.4 pypi_0 pypi
psutil 5.9.0 pypi_0 pypi
py-cpuinfo 8.0.0 pypi_0 pypi
pyasn1 0.4.8 pypi_0 pypi
pyasn1-modules 0.2.8 pypi_0 pypi
pyclipper 1.3.0.post2 pypi_0 pypi
pycocotools 2.0.4 pypi_0 pypi
pycparser 2.21 pyhd3eb1b0_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main
pydicom 2.3.0 pypi_0 pypi
pyhocon 0.3.59 pypi_0 pypi
pyopenssl 22.0.0 pyhd3eb1b0_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main
pyparsing 2.4.7 pypi_0 pypi
pyqt5 5.15.4 pypi_0 pypi
pyqt5-plugins 5.15.4.2.2 pypi_0 pypi
pyqt5-qt5 5.15.2 pypi_0 pypi
pyqt5-sip 12.9.1 pypi_0 pypi
pyqt5-tools 5.15.4.3.2 pypi_0 pypi
pysocks 1.7.1 py37_1 https://mirrors.ustc.edu.cn/anaconda/pkgs/main
python 3.7.1 h33f27b4_4 https://mirrors.ustc.edu.cn/anaconda/pkgs/main
python-dateutil 2.8.2 pypi_0 pypi
python-dotenv 0.20.0 pypi_0 pypi
pytorch 1.11.0 py3.7_cuda11.3_cudnn8_0 pytorch
pytorch-mutex 1.0 cuda https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch
pytz 2021.3 pypi_0 pypi
pywavelets 1.3.0 pypi_0 pypi
pyyaml 6.0 pypi_0 pypi
pyzmq 22.3.0 pypi_0 pypi
qt5-applications 5.15.2.2.2 pypi_0 pypi
qt5-tools 5.15.2.1.2 pypi_0 pypi
rawpy 0.17.1 pypi_0 pypi
regex 2022.3.15 pypi_0 pypi
requests 2.27.1 pyhd3eb1b0_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main
requests-oauthlib 1.3.1 pypi_0 pypi
rsa 4.8 pypi_0 pypi
sacremoses 0.0.49 pypi_0 pypi
scikit-image 0.19.2 pypi_0 pypi
scikit-learn 0.24.2 pypi_0 pypi
scipy 1.5.4 pypi_0 pypi
seaborn 0.11.2 pypi_0 pypi
sentencepiece 0.1.96 pypi_0 pypi
sentry-sdk 1.5.8 pypi_0 pypi
setproctitle 1.2.2 pypi_0 pypi
setuptools 58.0.4 py37haa95532_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
shapely 1.8.1.post1 pypi_0 pypi
shortuuid 1.0.8 pypi_0 pypi
six 1.16.0 pyhd3eb1b0_1 https://mirrors.ustc.edu.cn/anaconda/pkgs/main
sklearn 0.0 pypi_0 pypi
smmap 5.0.0 pypi_0 pypi
soupsieve 2.3.1 pypi_0 pypi
sympy 1.10.1 pypi_0 pypi
tensorboard 2.8.0 pypi_0 pypi
tensorboard-data-server 0.6.1 pypi_0 pypi
tensorboard-plugin-wit 1.8.1 pypi_0 pypi
tensorflow-hub 0.12.0 pypi_0 pypi
tensorflowjs 3.15.0 pypi_0 pypi
texttable 1.6.4 pypi_0 pypi
thop 0.0.31-2005241907 pypi_0 pypi
threadpoolctl 3.1.0 pypi_0 pypi
tifffile 2021.11.2 pypi_0 pypi
tokenizers 0.10.3 pypi_0 pypi
torch 1.11.0 pypi_0 pypi
torchfile 0.1.0 pypi_0 pypi
torchvision 0.12.0 pypi_0 pypi
tornado 6.1 pypi_0 pypi
tqdm 4.63.0 pypi_0 pypi
transformers 4.16.2 pypi_0 pypi
typing_extensions 4.1.1 pyh06a4308_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main
urllib3 1.26.8 pyhd3eb1b0_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main
vc 14.2 h21ff451_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
visdom 0.1.8.9 pypi_0 pypi
vs2015_runtime 14.27.29016 h5e58377_2 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
webencodings 0.5.1 pypi_0 pypi
websocket-client 1.3.2 pypi_0 pypi
werkzeug 2.0.3 pypi_0 pypi
wheel 0.37.1 pyhd3eb1b0_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
win_inet_pton 1.1.0 py37haa95532_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main
wincertstore 0.2 py37haa95532_2 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
yaspin 2.1.0 pypi_0 pypi
you-get 0.4.1555 pypi_0 pypi
zipp 3.7.0 pypi_0 pypi
尝试过的解决方法
1.这个小哥有和我相同的问题，但没提到怎么解决：https://stackoverflow.com/questions/63063723/how-to-fix-pickle-module-loadf-pickle-load-args-pickle-unpicklingerror-i
我想要达到的结果
使用IR的xml文件进行视频推理，查看推理速度和帧数，想要查看yolov5上部署openvino的效果。感谢您的帮助！~
