Advertisement

深层神经网络模型用于小猫分类

阅读量:

1 问题背景

从零构建一个具备五层或更多层级的二分类模型,用于判断图像中是否存在小猫,无需依赖任何现成工具包。输入数据为64*64尺寸的RGB图像,当然也可根据需求调整输入尺寸,具体可通过两种方式实现:其一是对图像本身进行缩放处理,其二是修改输入层的参数配置。输出结果目前仅包含0或1两个类别,但通过调整输出层的规模,亦可拓展至更复杂的多类别分类任务。

实验环境配置与搭建

复制代码
 import numpy as np    #基本操作

    
 import h5py    #数据集
    
 import matplotlib.pyplot as plt    #画图
    
 from PIL import Image        #图像处理
    
 from dnn_app_utils_v2 import * #
    
  
    
 %matplotlib inline #画图的设置
    
 plt.rcParams['figure.figsize'] = (5.0, 4.0) # set default size of plots
    
 plt.rcParams['image.interpolation'] = 'nearest'
    
 plt.rcParams['image.cmap'] = 'gray'

全部评论 (0)

还没有任何评论哟~