我图片的文件路径是在D:\wjd下,想把分类结果保存在D:\jk这个路径下,
时间: 2024-02-27 18:57:23 浏览: 76
若要将分类结果保存到 `D:\jk` 文件夹下,您只需要将代码中的 `path` 变量修改为 `D:\wjd` 文件夹的路径,同时将 `cluster_i_path` 的路径修改为 `D:\jk\cluster_{i}`,修改后的代码如下:
```
import os
def cluster_images(path, n_clusters):
# 代码略
# 将聚类结果写入文件
with open(os.path.join("D:\jk", f"cluster_result_{n_clusters}.txt"), "w") as f:
for i, label in enumerate(labels):
f.write(f"Ag{i + 10000}.png: {label}\n")
# 将聚类结果保存到对应的文件夹中
for i in range(n_clusters):
cluster_i_path = os.path.join("D:\jk", f"cluster_{i}")
if not os.path.exists(cluster_i_path):
os.mkdir(cluster_i_path)
for j, label in enumerate(labels):
if label == i:
old_path = os.path.join(path, f"Ag{j + 10000}.png")
new_path = os.path.join(cluster_i_path, f"Ag{j + 10000}.png")
os.rename(old_path, new_path)
# 遍历不同的聚类数目
for n_clusters in range(2, 11):
cluster_images("D:\wjd", n_clusters)
```
其中,`os.path.join()` 函数可以自动将路径拼接为正确的格式,无需手动添加斜杠。这里将聚类结果保存到了 `D:\jk` 文件夹下,并将聚类编号文件夹命名为 `cluster_{i}`。请注意修改路径以及聚类编号文件夹的命名方式,以适应您的具体需求。
阅读全文