python - 从边界框获取物体【物体检测】

Question

我有一个 .txt 文件，其中每行包含 path/to/image.jpg,xmin,ymin,xmax,ymax 以及一个 img 文件夹，其中包含jpg 图像。使用 python 提取每个文件坐标内的“对象”并查看边界框是否设置正确的最佳方法是什么？

谢谢!

Answer 1

您可以使用opencv-python库:

import cv2

# Reading the file
with open("filename.txt") as iostream:
    content = iostream.read()

# Visualizing the data
color = (0, 0, 255) # RED
for line in content.split("\n"):
    image_path, xmin, ymin, xmax, ymax = line.split(",")
    image = cv2.imread(image_path)
    pt1 = (int(xmin), int(ymin))
    pt2 = (int(xmax), int(ymax))
    cv2.rectangle(image, pt1, pt2, color)
    cv2.imshow("Visualization bounding box", image)
    cv2.waitKey()

此代码将显示每个图像，并用红色矩形来显示边界框。如果按任意键，它将切换到下一个键。

如果你想保存裁剪后的图像，你可以使用这样的东西:

    outfile = image_path[:-4] + "_bbox.jpg"
    outimage = image[int(ymin):int(ymax), int(xmin):int(xmax)]
    cv2.imwrite(outfile, outimage)