在没有暗网/火炬的情况下创建预训练权重以进行检测

Create pre-trained weights for detection without darknet / torch

我想训练我的数据并创建一个权重文件来训练我的 YOLOV3 网络。 我数据库中的对象不在 COCO 的 classes 中,所以我不想使用它们的权重文件。 我也有一个限制 - 我不能使用 darknet,我也不喜欢使用 torch 库。

我拥有的文件:(根据https://github.com/AlexeyAB/Yolo_mark/issues/60#issuecomment-401854885

  1. yolov3-custom.cfg 根据我的 classes
  2. obj.names 加上我 class 的名字
  3. train.txt + test.txt 图片位置列表
  4. 包含图像+标签的文件夹 yolo 格式(对象-class、x、y、宽度、高度)
  5. obj.data

我能做什么?

如果您想从预训练的权重开始,您仍然可以在 ImageNet 上使用预训练的权重。如果您的 类 与 COCO 数据集不同,那没问题。您可以定义自己的 类,然后使用预训练的权重开始训练。在训练期间,权重将更新以检测您的自定义 类。

您可以将这些权重用于 YOLOv3:“https://pjreddie.com/media/files/yolov3.weights”

我按照github训练了我的习惯。com/qqwweee/keras-yolo3