v0.1.0

Author: hzkitty

README.md

@@ -50,9 +50,9 @@ KittyDoc 是一个轻量级、专注于文档解析的开源框架，支持 **OC
 
 ## 🛠️ 安装KittyDoc
 
-#### 使用pip安装KittyDoc （暂未发布）
+#### 使用pip安装KittyDoc
 ```bash
-pip install kitty_doc -i https://mirrors.aliyun.com/pypi/simple
+pip install kitty-doc -i https://mirrors.aliyun.com/pypi/simple
 ```
 
 #### 通过源码安装KittyDoc
@@ -101,9 +101,10 @@ os.environ['MINERU_DEVICE_MODE'] = "cuda:1"
 
 ## 📌 TODO
 
-- [x] 复选框识别，包括勾选/未勾选（默认关闭、存在误检）
 - [x] 表格非OCR文本提取
 - [x] 跨页表格合并
+- [x] 复选框识别，使用opencv（默认关闭、opencv识别存在误检）
+- [ ] 复选框识别，使用模型
 - [ ] 四方向分类旋转表格解析 rapid_orientation
 - [ ] 表格内公式提取
 - [ ] 表格内图片提取

demo.py

@@ -102,7 +102,7 @@ def do_parse(
         }
 
         checkbox_config = {
-            "checkbox_enable": True, # 是否识别复选框，默认不识别，基于opencv，有可能会误检
+            # "checkbox_enable": False, # 是否识别复选框，默认不识别，基于opencv，有可能会误检
         }
 
 

demo/demo.py

@@ -97,7 +97,7 @@ def do_parse(
             # "wireless_table.model_dir_or_path": "", # 无线表结构模型地址，配置SLANEXT时使用,
         }
         checkbox_config = {
-            "checkbox_enable": True, # 是否识别复选框，默认不识别，基于opencv，有可能会误检
+            # "checkbox_enable": False, # 是否识别复选框，默认不识别，基于opencv，有可能会误检
         }
         for idx, pdf_bytes in enumerate(pdf_bytes_list):
             new_pdf_bytes = convert_pdf_bytes_to_bytes_by_pypdfium2(pdf_bytes, start_page_id, end_page_id)

setup.py

@@ -31,7 +31,8 @@ def parse_requirements(filename):
         author_email='',  # 作者邮箱
         packages=find_packages() + ["kitty_doc.resources"],  # 包含所有的包
         package_data={
-            "kitty_doc.resources": ["**"],  # 包含magic_pdf.resources目录下的所有文件
+            "kitty_doc.resources": ["**"],  # 包含 kitty_doc/resources 目录下的所有文件
+            "": ["*.yaml"],  # 包含所有包里的 .yaml 文件
         },
         install_requires=parse_requirements('requirements.txt'),  # 项目依赖的第三方库
         # extras_require={