|
圖片解壓縮在目錄下的dataset
安裝套件
- conda install keras
- pip install imutils
- pip install sklearn
- conda install tensorflow
複製代碼
程式碼:
- from imutils import paths
- import numpy as np
- import cv2,os
- print('資料夾影像的載入')
- imagePaths = list(paths.list_images('./dataset'))
- print(imagePaths)
- data = []
- labels = []
- for imagePath in imagePaths:
- print(imagePath)
- label = imagePath.split(os.path.sep)[-2]
- print(os.path.sep)
- print(label) # 分類時的目標,代表這是covid或normal
- image = cv2.imread(imagePath)
- image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
- image = cv2.resize(image, (224,224)) # 轉換為相同的大小
- # cv2.imshow('image',image)
- # cv2.waitKey(0)
- data.append(image)
- labels.append(label)
- data = np.array(data)/255.0
- labels = np.array(labels)
- print(labels)
複製代碼 結果圖
|
|