【TensorFlow训练石头剪刀布数据集】

最新推荐文章于 2024-04-13 08:52:55 发布

wly_ok

最新推荐文章于 2024-04-13 08:52:55 发布

阅读量189

点赞数

分类专栏： android 文章标签： tensorflow 人工智能 python

本文链接：https://blog.csdn.net/wly_ok/article/details/131353453

版权

本文介绍如何利用TensorFlow的Keras接口训练一个石头剪刀布的数据集。首先解压并检查数据集，然后展示部分训练图片。接着，通过Keras构建并训练模型，最后展示训练和验证结果。

摘要由CSDN通过智能技术生成

TensorFlow训练石头剪刀布数据集

解压下载的数据集

import os
import zipfile

local_zip = 'D:/mldownload/rps.zip'
zip_ref = zipfile.ZipFile(local_zip, 'r')
zip_ref.extractall('D:/mldownload/')
zip_ref.close()

local_zip = 'D:/mldownload/rps-test-set.zip'
zip_ref = zipfile.ZipFile(local_zip, 'r')
zip_ref.extractall('D:/mldownload/')
zip_ref.close()

检测数据集的解压结果，打印相关信息

rock_dir = os.path.join('D:/mldownload/rps/rock')
paper_dir = os.path.join('D:/mldownload/rps/paper')
scissors_dir = os.path.join('D:/mldownload/rps/scissors')

print('total training rock images:', len(os.listdir(rock_dir)))
print('total training paper images:', len(os.listdir(paper_dir)))
print('total training scissors images:', len(os.listdir(scissors_dir)))

rock_files = os.listdir(rock_dir)
print(rock_files[:10])

paper_files = os.listdir(paper_dir)
print(paper_files[:10])

scissors_files = os.listdir(scissors_dir)
print(scissors_files[:10])