zl程序教程

您现在的位置是:首页 >  硬件

当前栏目

机器学习笔记 - 使用自己收集的图片以及前馈神经网络,进行图像分类训练

2023-09-14 09:01:36 时间

一、数据收集

首先从ImageNet(编写程序下载)和百度(手动找了一些)搜索了一些图片,分三类鱼(700张左右)、浣熊(1100张左右)、猫(1000张左右,可能有几张老虎)。这个工作可能比较无聊,但是尝试并了解如何使用自己的数据训练,还是挺有意思的。

数据集下载地址:自定义采集的图像数据集,浣熊、鱼、猫,用于训练神经网络-深度学习文档类资源-CSDN下载

二、 参考代码如下

前馈神经网络是一种最简单的神经网络,各神经元分层排列。每个神经元只与前一层的神经元相连。接收前一层的输出,并输出给下一层.各层间没有反馈。

数据集很,运行起来很快,我的i7cpu,1s / epoch 

# 标准前馈神经网络,进行图像分类,效果不好
# set the matplotlib backe