Объединение данных изображения и меток вместе - PullRequest
Новая
       37

Объединение данных изображения и меток вместе

2 голосов
/ 23 апреля 2020

У меня есть материнская папка, которая содержит несколько подпапок, в которых я храню изображения номерных знаков. В материнском каталоге также находится папка с именем train Val, которая представляет собой csv-файл, который содержит метки данных. trainVal содержит следующие столбцы:

  • track_id - идентификатор заданного c трека на основе трекера
  • image_path - путь к изображению в структуре архива
  • lp - земля Текст истины для номерного знака
  • train - Поезд / тестовый сплит. 0 - тест, 1 - поезд

TrainVal

На мой взгляд, track_id не нужен для вывода меток (метки - "lp", основной текст истины, например, 7C24698) и номерные знаки вместе. Мне интересно, как их собрать.

Я читаю на изображениях с кодом ниже: 

import cv2
import numpy
import glob
import matplotlib.pyplot as plt

folders = glob.glob(r'path\to\mother\directory\*')
imagenames_list = []
for folder in folders:
    for f in glob.glob(folder+'/*.png'):
        imagenames_list.append(f)

read_images = []        

for image in imagenames_list:
    read_images.append(cv2.imread(image, cv2.IMREAD_GRAYSCALE))

Я читаю в trainVal данные, используя этот код: 

import csv

track_id = []
image_path = []
lp = []
train = []

with open(r'path\to\trainVal.csv') as csvDataFile:
    csvReader = csv.reader(csvDataFile)
    for row in csvReader:
        track_id.append(row[0])
        image_path.append(row[1])
        lp.append(row[2])
        train.append(row[3])

Теперь я получил метки и все другие данные trainVal в виде массивов.

Есть ли способ объединить метки и изображения, расширив код, который я использовал, или мне нужно читать в изображениях по-другому или я не должен делать что-то подобное?

...