지정된 폴더의 모든 이미지를 읽는 Python 방법

다음은 지정된 폴더의 모든 이미지를 읽는 Python 방법입니다. 좋은 참조 값이 있으며 모든 사람에게 도움이 되기를 바랍니다. 함께 살펴보겠습니다

(1) 데이터 준비

데이터 세트 소개:

데이터 세트에는 756개의 네거티브 샘플(이미지 이름은 0.1~0.756)과 458개의 포지티브 샘플을 포함하여 1223개의 이미지가 저장되어 있습니다. (이미지 이름은 1.1~1.458), 여기서 "." 앞의 레이블은 샘플 레이블이고 "." 뒤의 레이블은 샘플 일련 번호입니다.

(2) Python을 사용하여 폴더의 모든 이미지를 읽습니다.

'''
Load the image files form the folder
input:
  imgDir: the direction of the folder
  imgName:the name of the folder
output:
  data:the data of the dataset
  label:the label of the datset
'''
def load_Img(imgDir,imgFoldName):
  imgs = os.listdir(imgDir+imgFoldName)
  imgNum = len(imgs)
  data = np.empty((imgNum,1,12,12),dtype="float32")
  label = np.empty((imgNum,),dtype="uint8")
  for i in range (imgNum):
    img = Image.open(imgDir+imgFoldName+"/"+imgs[i])
    arr = np.asarray(img,dtype="float32")
    data[i,:,:,:] = arr
    label[i] = int(imgs[i].split('.')[0])
  return data,label

여기에서 얻은 데이터와 레이블은 ndarray 데이터입니다.

data: (1223,1,12,12)

label: (1223,)

참고: Nddary 데이터타입 numpy에서 제공하는 데이터타입, 즉 N차원 배열로, 파이썬의 배열이 다차원성을 지원하지 않는 단점을 보완해줍니다

(3) 호출방식

craterDir = "./data/CraterImg/Adjust/"
foldName = "East_CraterAdjust12"
data, label = load_Img(craterDir,foldName)

관련 권장 사항:

Python이 csv 파일을 읽고 해당 파일을 목록에 넣는 방법에 대한 예

Python은 파일의 그림에 대한 레이블이 있는 txt 파일을 생성하는 방법을 구현합니다

위 내용은 지정된 폴더의 모든 이미지를 읽는 Python 방법의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!