python水质二维模型_基于深度学习的水质图像分类(python源码)

运行环境import matplotlib.pyplot as plt

import os

import tensorflow as tf

import tensorflow.contrib.eager as tfe

实验内容

某地区的多个罗非鱼池塘水样数据，包含水产专家按水色判断水质分类的数据，以及用数码相机按照标准进行水色采集的数据，如表1和图1所示。每个水质图片命名规则是“类别-编号.jpg”，如“1_1.jpg”是第一类样本的图片。请根据这些样本，利用数字图像处理技术，通过水色图像实现水质的自动评价。

实验过程

观察样本图片，发现其几乎每张图片都有背景噪声，所以需要先对图像进行预处理，如下代码：

输入fromdir为图片文件夹，todir为输出文件夹。

对处理过的图片进行HSV颜色矩提取并打上标签：

接下来编写神经网络相关代码：

分析csv数据：

读取颜色矩：

定义网络变量：class NNetConfig():

num_classes = 5 # 多分类的种类