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文件
在这里插入图片描述

之后会上传该算法相关的所有文件!

评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值