文件名称：shape-classifier-cnn:使用CNN的形状分类器
文件大小：23.33MB
文件格式：ZIP
更新时间：2024-05-21 14:00:23
machine-learning computer-vision deep-learning keras cnn
形状分类器（深度学习）：是使用CNN /深度学习将图像成像为三角形，正方形还是圆形。 形状有很多不同的类型，能够区分它们很重要。 在该项目中，我们创建了深度卷积神经网络，以对不同的绘图图像进行分类并将其标记为三角形，正方形或圆形。 首先，我们使用具有3个主要类别的训练数据集来训练模型。 之后，我们将训练好的模型保存到h5文件中。 我们将使用此预训练模型从测试数据集中预测我们的自定义图像（单个）或图像集（多个图像）。 我们正在使用matplotlib库在每个时期之后确保准确性和有效性的丧失。 还有另一个数据集可以使用四种分类形状进行练习。 安装 要开始项目，只需遵循以下几个步骤 $ git clone https://github.com/keyurr2/shape-classifier-cnn.git $ pip install -r requirements.txt 这将安