欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

TensorFlow2.X使用圖片制作簡單的數(shù)據(jù)集訓(xùn)練模型

 更新時間:2020年04月08日 14:18:59   作者:FlyDremever  
這篇文章主要介紹了TensorFlow2.X使用圖片制作簡單的數(shù)據(jù)集訓(xùn)練模型,本文通過截圖實例代碼相結(jié)合給大家介紹的非常詳細,對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值,需要的朋友可以參考下

Tensorflow內(nèi)置了許多數(shù)據(jù)集,但是實際自己應(yīng)用的時候還是需要使用自己的數(shù)據(jù)集,這里TensorFlow 官網(wǎng)也給介紹文檔,官方文檔。這里對整個流程做一個總結(jié)(以手勢識別的數(shù)據(jù)集為例)。

1、 收集手勢圖片

數(shù)據(jù)集下載

方法多種多樣了。我通過攝像頭自己采集了一些手勢圖片。保存成如下形式,

在這里插入圖片描述

以同樣的形式在建立一個測試集,當(dāng)然也可以不弄,在程序里處理。

2、構(gòu)建數(shù)據(jù)集

導(dǎo)入相關(guān)的包

import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import datasets, layers, optimizers, Sequential, metrics
from tensorflow.python.framework.convert_to_constants import convert_variables_to_constants_v2
import os
import pathlib
import random
import matplotlib.pyplot as plt

讀取文件

data_root = pathlib.Path('D:\code\PYTHON\gesture_recognition\Dataset')
print(data_root)
for item in data_root.iterdir():
 print(item)

運行結(jié)果

讀取圖片路徑到list中

all_image_paths = list(data_root.glob('*/*'))
all_image_paths = [str(path) for path in all_image_paths]
random.shuffle(all_image_paths)
image_count = len(all_image_paths)
print(image_count) ##統(tǒng)計共有多少圖片
for i in range(10):
 print(all_image_paths[i])

在這里插入圖片描述

label_names = sorted(item.name for item in data_root.glob('*/') if item.is_dir())
print(label_names) #其實就是文件夾的名字
label_to_index = dict((name, index) for index, name in enumerate(label_names))
print(label_to_index)
all_image_labels = [label_to_index[pathlib.Path(path).parent.name]
     for path in all_image_paths]

print("First 10 labels indices: ", all_image_labels[:10])

在這里插入圖片描述

預(yù)處理

def preprocess_image(image):
 image = tf.image.decode_jpeg(image, channels=3)
 image = tf.image.resize(image, [100, 100])
 image /= 255.0 # normalize to [0,1] range
 # image = tf.reshape(image,[100*100*3])
 return image

def load_and_preprocess_image(path,label):
 image = tf.io.read_file(path)
 return preprocess_image(image),label

構(gòu)建一個 tf.data.Dataset

ds = tf.data.Dataset.from_tensor_slices((all_image_paths, all_image_labels))
train_data = ds.map(load_and_preprocess_image).batch(16)

同樣的方式在制作一個測試集,就可以用于模型訓(xùn)練和測試了。

總結(jié)

到此這篇關(guān)于TensorFlow2.X使用圖片制作簡單的數(shù)據(jù)集訓(xùn)練模型的文章就介紹到這了,更多相關(guān)TensorFlow數(shù)據(jù)集訓(xùn)練模型內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!

相關(guān)文章

  • 使用with torch.no_grad():顯著減少測試時顯存占用

    使用with torch.no_grad():顯著減少測試時顯存占用

    這篇文章主要介紹了使用with torch.no_grad():顯著減少測試時顯存占用問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教
    2023-08-08
  • 高考考python編程是真的嗎

    高考考python編程是真的嗎

    在本篇文章里小編給大家整理的是一篇關(guān)于高考考python編程的相關(guān)信息,有興趣的朋友們閱讀下。
    2020-07-07
  • Python HTML解析模塊HTMLParser用法分析【爬蟲工具】

    Python HTML解析模塊HTMLParser用法分析【爬蟲工具】

    這篇文章主要介紹了Python HTML解析模塊HTMLParser用法,結(jié)合實例形式分析了HTMLParser模塊功能、常用函數(shù)及作為爬蟲工具相關(guān)使用技巧,需要的朋友可以參考下
    2019-04-04
  • Pycharm漢化兩種方法(pycharm改為中文版)

    Pycharm漢化兩種方法(pycharm改為中文版)

    PyCharm是一款流行的Python開發(fā)環(huán)境,提供了豐富的功能和工具,使得Python的開發(fā)和調(diào)試變得更加方便和高效,下面這篇文章主要給大家介紹了Pycharm漢化的兩種方法,所謂漢化就是將pycharm改為中文版,需要的朋友可以參考下
    2023-06-06
  • 使用批處理腳本自動生成并上傳NuGet包(操作方法)

    使用批處理腳本自動生成并上傳NuGet包(操作方法)

    這篇文章主要介紹了使用批處理腳本自動生成并上傳NuGet包的操作方法,非常不錯,具有一定的參考借鑒價值,需要的朋友可以參考下
    2019-11-11
  • Python中SQLAlchemy庫的使用方法分析

    Python中SQLAlchemy庫的使用方法分析

    SQLAlchemy 是Python SQL工具包和對象關(guān)系映射器(ORM),允許Python開發(fā)者在應(yīng)用程序中使用SQL來交互,而無需處理數(shù)據(jù)庫的具體細節(jié),本文給大家詳細分析了Python中SQLAlchemy庫的使用方法,需要的朋友可以參考下
    2024-08-08
  • win10下python3.8的PIL庫安裝過程

    win10下python3.8的PIL庫安裝過程

    這篇文章主要介紹了win10下python3.8的PIL庫安裝方法,本文給大家介紹的非常詳細,對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值,需要的朋友可以參考下
    2020-06-06
  • python主動拋出異常raise的方法實現(xiàn)

    python主動拋出異常raise的方法實現(xiàn)

    本文主要介紹了python主動拋出異常raise的方法實現(xiàn),文中通過示例代碼介紹的非常詳細,對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
    2023-12-12
  • python3文件復(fù)制、延遲文件復(fù)制任務(wù)的實現(xiàn)方法

    python3文件復(fù)制、延遲文件復(fù)制任務(wù)的實現(xiàn)方法

    這篇文章主要給大家介紹了關(guān)于python3文件復(fù)制、延遲文件復(fù)制任務(wù)的實現(xiàn)方法,文中通過示例代碼介紹的非常詳細,對大家學(xué)習(xí)或者使用python3具有一定的參考學(xué)習(xí)價值,需要的朋友們下面來一起學(xué)習(xí)學(xué)習(xí)吧
    2019-09-09
  • Python中http請求方法庫匯總

    Python中http請求方法庫匯總

    最近在使用python做接口測試,發(fā)現(xiàn)python中http請求方法有許多種,今天抽點時間把相關(guān)內(nèi)容整理,對python http請求相關(guān)知識感興趣的朋友一起學(xué)習(xí)吧
    2016-01-01

最新評論