如何在 Yolo 3D 中获取边界框标签及其格式

Question

我已经在我的工作空间中实现了 Yolo 3D 模型，并且运行良好。我得到一个图像作为输出。所以我想知道在哪里可以获得 3D 边界框的标签。我什至想知道标签存储的格式是什么。

x,y,x,y... 格式或 x,y,z,x,y,z... 格式。

（编辑） 我已经实现了 Yolo 3D 模型： https://github.com/ruhyadi/YOLO3D

我下载了预训练的权重，并使用以下方法运行推理：

python inference.py \
    --weights yolov5s.pt \
    --source eval/image_2 \
    --reg_weights weights/resnet18.pkl \
    --model_select resnet18 \
    --output_path runs/ \
    --show_result --save_result

现在在运行文件夹中，我只得到一个 PNG 图像作为输出。我的输出是：

现在我什至想获得该图像中 3D 边界框的点，我什至想知道这些点的存储格式是 x,y,x,y.. 格式还是 x ,y,z,x,y,z..格式？

Answer 1

大家好，我已经找到了我的问题的解决方案。库文件夹中有一个 Math.py 脚本。 Math.py 中有一个函数 create_corners，我们可以在 inference.py 脚本中使用该函数来获取 3d 边界框的标签。我们的 3D 边界框的格式是 x,y,z,x,y,z... ，仅在 Math.py 脚本中提到。