从暗网的预训练权重中获取权重值的方法？

Question

我目前正在尝试在 C 中实现 YOLOv3 对象检测模型（仅检测，不训练）。

我已经用任意值测试了我的卷积方法，它似乎按我的预期工作。

在堆叠多个方法调用以进行前向传播之前，我认为使用实际预训练的权重文件数据进行测试是安全的。

当我查看 Darknet 的预训练权重文件时，它是一大块二进制文件。我尝试将其转换为十六进制和十进制，但要确定要使用的值的哪一部分仍然看起来并不简单。

所以，我的问题是，我应该如何提取权重或过滤器值的十进制数，以便我可以按照 YOLOv3 中发生的前向传播的相同顺序使用它们？

*我目前正在尝试使用https://www.itread01.com/content/1541167345.html中显示的结构图像构建我的C版本的YOLOv3

*我的 c 代码将与其他 HDL 代码一起在名为 MicroZed 的 FPGA 板上运行。

*我尝试将一些 printf 函数插入到 Darknet 代码的某些地方，以查看 YOLOv3 运行时正在移动的数据类型，但是，当我在 Linux 终端上运行它时，它没有显示任何新内容并保留输出相同的结果。

任何帮助或建议将不胜感激。谢谢！

Answer 1

我不太确定是否有直接读取暗网权重的方法，但是您可以将其转换为.h5 格式并从中获取权重值

您可以使用 this 存储库中的相应命令将暗网 yolov3 权重转换为 .h5 格式（由 keras 使用）。

您可以根据您的 Yolo 版本从链接存储库的自述文件中显示的列表中选择命令。对于标准的yolov3，转换的命令是

python tools/model_converter/convert.py cfg/yolov3.cfg weights/yolov3.weights weights/yolov3.h5 一旦你有了.h5weights，你可以使用下面的代码 sn-p 来获取 权重的值。 credit/source

import h5py

path = "<path to weights>.h5"
weights = {}
keys = []
with h5py.File(path, 'r') as f: # open file
  f.visit(keys.append) # append all keys to list
  for key in keys:
      if ':' in key: # contains data if ':' in key
          param_name = f[key].name
          weights[f[key].name] = f[key].value
          print(param_name,weights[f[key].name])