Rise-group / DETR_for_commerce

This repository contains the data and the code used in the implementation of the DETR for the detection of businesses visible in street images.

DEtection Transformer for Commerce

Description

This repository contains the data and the code used in the implementation of the DETR for the detection of businesses visible in street images:

Juan C. Saldarriaga¹

¹ RiSE-group, Department of Mathematical Sciences, Universidad EAFIT, Medellin, Colombia

maintainer = "RiSE Group" (http://www.rise-group.org/). Universidad EAFIT

Corresponding author = [e-mail] (JCD)

# PREDICCIÓN
python prediction.py --fg_folder=best_model 
                     --score_th=0.1
                     --folder_images=data/tuning/
                     --output_directory=predictions/prediction_tuning_best_model.csv
                     
# RENDIMIENTO
python model_performance.py --prediction_folder=predictions/prediction_tuning_best_model
                            --labels_file = data/tuning_labels.csv
                            --score_th=0.1
                            --IOU_th = 0.45