羊毛毛毛毛毛 2023-02-21 13:33 采纳率: 20%
浏览 12
已结题

想用肘关节法看图片最优聚类 输入维度有问题怎么解决?


img=cv2.imread('/content/drive/MyDrive/no-fog vs fog/8_8_1.jpg')
Z = img.reshape((-1, 3))
Z = np.array(Z).reshape(1, -1)
whw_df = np.float32(Z)
whw_df = [[i] for i in whw_df]
##KMeans进行选取聚类数目
#分别利用手肘法和轮廓系数对聚类的数目进行探索:
#手肘法
# 利用SSE选择k
whw_SSE = []  # 存放每次结果的误差平方和
for i in range(2, 8):  # 尝试要聚成的类数
    whw_estimator = KMeans(n_clusters=i)  # 构造聚类器
    whw_estimator.fit(whw_df) #存入数组(数据的列名标签)
    whw_SSE.append(whw_estimator.inertia_)
X = range(2, 8)  # 跟k值要一样
fig = plt.figure(figsize=(8,6))
plt.xlabel('聚类数目')
plt.ylabel('SSE')
plt.plot(X, whw_SSE,'o-' )
plt.title("K值在肘关节处最优")
plt.show()  # 画出图
  • 写回答

2条回答 默认 最新

  • MarkHan_ 2023-02-21 13:42
    关注

    该回答引用GPTᴼᴾᴱᴺᴬᴵ

    从您提供的代码看,图片已经被读入,reshape 成为 (n_samples, n_features) 的二维数组 Z,再转化为一维数组 whw_df。但是在这里,一维数组 whw_df 中的每个元素又被包装成一个列表,这一步看上去没有必要。因此,可以直接把 whw_df 的元素改成三个维度,即 whw_df = whw_df.reshape(-1, 1, 3)。

    此外,聚类算法在图像分割中通常使用 RGB 颜色空间,而不是将每个像素的 R、G、B 三个分量分开考虑。因此,将 Z 按行展平为一维数组后,应该将其转化为一个二维数组,即 Z = img.reshape((-1, 3))。

    修改后的代码如下:

    img = cv2.imread('/content/drive/MyDrive/no-fog vs fog/8_8_1.jpg')
    Z = img.reshape((-1, 3))
    whw_df = np.float32(Z).reshape(-1, 1, 3)
    
    # 利用SSE选择k
    whw_SSE = []  # 存放每次结果的误差平方和
    for i in range(2, 8):  # 尝试要聚成的类数
        whw_estimator = KMeans(n_clusters=i)  # 构造聚类器
        whw_estimator.fit(whw_df)  # 存入数组(数据的列名标签)
        whw_SSE.append(whw_estimator.inertia_)
    
    X = range(2, 8)  # 跟k值要一样
    fig = plt.figure(figsize=(8,6))
    plt.xlabel('聚类数目')
    plt.ylabel('SSE')
    plt.plot(X, whw_SSE,'o-')
    plt.title("K值在肘关节处最优")
    plt.show()  # 画出图
    

    希望这能解决您的问题。

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(1条)

报告相同问题?

问题事件

  • 系统已结题 3月1日
  • 已采纳回答 2月21日
  • 创建了问题 2月21日

悬赏问题

  • ¥15 求MCSCANX 帮助
  • ¥15 机器学习训练相关模型
  • ¥15 Todesk 远程写代码 anaconda jupyter python3
  • ¥15 我的R语言提示去除连锁不平衡时clump_data报错,图片以下所示,卡了好几天了,苦恼不知道如何解决,有人帮我看看怎么解决吗?
  • ¥15 在获取boss直聘的聊天的时候只能获取到前40条聊天数据
  • ¥20 关于URL获取的参数,无法执行二选一查询
  • ¥15 液位控制,当液位超过高限时常开触点59闭合,直到液位低于低限时,断开
  • ¥15 marlin编译错误,如何解决?
  • ¥15 有偿四位数,节约算法和扫描算法
  • ¥15 VUE项目怎么运行,系统打不开