Object and Edge Detection on Entity-Relationship Diagrams

This repository contains the code for the paper "Object and Edge Detection on Entity-Relationship Diagrams."

Installation

Follow these steps to set up the project:

1. Clone the repository:

   git clone <repository-url>
   cd <repository-folder>

2. Install Poetry: Follow the instructions provided in the Poetry documentation.

3. Install project dependencies:

   poetry install

4. Add the weights for YOLO and SAM:

   • Create a weights folder in the top level of the project directory:

     mkdir weights

   • Download the YOLO weights from here and place them in the weights folder.

   • Download the SAM weights from here and place them in the weights folder.

Running the Code

Before executing the code, update the file erd_detection/main.py at lines 22 and 29 with the appropriate paths.

To execute the code, use the following command in the project directory:

poetry run python erd_detection/main.py

Own handwritten Entity-Relationship diagrams can be tested by adding other images to the data folder.