建立tf record檔案 直觀版

2021-09-11 22:09:14 字數 1331 閱讀 2861

import os 

import tensorflow as tf

from pil import image #注意image,後面會用到

import matplotlib.pyplot as plt

import numpy as np

writer= tf.python_io.tfrecordwriter("cifar10_train.tfrecords")

writer_2 = tf.python_io.tfrecordwriter("cifar10_test.tfrecords")

root = '101_objectcategories'

class_num =

path_all =

i = 0

num_test = 0

num_train = 0

for dirpath, dirnames, filenames in os.walk(root):

j = 0

for filepath in filenames:

img_path = os.path.join(dirpath, filepath)

img=image.open(img_path)

img= img.resize((224,224))

img =img.convert("rgb")

img_raw=img.tobytes()

example = tf.train.example(features=tf.train.features(feature=))

if j%10==0 :

writer_2.write(example.serializetostring())

num_test=num_test+1

else:

writer.write(example.serializetostring())

num_train = num_train+1

j=j+1

for dir_path in dirnames:

i=i+1

writer.close()

print(str(num_test)+'\n')

print(str(num_train))

with open('label.txt','w') as f:

for i in range(len(class_num)):

f.write(str(class_num[i])+'\n')

將資料夾下分類好的子資料夾中的轉換成tf_record的形式,然後再生成乙個label的txt檔案方便後面分類後對分類物體具體類別進行對應。

通訊原理初篇 建立直觀概念

通訊原理的分享,就更加少的可憐。上面是陳愛軍的分享所說,我感覺很對,只有廣泛的分享,和別人討論,也才能真正驗證自己是不是真的學會了。下面說說我學通訊原理的一些規劃 1 一定要借助matlab,人學習,容易記住的是圖表,而不是公式,文字 2 一定要實踐,動手寫 測試,折騰,3 借助好書,step by...

新手建立oracle資料庫最簡直觀教程

oracle資料庫建在遠端linux伺服器上,通過securecrt連線 1 通過su ora10g 乙個資料庫例項名 password sqlplus nolog sql connect as sysdba連線管理員使用者,以下都以管理員身份來做 2.通過步驟1,下來先要建立表空間 其格式為 格式...

建立FTP使用者(簡單版)

簡單版是說vsftpd已安裝,且引數配置完整,只需輸命令建立新使用者。下面都以建立名字為grace的使用者為例 新建使用者 adduser d ftp grace g ftp s sbin nologin grace d 指定使用者目錄 s sbin nologin 使該使用者不能登陸系統。pass...