- traffic_data
- images
- 内含图像
- yolo_label
- 内含标签文件,除了后缀,文件名与图像名一一对应
- new_label
- 空文件夹
- traffic_light_one_73
- images
- train
-空文件夹 - val
-空文件夹
- train
- images
- train
-空文件夹 - val
-空文件夹
- train
- images
- get_one.ipynb
- show_label_box_on_image.ipynb
traffic_data为根目录,以下代码存放在show_label_box_on_image.ipynb
1)构建函数
- images
import cv2
import os
def draw_box_in_single_image(image_path, txt_path):
# 读取图像
image = cv2.imread(image_path)
# 读取txt文件信息
def read_list(txt_path):
pos = []
with open(txt_path, 'r') as file_to_read:
while True:
lines = file_to_read.readline() # 整行读取数据
if not lines:
break
pass
# 将整行数据分割处理,如果分割符是空格,括号里就不用传入参数,如果是逗号, 则传入‘,'字符。
p_tmp = [float(i) for i