dog-vs-cat-classification:狗与猫的图像分类

时间:2024-06-07 06:46:04
【文件属性】:

文件名称:dog-vs-cat-classification:狗与猫的图像分类

文件大小:3KB

文件格式:ZIP

更新时间:2024-06-07 06:46:04

Python

狗与猫的图像分类 这是一个简单的小型项目,借助称为卷积神经网络(CNN)的机器学习算法预测狗或猫的图像。 这个想法来自 模型实施 这是一个简单的模型,最后使用2个转换层,2个Maxpool层和2个完全连接的层。 层顺序 [Input (64, 64, 3)] -> [Conv (128)] -> [Pool] ->[Conv (128)] -> [Pool] -> [Full (128)] -> [Full (1)] -> [Output] 卷积层滤镜大小= 3 X 3 最大池使用2 X 2的大小 学习率= 0.001 压差= 0.2可防止过度拟合 no_of_epochs = 50 train_test_split = 80%-20% ->整个程序执行所需的总时间为25.48小时。 笔记本电脑的配置是: Microsoft Windows 10 GPU-Nvidi


【文件预览】:
dog-vs-cat-classification-master
----_config.yml(25B)
----LICENSE(1KB)
----README.md(1KB)
----CNN.py(2KB)

网友评论