首页 > 其他分享 >TableMASTER-mmocr运行环境搭建

TableMASTER-mmocr运行环境搭建

时间:2023-01-04 01:11:24浏览次数:57  
标签:mmocr demo cu102 TableMASTER install pip 搭建

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

相关文章