Как загрузить несколько изображений из одной папки? - PullRequest
Новая
       266

Как загрузить несколько изображений из одной папки?

1 голос
/ 24 января 2020

Я хочу загрузить несколько изображений из одной папки. Однако приведенный ниже код выдает ошибку: 

(TypeError Traceback (most recent call last)
<ipython-input-22-a30c12347c11> in <module>
2 import glob
3 
----> 4 image_list = map(Image.open, glob('/Users/name/images/*.jpg'))
5 
6 object_detection_api(image_list, rect_th=2, text_th=1, text_size=1)

TypeError: 'module' object is not callable)

Я уже проверил некоторые страницы устранения неполадок, но данное уведомление продолжает появляться. Что я должен изменить в коде? 

from PIL import Image
import glob

image_list = map(Image.open, glob('/Users/name/images/*.jpg'))

object_detection_api(image_list, rect_th=2, text_th=1, text_size=1)

В соответствии с просьбой, я также вставляю ниже код для object_detection_api: 

def object_detection_api(img_path, threshold=0.7, rect_th=3, text_size=3, text_th=3):
  """
  object_detection_api
    parameters:
      - img_path - path of the input image
      - threshold - threshold value for prediction score
      - rect_th - thickness of bounding box
      - text_size - size of the class label text
      - text_th - thichness of the text
    method:
      - prediction is obtained from get_prediction method
      - for each prediction, bounding box is drawn and text is written
        with opencv
      - the final image is displayed
  """
  boxes, pred_cls = get_prediction(img_path, threshold)
  img = cv2.imread(img_path)
  img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
  for i in range(len(boxes)):
    cv2.rectangle(img, boxes[i][0], boxes[i][1],color=(0, 255, 0), thickness=rect_th)
    cv2.putText(img,pred_cls[i], boxes[i][0], cv2.FONT_HERSHEY_SIMPLEX, text_size, (0,255,0),thickness=text_th)
  plt.figure(figsize=(20,30))
  plt.imshow(img)
  plt.xticks([])
  plt.yticks([])
  plt.show()

1 Ответ

3 голосов
/ 24 января 2020

Что-то вроде 

import os
ls = [x for x in os.listdir('/Users/name/images/') if x.endswith('.jpg')]
im_list = ['/Users/name/images/'+x for x in ls]
for img_path in im_list:
   object_detection_api(img_path)

Может работать на вас.

Имейте в виду, что map() и filter создают генераторы, и если вам нужно с нетерпением их оценить, вы можете позвоните им list(), чтобы поместить их в список.

Благодарим @ Gwang-JinKim за помощь!

...