import cv2 import numpy as np from sklearn.metrics import classification_report from sklearn.model_selection import train_test_split from torchvision.datasets import ImageFolder dataset = ImageFolder('C:\\Users\\LENOVO\\Desktop\\Images\\') a = cv2.imread('C:/Users/LENOVO/Desktop/Images/0/agricultural00.tif') dataset = dataset.imgs print(dataset) data_path = [] data_label = [] for i in range(dataset.__len__()): data_path.append(dataset[i][0]) data_label.append(dataset[i][1]) data_arr=[] for i in range(len(data_path)): p = cv2.imread(data_path[i]) p = cv2.resize(p,(224,224)) data_arr.append(np.array(p).flatten()) print(type(data_arr)) data_arr = np.array(data_arr) data_label = np.array(data_label) X_train, X_test, y_train, y_test = train_test_split(data_arr, data_label, test_size=0.1, random_state=200) from sklearn import svm predictor = svm.SVC(gamma='scale', C=1.0,max_iter = 1000) predictor.fit(X_train, y_train) predictions_labels = predictor.predict(X_test) print(classification_report(y_test, predictions_labels,digits=3))
SVM图片分类经常会遇到的问题
最新推荐文章于 2024-06-30 17:10:06 发布