diff --git a/docs/reference/object-detection-2d-nms-seq2seq_nms.md b/docs/reference/object-detection-2d-nms-seq2seq_nms.md index c1269c108f..8f530743bb 100644 --- a/docs/reference/object-detection-2d-nms-seq2seq_nms.md +++ b/docs/reference/object-detection-2d-nms-seq2seq_nms.md @@ -302,4 +302,4 @@ TABLE-1: Average Precision (AP) achieved by pretrained models on the person dete #### References -[1] Neural Attention-driven Non-Maximum Suppression for Person Detection, [TechRxiv](https://www.techrxiv.org/articles/preprint/Neural_Attention-driven_Non-Maximum_Suppression_for_Person_Detection/16940275). +[1] C. Symeonidis, I. Mademlis, I. Pitas and N. Nikolaidis, "Neural Attention-Driven Non-Maximum Suppression for Person Detection" in IEEE Transactions on Image Processing, vol. 32, pp. 2454-2467, 2023, doi: 10.1109/TIP.2023.3268561. diff --git a/src/opendr/perception/object_detection_2d/nms/seq2seq_nms/README.md b/src/opendr/perception/object_detection_2d/nms/seq2seq_nms/README.md index 4e03fce80c..3da814929e 100644 --- a/src/opendr/perception/object_detection_2d/nms/seq2seq_nms/README.md +++ b/src/opendr/perception/object_detection_2d/nms/seq2seq_nms/README.md @@ -1,7 +1,9 @@ Seq2Seq-NMS ====== -This folder contains an implementation of Seq2Seq-NMS [[1]](#seq2seq_nms-1). +This folder contains an implementation of the Seq2Seq-NMS algorithm, for neural Non-Maximum Suppression in visual person detection. If one uses any part of this implementation in his/her work, he/she is kindly asked to cite the following paper: + +- C. Symeonidis, I. Mademlis, I. Pitas and N. Nikolaidis, "[Neural Attention-Driven Non-Maximum Suppression for Person Detection](https://ieeexplore.ieee.org/abstract/document/10107719)" in IEEE Transactions on Image Processing, vol. 32, pp. 2454-2467, 2023, doi: 10.1109/TIP.2023.3268561. TABLE-1: Average Precision (AP) achieved by pretrained models on the person detection task on the validation sets. The maximum number or RoIs, employed for the performance evaluation was set to 800. | **Pretrained Model** | **Dataset** | **Detector** | **Type of Appearance-based Features** | **Pre-processing IoU Threshold** | **AP@0.5 on validation set** | **AP@0.5 on test set** | @@ -14,4 +16,3 @@ TABLE-1: Average Precision (AP) achieved by pretrained models on the person dete \* The minival set was used as validation set.
\*\* The minitest set was used as test set. -[1] Neural Attention-driven Non-Maximum Suppression for Person Detection, [TechRxiv](https://www.techrxiv.org/articles/preprint/Neural_Attention-driven_Non-Maximum_Suppression_for_Person_Detection/16940275).