Сравнение изображения со столбцом csv в л oop python - PullRequest
Новая
       56

Сравнение изображения со столбцом csv в л oop python

1 голос
/ 09 июля 2020

Gretings,

Проблема: Я застрял в al oop. Я хочу, чтобы l oop выбирал изображение и сравнивал его со столбцом (именем) csv, если он существует, а затем переходил к изменению размера. Изображения были в архиве, сначала я вынул их из архива; 

from zipfile import ZipFile
file_name = "human_detection_images.zip"
with ZipFile(file_name, 'r') as zip:
   zip.extractall()
print('Done')

PATH_TO_TEST_IMAGES_DIR = pathlib.Path('/content/human_detection_images')
TEST_IMAGE_PATHS = sorted(list(PATH_TO_TEST_IMAGES_DIR.glob("*.jpg")))
TEST_IMAGE_PATHS

Затем я загружаю csv, в котором есть ярлыки и названия изображений (такие же, как у папки, но не последовательно). 

import pandas as pd
import numpy as np
data1 = pd.read_csv('/content/csvnew.csv')
df = pd.DataFrame(data1)
df.shape,df.columns

Теперь мне нужно запустить al oop, который выбирает изображения 1 на 1 и сравнивает их с именами в csv. Если похоже, измените размер изображения. доступен код изменения размера, 

for image_path in TEST_IMAGE_PATHS:
   if TEST_IMAGE_PATHS == df['names']:
     print("done") #error here: it prints nothing
  #  return False
     imgs = read_given_images("./",TEST_IMAGE_PATHS)
     points = df['labels']
     points = np.array(points)

# resizing starts

Мне нужна помощь с l oop. если кому пожалуйста ..

1 Ответ

0 голосов
/ 23 июля 2020

Это просто. Сначала выберите имя из csv и найдите его в папке. ваш код: 

     imgs = read_given_images("./",TEST_IMAGE_PATHS)
 points = df['labels']
 points = np.array(points)

измените его на: 

imgs = read_given_images("/content/human_detection_images",df.names.values)
points = df['labels'].values
points = np.array(points)

таким образом изображения и точки будут собраны вместе.

...