準備圖片和標簽

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

//附帶一小段主要代碼 主要代碼太長了 // 保存圖片到指定的輸出流public static void output (BufferedImage image, OutputStream out) throws IOException {ImageIO.write(image, "JPEG", out);}//圖片創建public static void getmoreImage() throws Exception{//遍歷多少次創建多少張for(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("打印完畢");}

第二步 對圖片打標簽儲存

//這里處理也使用java代碼 public class Sortimage {/** 1 獲取每個圖片名字* 2 分割符分割* 3 獲取每個圖片名字 和 值 寫* 入文本文件中 讀取一個圖片名字 換行下一個讀取 * */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);//創建字節輸出流對象FileOutputStream fos=new FileOutputStream ("fuhao_train_jpg_84000.txt");// 如果這個路徑是文件夾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);//寫數據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();} }

開始的時候想使用java處理這部分但是發現有點問題 果斷使用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)#二進制閾值化處理 這里闕值闕值處理有很多方法 目測THRESH_BINARY_INV這個效果最好r, b = cv2.threshold(GrayImage, 127, 255, cv2.THRESH_BINARY_INV)#print (r)#設置卷積核 這里你自己可以調 卷積核大小 #kernel = np.ones((2,2), np.uint8)#圖像膨脹處理 這步看你怎么處理圖片 目的是對目標膨脹處理#erosion = cv2.dilate(b, kernel)#設置卷積核#kernel = np.ones((1,2), np.uint8)#圖像腐蝕處理 同上膨脹處理相反 這步建議不要 這里只是給你們參考#erosion = cv2.erode(erosion, kernel)cv2.imwrite(out_dir+i,b,[cv2.IMWRITE_JPEG_QUALITY,100])# pyhton 簡單方便

第四步
創建自己的文件路徑 、
看下圖 （訓練集 測試集）

好了，前期的數據已經準備完畢 我們開始制作tfrecord 數據文件了 下一篇 接著來

總結

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

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。