generated from LufsX/Experiment-kNN
This commit is contained in:
+1
-1
@@ -74,7 +74,7 @@ def load_dataset(directory):
|
||||
labels = []
|
||||
|
||||
for i, file_name in enumerate(file_list):
|
||||
# 文件名格式:label_index.txt,例如 0_1.txt,标签为 0
|
||||
# 文件名格式 label_index.txt,例如 0_1.txt,标签为 0
|
||||
label = int(file_name.split('_')[0])
|
||||
labels.append(label)
|
||||
data_matrix[i, :] = img_to_vector(os.path.join(directory, file_name))
|
||||
|
||||
Reference in New Issue
Block a user