首页 > 解决方案 > ValueError: 本地文件 yolov3.weights 的 MD5 校验和不同于 c84e5b99d0e52cd466ae710cadf6d84c

问题描述

我遇到了一个问题,我试图用 tensorrt 对我定制的 YOLO 模型(yolov3.weight)进行预测。我在下面的步骤中被打动了(将 yolo 转换为 onnx)。我有自己的 yolov3.weights。

为了将 yolo 转换为 onxx,我使用了 /usr/src/tensorrt/samples/python/ 中的 python 文件 (yolov3_to_onnx.py)。在这个文件中,我正在加载我定制的训练有素的 yolov3.weight。

运行文件时

python2 yolov3_to_onnx.py

我收到以下错误

Traceback (most recent call last):
File "yolov3_to_onnx.py", line 812, in <module>
main()
File "yolov3_to_onnx.py", line 793, in main
'c84e5b99d0e52cd466ae710cadf6d84c')
File "yolov3_to_onnx.py", line 750, in download_file
(local_path, checksum_reference))
ValueError: The MD5 checksum of local file yolov3.weights differs from c84e5b99d0e52cd466ae710cadf6d84c, please manually remove the file and try again

我在这里很震惊,不知道如何找到本地文件yolov3.weights的MD5校验和。我相信 yolov3_to_onnx.py 正在使用预训练的 yolov3.weight 的 MD5 校验和。

我在互联网上搜索了 MD5 校验和,但不知道如何为我的自定义 yolov3.weight 找到这样的校验和。我也开始知道这对于检查真实性非常重要。

请查看我感到震惊的文件yolov3_to_onnx.py 。如需更多帮助,我将添加此文件中的代码片段,这是此问题的真实案例。

weights_file_path = download_file(
        'yolov3.weights',
        'https://pjreddie.com/media/files/yolov3.weights',
        'c84e5b99d0e52cd466ae710cadf6d84c')

我希望,如果 c84e5b99d0e52cd466ae710cadf6d84c 校验和被我体重的 md5 校验和替换,那么问题将得到解决,但我不知道如何找到我自定义的 yolov3 权重的校验和。

标签: yolotensorrt

解决方案


推荐阅读