как получить имя изображения с помощью python open-cv

Question

Я читаю одно изображение с помощью функции cv2.imread ().Я хочу сохранить заголовок изображения 10_left.jpeg в одной переменной, а это имя записать в один файл CSV.код как ниже:

import cv2

img = cv2.imread('home/pycharmprojects/diabetic/testing/10_left.jpeg')
print("shape of original image ",img.shape)
cv2.imshow('Orgninal Image',img)

Answer 1

Вы можете создать собственный класс для объекта изображения и сохранить фактическое изображение и имя файла в отдельных переменных.

class MyImage:
    img = None
    name = ''

    def __init__(self,name):
        self.name = name
        self.img = cv2.imread(name)

myImage = MyImage('10_left.jpeg')

Чтобы записать имя в CSV-файл, просто получите значение атрибута myImage.name

Answer 2

Чтобы уточнить, вы хотите прочитать имя изображения из файла CSV и использовать его, чтобы открыть изображение с open-cv?В этом случае вам просто нужно проанализировать ваш CSV-файл с помощью CSV-модуля .Если ваш CSV - это просто строки с именами изображений, это довольно просто.

import cv2
import csv

ifile = open(‘test.csv’, “rb”)
reader = csv.reader(ifile)

imageNames = []
for row in reader
    imageNames.append(row)

img = cv2.imread(imageNames[0])
print("shape of original image ",img.shape)
cv2.imshow('Original Image',img)
cv2.waitkey(0)

Если ваш CSV более вовлечен, ссылка выше должна помочь вам в вашем пути.