README.md 2.03 KB
Newer Older
zhougaofeng's avatar
zhougaofeng committed
1
2
# magic_pdf

zhougaofeng's avatar
zhougaofeng committed
3
4
## 安装

zhougaofeng's avatar
zhougaofeng committed
5
### 以下演示在223节点安装pdf解析模块(可以直接使用镜像:1177ea7959ce)
zhougaofeng's avatar
zhougaofeng committed
6

zhougaofeng's avatar
zhougaofeng committed
7
下载本项目
zhougaofeng's avatar
zhougaofeng committed
8

zhougaofeng's avatar
zhougaofeng committed
9
`git clone http://developer.sourcefind.cn/codes/zhiAn123/magic_pdf.git`
zhougaofeng's avatar
zhougaofeng committed
10

zhougaofeng's avatar
zhougaofeng committed
11
下载需要的模型库
zhougaofeng's avatar
zhougaofeng committed
12
13

`git lfs clone https://www.modelscope.cn/opendatalab/PDF-Extract-Kit.git`
zhougaofeng's avatar
zhougaofeng committed
14

zhougaofeng's avatar
zhougaofeng committed
15
16
17
18
19
20
21
22
使用魔搭下载
`
    pip install modelscope
    from modelscope import snapshot_download
    model_dir = snapshot_download('opendatalab/PDF-Extract-Kit')
`


zhougaofeng's avatar
zhougaofeng committed
23
### 1、安装需要的依赖库
zhougaofeng's avatar
zhougaofeng committed
24

zhougaofeng's avatar
zhougaofeng committed
25
26
`cd magic_pdf-main`

zhougaofeng's avatar
zhougaofeng committed
27
28
#### pip install -e .

zhougaofeng's avatar
zhougaofeng committed
29
### 2、安装需要的模型
zhougaofeng's avatar
zhougaofeng committed
30

zhougaofeng's avatar
zhougaofeng committed
31
#### 修改magic-pdf.template.json
zhougaofeng's avatar
zhougaofeng committed
32

zhougaofeng's avatar
zhougaofeng committed
33
34
35
36
<div align=center>
    <img src="doc/image (9).png"/>
</div>
需要注意,"models-dir":"/home/practice/model/PDF-Extract-Kit/models" 路径指向PDF-Extract-Kit/models
zhougaofeng's avatar
zhougaofeng committed
37

zhougaofeng's avatar
zhougaofeng committed
38
39
40
41
将magic-pdf.template.json 拷贝到/root目录下并改名为magic-pdf.json
<div align=center>
    <img src="doc/image (10).png"/>
</div>
zhougaofeng's avatar
zhougaofeng committed
42

zhougaofeng's avatar
zhougaofeng committed
43
### 4、启动qwen-ocr模块:
zhougaofeng's avatar
zhougaofeng committed
44
下载qwen模型:[快速下载通道](http://113.200.138.88:18080/aimodels/qwen/Qwen2-VL-7B-Instruct.git)
zhougaofeng's avatar
zhougaofeng committed
45

zhougaofeng's avatar
zhougaofeng committed
46
47
48
49
50
修改magic_pdf-main/magic_pdf/dict2md/ocr_server.py文件中模型路径地址

<div align=center>
    <img src="doc/image11.png"/>
</div>
zhougaofeng's avatar
zhougaofeng committed
51
52
53

#### qwen-ocr服务启动代码:

zhougaofeng's avatar
zhougaofeng committed
54
`python magic_pdf/dict2md/ocr_server.py`
zhougaofeng's avatar
zhougaofeng committed
55

zhougaofeng's avatar
zhougaofeng committed
56
默认使用6020端口,0号DCU卡 ,可以通过--dcu_id 指定卡,--server_port指定端口号,-c 指定qwen模型地址
zhougaofeng's avatar
zhougaofeng committed
57

zhougaofeng's avatar
zhougaofeng committed
58
59
60
61
62
63
qwen-ocr模块启动成功:
<div align=center>
    <img src="doc/image (5).png"/>
</div>

### 5、启动pdf-server解析服务:
zhougaofeng's avatar
zhougaofeng committed
64
65
66

#### pdf-server解析服务启动代码:

zhougaofeng's avatar
zhougaofeng committed
67
`python magic_pdf/tools/pdf_server.py`
zhougaofeng's avatar
zhougaofeng committed
68

zhougaofeng's avatar
zhougaofeng committed
69
70
默认使用6030端口,0号DCU卡 ,可以通过--dcu_id 指定卡,--pdf_port指定端口号

zhougaofeng's avatar
zhougaofeng committed
71
72
73
<div align=center>
    <img src="doc/image (6).png"/>
</div>
zhougaofeng's avatar
zhougaofeng committed
74

zhougaofeng's avatar
zhougaofeng committed
75
76
77
78
启动成功:
<div align=center>
    <img src="doc/image (7).png"/>
</div>
zhougaofeng's avatar
zhougaofeng committed
79

zhougaofeng's avatar
zhougaofeng committed
80
### 6、解析pdf
zhougaofeng's avatar
zhougaofeng committed
81
`python magic_pdf/parse/common_parse.py -p [文件/目录 路径] -o [输出地址]`
zhougaofeng's avatar
zhougaofeng committed
82
83
84
85
<div align=center>
    <img src="doc/image (8).png"/>
</div>
-p指定pdf路径,-o指定输出路径