69 lines
1.4 KiB
Markdown
69 lines
1.4 KiB
Markdown
# Vidocp
|
|
|
|
This repository implements computer vision based approaches for detecting and parsing visual features such as tables or
|
|
previous redactions in documents.
|
|
|
|
## Installation
|
|
|
|
```bash
|
|
git clone ssh://git@git.iqser.com:2222/rr/vidocp.git
|
|
cd vidocp
|
|
|
|
python -m venv env
|
|
source env/bin/activate
|
|
|
|
pip install -e .
|
|
pip install -r requirements.txt
|
|
|
|
dvc pull
|
|
```
|
|
|
|
## Usage
|
|
|
|
### As an API
|
|
|
|
The module provided functions for the individual tasks that all return some kid of collection of points, depending on
|
|
the specific task. Example for finding the outlines of previous redactions.
|
|
|
|
```python
|
|
|
|
from vidocp.redaction_detection import find_redactions
|
|
import pdf2image
|
|
import numpy as np
|
|
|
|
pdf_path = ...
|
|
page_index = ...
|
|
|
|
|
|
page = pdf2image.convert_from_path(pdf_path, first_page=page_index, last_page=page_index)[0]
|
|
page = np.array(page)
|
|
|
|
redaction_contours = find_redactions(page)
|
|
```
|
|
|
|
|
|
### As a CLI Tool
|
|
|
|
|
|
Core API functionalities can be used through a CLI.
|
|
|
|
|
|
#### Table Parsing
|
|
|
|
The tables parsing utility detects and segments tables into individual cells.
|
|
```bash
|
|
python scripts/annotate.py data/test_pdf.pdf 2 --type redaction
|
|
```
|
|
|
|
|
|
#### Redaction Detection
|
|
|
|
The redaction detection utility detects previous redactions in PDFs (black filled rectangles).
|
|
```bash
|
|
python scripts/annotate.py <path to pdf> 0 --type redaction
|
|
```
|
|
|
|
The below image shows the detected redactions with green outlines.
|
|
|
|
|