By Md. Sadrul Islam Toaha, Sakib Bin Asad, Chowdhury Rafeed Rahman, S. M. Shahriar Haque, Mahfuz Ara Proma, Md. Ahsan Habib Shuvo, Tashin Ahmed, Md. Amimul Basher
Automatic signboard detection in a developing or less developed city is a difficult task due to its challenging urban features. To approach this problem, we introduce a robust model with a new signboard dataset. The signboard detection model is based on Faster R-CNN architecture including a smart proposal box generator algorithm and specialized pretrained techniques.
This repository contains code for signboard detection which will return the segmented region of a signboard with localization details automatically.
Visit Google Colab, and clone the Detection.ipynb
file from https://github.com/sadrultoaha/Signboard-Detection.git
%tensorflow_version 1.x
!pip install Keras==2.2.4
- Set the required path as follows and run all the notebook cells.
test_file_path = 'Test'
output_csv = 'output.csv'
output_zip = 'result.zip'
- Upload and connect your custom dataset into Google Colab environment.
- Set the required path as follows and run all the notebook cells.
test_file_path = 'your_test_data_path'
output_csv = 'output.csv'
output_zip = 'result.zip'
git clone https://github.com/sadrultoaha/Signboard-Detection.git
cd Signboard-Detection
make sure that you have Tensorflow==1.15.2
and Keras==2.2.4
installed. (see: Tensorflow installation instructions)
pip install -r requirements.txt
python detection.py -test_file_path "Test" -output_csv output.csv -output_zip result.zip
To run the signboard model on your desired test data, set the test_file_path to the your own custom test dataset path.
python detection.py -test_file_path "your_test_data_path" -output_csv output.csv -output_zip result.zip
Arguments Details:
- -test_file_path: Path to the testing images folder, i.e., Path to the Public test data or Path to the Custom test data.
- -output_csv: Path to output the predicted localization and classification details.
- -output_zip: Path to output the segmented signboards on input images.