本文轉載自【微信公眾號:機器學習算法與Python精研 ,ID:AITop100】,經微信公眾號授權轉載,如需轉載原文作者聯繫
來自:GitHub
測試環境
Ubuntu 18.04python 3.6.7numpy 1.16.4tensorflow-gpu 1.13.1 或者是cpu版本keras 2.2.4opencv-python 4.1.0.25PyQt5 5.12.2CUDA 10.0.130cuDNN 7.5
環境搭建
pip3 install numpy==1.16.4pip3 install tensorflow-gpu==1.13.1 #或者使用cpu版本pip3 install keras==2.2.4pip3 install opencv-python==4.1.0.25pip3 install PyQt5==5.12.2
CUDA與cuDNN安裝
具體安裝過程可以在NVIDIA官網查看
啟動demo
執行demo.py打開銀行卡號識別的GUI界面。batch_test.py用於對銀行卡進行批量識別,將要識別的銀行卡放在同目錄下的test_images文件中。定位結果以及識別結果存放在test_results中。
模型訓練
在train文件中首先執行datasets.py進行數據增強,並生成對應圖片路徑及標籤文件,再執行run.py文件開始模型的訓練,訓練大概20個epoch後loss不再收斂可以停止。訓練的模型被保存在train/model中。
測試模型
選取銀行卡圖片進行測試時,銀行卡儘量充滿圖片,橫向放置。