toni99c's picture
Update README.md
23c5705 verified
metadata
license: unknown

The purpose of this model is to classify a single document page image to define if it is the beginning page of the document or the middle/end page of a document. Single-page documents are classified as beginning page. It is a first step of the more general document boundary classification problem.

To generate the embeddings use google/siglip2-so400m-patch16-512 with no fine tuning. You have a tiny script in generate_embeddings.py to generate a pickle file with the embeddings, provided a Pandas DataFrame tasks_df with a col "image_path" that contains all the images paths.

Then you can use the resulting embeddings with the model here uploaded.

Output meaning:

0 -> Middle or end page of a document

1 -> Beginning page of a document (or single-page document)