Aster文字矯正算法訓練後續(docker容器內)

上一篇相關文章中實現了aster的demo腳本,繼而我們實現aster算法的訓練模型:
編寫dockerfile:
在這裏插入圖片描述修改train.py:
在這裏插入圖片描述
運行builders目錄下的input_reader_builder.py, 用該腳本獲得訓練數據:
在data目錄下生成
在這裏插入圖片描述
啓動鏡像:

docker run -i -t aster-test:v2 /bin/bash

運行train.py文件:

docker run --gpus all -v/data/ai_lls/qiao_q/aster-code/aster-master:/aster-code/aster-master/  -w/aster-code/aster-master/ asiter-train:v1 python train.py

之後在experiments/demo/log目錄下找到相關checkpoints文件
在這裏插入圖片描述

之後會上傳該算法相關的所有文件!

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章