如何利用K-Means将文件夹中图像进行分类？

点击上方“小白学视觉”，选择加"星标"或“置顶”

重磅干货，第一时间送达

K-Means聚类是最常用的无监督机器学习算法之一。顾名思义，它可用于创建数据集群，从本质上将它们隔离。

现在，我们将做一个简单的示例，将文件夹中的图像进行分离，该文件夹既有猫也有狗的图像。并且将创建两个单独的文件夹（群集），我们将介绍如何自动确定K的最佳值。

猫和狗的图像数据集

首先，我们将从导入所需的库开始。

import numpy as np
import tensorflow as tf
import matplotlib.pyplot as plt
from sklearn.cluster import KMeans
from sklearn.metrics import