postgresql - 如何在数据库的 jsonb 字段中保存 YOLO(You Only Look Once)对象检测的预测

Question

我想在大量图像上运行 Darknet(YOLO) 并将其预测存储在 PostgreSQL 数据库中。

这是我的表的结构:

sample=> \d+ prediction2;
                       Table "public.prediction2"
   Column    | Type  | Modifiers | Storage  | Stats target | Description 
-------------+-------+-----------+----------+--------------+-------------
 path        | text  | not null  | extended |              | 
 pred_result | jsonb |           | extended |              | 
Indexes:
    "prediction2_pkey" PRIMARY KEY, btree (path)

Darknet(YOLO) 的源文件是用 C 语言编写的。

我已经将 Caffe 的预测存储在数据库中，如下所示。作为示例，我在此处列出了我的数据库中的一行。

 path                                            |   pred_result                                                                                                                                           
-------------------------------------------------+------------------------------------------------------------------------------------------------------------------
 /home/reena-mary/Pictures/predict/gTe5gy6xc.jpg | {"bow tie": 0.00631, "lab coat": 0.59257, "neck brace": 0.00428, "Windsor tie": 0.01155, "stethoscope": 0.36260}

我想将 YOLO 的预测添加到 pred_result 的 jsonb 数据中，即对于已存储在数据库中的每个图像路径和 Caffe 预测结果，我想附加 Darknet(YOLO)的预测。

我想这样做的原因是为每个图像添加搜索标签。因此，通过在图像上运行 Caffe 和 Darknet，我希望能够获得足够的标签来帮助我更好地搜索图像。

请帮助我如何在 Darknet 中执行此操作。

Answer 1

我也遇到过这个问题。实际上YOLO并没有提供JSON输出接口(interface)，所以没有办法得到和Caffe一样的输出。

但是，您可以在此处合并一个拉取请求以获得可行的输出:https://github.com/pjreddie/darknet/pull/34/files .它输出 CSV 数据，您可以将其转换为 JSON 以存储在数据库中。

当然你也可以修改 YOLO 的源代码来制作你自己的直接输出 JSON 的实现。