準(zhǔn)備圖片和標(biāo)簽

這里我使用的是tfrecord數(shù)據(jù)文件來(lái)進(jìn)行讀取
第一步 圖片獲取
我使用的是java代碼來(lái)制作圖片（需要具體制作圖片代碼可以和我私聊）
（圖片多少自己決定 但我可以給你建議 開(kāi)始 每個(gè)類別 測(cè)試集300 訓(xùn)練集 3000 個(gè)人建議開(kāi)始做的時(shí)候 后面可以加
）

//附帶一小段主要代碼 主要代碼太長(zhǎng)了 // 保存圖片到指定的輸出流public static void output (BufferedImage image, OutputStream out) throws IOException {ImageIO.write(image, "JPEG", out);}//圖片創(chuàng)建public static void getmoreImage() throws Exception{//遍歷多少次創(chuàng)建多少?gòu)坒or(int i=0;i<20;i++){CreateImage_test img=new CreateImage_test();BufferedImage image=img.getImage();//保存到指定路徑String imagename="E:/temp/test/image/"+i+"_"+ img.getText()+".jpg";OutputStream out=new FileOutputStream(imagename);CreateImage_test.output(image,out);}System.out.println("打印完畢");}

第二步 對(duì)圖片打標(biāo)簽儲(chǔ)存

//這里處理也使用java代碼 public class Sortimage {/** 1 獲取每個(gè)圖片名字* 2 分割符分割* 3 獲取每個(gè)圖片名字 和 值 寫(xiě)* 入文本文件中 讀取一個(gè)圖片名字 換行下一個(gè)讀取 * */public static void main(String[] args) throws Exception {//圖片路徑String path="image0";Sortimage.getFiles(path);}public static void getFiles(String path) throws Exception {File file = new File(path);//創(chuàng)建字節(jié)輸出流對(duì)象FileOutputStream fos=new FileOutputStream ("fuhao_train_jpg_84000.txt");// 如果這個(gè)路徑是文件夾if (file.isDirectory()) {// 獲取路徑下的所有文件File[] files = file.listFiles();System.out.println(files.length);for (int i = 0; i < files.length; i++) {// 如果還是文件夾 遞歸獲取里面的文件 文件夾if (files[i].isDirectory()) {//System.out.println("目錄：" + files[i].getName());getFiles(files[i].getPath());} else {System.out.println("文件：" + files[i].getName());String[] strings=files[i].getName().split("_");// System.out.println(strings.length);String num=strings[1].substring(0, 1);//寫(xiě)數(shù)據(jù)String text =files[i].getName()+" "+num+"\r\n";System.err.println(files[i].getName()+" "+num);fos.write(text.getBytes());//System.out.println("前綴："+string);}}} else {System.out.println("文件：" + file.getPath());}fos.close();} }

開(kāi)始的時(shí)候想使用java處理這部分但是發(fā)現(xiàn)有點(diǎn)問(wèn)題 果斷使用python
第三步 圖片灰度化
`

from PIL import Image import os import cv2 import numpy as np input_dir = 'test/image/' #原地址 out_dir = 'test/image1/' #修改后地址 a = os.listdir(input_dir) for i in a:print(i)src = cv2.imread(input_dir+i)GrayImage = cv2.cvtColor(src,cv2.COLOR_BGR2GRAY)#二進(jìn)制閾值化處理 這里闕值闕值處理有很多方法 目測(cè)THRESH_BINARY_INV這個(gè)效果最好r, b = cv2.threshold(GrayImage, 127, 255, cv2.THRESH_BINARY_INV)#print (r)#設(shè)置卷積核 這里你自己可以調(diào) 卷積核大小 #kernel = np.ones((2,2), np.uint8)#圖像膨脹處理 這步看你怎么處理圖片 目的是對(duì)目標(biāo)膨脹處理#erosion = cv2.dilate(b, kernel)#設(shè)置卷積核#kernel = np.ones((1,2), np.uint8)#圖像腐蝕處理 同上膨脹處理相反 這步建議不要 這里只是給你們參考#erosion = cv2.erode(erosion, kernel)cv2.imwrite(out_dir+i,b,[cv2.IMWRITE_JPEG_QUALITY,100])# pyhton 簡(jiǎn)單方便

第四步
創(chuàng)建自己的文件路徑 、
看下圖 （訓(xùn)練集 測(cè)試集）

好了，前期的數(shù)據(jù)已經(jīng)準(zhǔn)備完畢 我們開(kāi)始制作tfrecord 數(shù)據(jù)文件了 下一篇 接著來(lái)

總結(jié)

以上是生活随笔為你收集整理的英文字母tfrecord文件的制作（一）的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。