使用 pytorch 建立的 CNN 模型,並建立 CAPTCHA_Translater()
以供調用。
透過漏洞爬取的訓練集資料,共有26萬種數字,每種數字各有10張 jpg 檔,可透過以下連結下載。
將檔案透過 cv2 讀取後,調用 CAPTCHA_Translater.translate()
,範例如下:
from cnn_translate import CAPTCHA_Translater
import config
import utils
import cv2
filename = '000998_3.jpg'
img = cv2.imread(utils.get_img_path(filename))
translater = CAPTCHA_Translater()
print(translater.translate(img))