Skip to content

Latest commit

 

History

History

Software

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
Data
Data
 
 
output
output
 
 
.DS_Store
.DS_Store
 
 
README.txt
README.txt
 
 
data_misc.txt
data_misc.txt
 
 
data_pp.txt
data_pp.txt
 
 
data_tos.txt
data_tos.txt
 
 
requirements.txt
requirements.txt
 
 
train.py
train.py
 
 

README.txt

1. DESCRIPTION
This is the implementation of STAPI pipeline. It is recommended to run STAPI under Python3.7 and a virtual environment.



2. INSTALLATION
To install required packages, run the following command in terminal.

pip install -r requirements.txt


To use our model, replace the Data folder with the downloaded dataset (../Data folder).



3. EXECUTION
To train STAPI from scratch, run the following command. The trained models (both filter and typographic classifier) will be saved in output folder. Confusion matrices and ROC curves will be generated.

python train.py