我如何将经过训练的模型(yolov4-obj_best)转换为 Pytorch(即)从 Darknet 到 Pytorch?

问题描述 投票:0回答:0

C:\Users\FHW>python demo_darknet2onnx.py yolov4-obj obj yolov4-obj_best train 64 追溯(最近一次通话): 文件“C:\Users\FHW\demo_darknet2onnx.py”,第 5 行,位于 “execution_count”:空, NameError:未定义名称“null”

目前我正在使用 Anaconda 环境中的 CUDA 11.2、CuDNN 11.2(v8.1.0.77)、OpenCV 4.5.2、Cmake 3.19.6 和 Python (3.10.9)。我目前正在使用 Yolov4 Darknet 进行自定义数据对象检测项目。成功完成模型的训练后,我想将训练好的模型(yolov4-obj_best)转换为Pytorch(即)从Darknet到Pytorch。这样我就可以将 Pytorch 模型转换为 onnx 模型以使用 TensorRT 进行推理。

我使用命令尝试了这个存储库“https://github.com/Tianxiaomo/pytorch-YOLOv4”

pip 安装 onnxruntime 蟒蛇 demo_darknet2onnx.py

执行的命令: pip install onnxruntime python demo_darknet2onnx.py yolov4-obj obj yolov4-obj_best 火车 64

yolov4-obj = 表示进行自定义配置更改的 cfg 文件 obj = 表示两个类的名称(即)杂草和胡萝卜 yolov4-obj_best = 训练后得到的权重 train = 这里的图片是 .txt 形式,其中包含图片的路径 64 =

在命令提示符下执行此命令“python demo_darknet2onnx.py yolov4-obj obj yolov4-obj_best train 64”后,我遇到了以下错误,我已将其附加到上面的错误消息中。我还更改了“执行计数”:从空到无。

我该如何解决这个错误?有人可以为我提供解决方案吗?或者谁能给我另一种方法来将我的 Darknet 权重更改为 ONNX 或 Pytorch?我将期待您的回复。谢谢。

pytorch nvidia onnx darknet tensorrt
© www.soinside.com 2019 - 2024. All rights reserved.