《相似度對(duì)比模型訓(xùn)練及在AidLux上部署應(yīng)用》#AidLux #深度學(xué)習(xí)方法 #相似度對(duì)比模型
1.基于相似度比對(duì)的真假識(shí)別算法介紹(1)出發(fā)點(diǎn):利用深度神經(jīng)網(wǎng)絡(luò)樟蠕,提取印章深度特征音半,同時(shí)學(xué)習(xí)印章之間的相似度琐驴,自己與自己相似黄鳍,自己與其它不相似。(2)常見的相似性度量的深度學(xué)習(xí)方法:Siamese網(wǎng)絡(luò)邢滑、Triplet Loss網(wǎng)絡(luò)萌业、本文方法侧蘸。
2.相似度比對(duì)模型訓(xùn)練與測(cè)試(1)訓(xùn)練步驟為:①按上述格式放置數(shù)據(jù)集,放在dataset.文件夾下偎捎;②將train.py當(dāng)中的train_own_data設(shè)置成True蠢终;③運(yùn)行train.py開始訓(xùn)練,可以觀察對(duì)應(yīng)step訓(xùn)練集和驗(yàn)證集的準(zhǔn)確率茴她。訓(xùn)練完成后可以看訓(xùn)練過程中oSs記錄和loss曲線寻拂,在log文件夾下。(2)模型轉(zhuǎn)換與測(cè)試①pth->onnx模型轉(zhuǎn)換:選取最優(yōu)的模型進(jìn)行轉(zhuǎn)換模型推理:onnx推理代碼②onnx->tflite打開網(wǎng)站:aimo.aidlux.com/#/model-con…輸入試用賬號(hào)和密碼:賬號(hào):AIMOTC001,密碼:AIMOTC00模型轉(zhuǎn)換模型推理③onnx->dlc進(jìn)行模型轉(zhuǎn)換
3.AidLux平臺(tái)部署進(jìn)行tflite部署丈牢、dlc部署祭钉。
4.實(shí)戰(zhàn)過程訓(xùn)練自己的相似度比對(duì)模型,并完成模型轉(zhuǎn)換到onnx己沛,然后使用AIMO轉(zhuǎn)換工具在線轉(zhuǎn)換為tflite模型和dlc模型慌核,針對(duì)test集中的2對(duì)圖片 (兩個(gè)同一樣式真章和一個(gè)真章和對(duì)應(yīng)的假章)進(jìn)行測(cè)試。
具體路程及結(jié)果見:https://www.bilibili.com/video/BV1Wu4y1E76a/