如何從 python 中的目錄中獲取圖像文件名

Question

我想在我的目錄中獲取文件名，以便在 ml model 中動態使用它。

import cv2
import numpy as np
import os

i=os.listdir('./upload')[0]
print(i)

img=cv2.imread(i)
im=cv2.resize(img,(960,540))

Answer 1

您的方法的問題是os.listdir返回文件名而不是文件路徑。 因此，您需要將目錄名稱添加到文件路徑中：

cv2.imread(os.path.join('upload', i))

或者您可以使用返回路徑的glob ：

import glob
import cv2

for image_path in glob.glob('./upload/*.jpg'):
    image = cv2.imread(image_path)