zl程序教程

您现在的位置是:首页 >  后端

当前栏目

python工具方法 22 基于相对值描述的目标检测标注文件的可视化

Python文件方法工具 基于 检测 可视化 目标
2023-09-14 09:15:04 时间

虽然在目标检测中有labelme、labelimg等标注利器,但是在一些公开的数据集上,却不是使用这些公开的格式。而是一种基于长宽比描述的标准信息,如卫星应用赛题——海上船舶智能检测的数据集,使用相对值描述标注目标,具体如下图1所示:

图1 常见的标注格式

针对于这种txt格式所描述的标注信息,无法快速的进行可视化,因此开发python脚本实现对这类数据的快速可视化。

1、读取单个图片可视化并保存

import glob,os
import numpy as np
from skimage import io
from PIL import Image,ImageDraw,ImageFont

def get_label(img_path):
    label_path=img_path.replace(".jpg",".txt")
    label_list=[]
    with open(label_path,"r") as f:
        label_list = f.readl