TableMASTER-mmocr运行环境搭建
pwd:TableMASTER-mmocr-master
# conda环境
conda create -n record python=3.8
1.pytorch
# pytorch 1.10 CUDA 10.2
pip install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio==0.10.1 -f https://download.pytorch.org/whl/cu102/torch_stable.html
2.按照REAMD.md进行installation
注意:requirements的两个lanms-proper
要换成lanms-neo
Install mmdetection. click here for details.
# We embed mmdetection-2.11.0 source code into this project.
# You can cd and install it (recommend).
cd ./mmdetection-2.11.0
pip install -v -e .
Install mmocr. click here for details.
# install mmocr
cd ..
pip install -v -e .
3.mmcv-full(1.2.4-1.4.0)
可以去官方文档找到对应版本,这里直接给了可行的对应 pytorch 1.10 CUDA 10.2 的版本
pip install mmcv-full==1.4.0 -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.10/index.html
4.运行demo
真正的demo在 ./table_recognition/demo
,有README.md
可以看
直接运行
python ./table_recognition/demo/demo.py
此时报错没有tqdm,直接pip安装了就好,然后再运行就没问题了。
但是里面路径会报错,直接修改模型路径过去。
标签:mmocr,demo,cu102,TableMASTER,install,pip,搭建 From: https://www.cnblogs.com/faf4r/p/17023809.html