上传者: 38507121
|
上传时间: 2021-12-30 16:45:14
|
文件大小: 301KB
|
文件类型: -
一.数据集准备
数据集共1400张机场或湖泊的图片,因此此分类为简单的二分类问题,通过CNN对数据集进行模型训练,得出相关指标。
数据集如下:
机场
湖泊
二.读取数据集
数据集路径
导入相关模块
import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt
import pathlib #使用pathlib对路径对象进行管理
import random
构造路径对象,获取所有图片路径,并打乱数据集
pic_dir = 'D:/tensorflowDataSet/2_class'
pic_ro