From c6a3633336bd1bfbb8bcadc7e753b1400de1899d Mon Sep 17 00:00:00 2001 From: majorli Date: Wed, 24 Jul 2024 17:22:52 +0800 Subject: [PATCH 1/7] update rtmdet igie results Signed-off-by: majorli --- models/cv/detection/rtmdet/igie/README.md | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/models/cv/detection/rtmdet/igie/README.md b/models/cv/detection/rtmdet/igie/README.md index 737d203c..25b1f9b3 100644 --- a/models/cv/detection/rtmdet/igie/README.md +++ b/models/cv/detection/rtmdet/igie/README.md @@ -25,12 +25,17 @@ pip3 install mmengine==0.10.4 ``` ### Download -Pretrained model: https://download.openmmlab.com/mmpose/v1/projects/rtmpose/rtmdet_nano_8xb32-100e_coco-obj365-person-05d8511e.pth + +Pretrained model: Dataset: to download the validation dataset. +```bash +wget https://download.openmmlab.com/mmpose/v1/projects/rtmpose/rtmdet_nano_8xb32-100e_coco-obj365-person-05d8511e.pth +``` ### Model Conversion + ```bash # export onnx model python3 export.py --weight rtmdet_nano_8xb32-100e_coco-obj365-person-05d8511e.pth --cfg rtmdet_nano_320-8xb32_coco-person.py --output rtmdet.onnx @@ -56,6 +61,6 @@ bash scripts/infer_rtmdet_fp16_performance.sh ## Results -| Model | BatchSize | Input Shape | Precision | FPS | mAP@0.5(%) | -| :-------: | :-------: | :---------: | :-------: | :-------: | :--------: | -| RTMDet | 32 | 320x320 | FP16 | 2627.15 | 0.619 | \ No newline at end of file +| Model | BatchSize | Input Shape | Precision | FPS | mAP@0.5(%) | +| :----: | :-------: | :---------: | :-------: | :------: | :--------: | +| RTMDet | 32 | 320x320 | FP16 | 4006.849 | 0.619 | -- Gitee From 40742e337be94d1f919d138d10491b8a2bb2d141 Mon Sep 17 00:00:00 2001 From: majorli Date: Wed, 24 Jul 2024 17:25:50 +0800 Subject: [PATCH 2/7] update fsaf igie results Signed-off-by: majorli --- models/cv/detection/fsaf/igie/README.md | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/models/cv/detection/fsaf/igie/README.md b/models/cv/detection/fsaf/igie/README.md index 14d56932..7a587dc0 100644 --- a/models/cv/detection/fsaf/igie/README.md +++ b/models/cv/detection/fsaf/igie/README.md @@ -24,12 +24,17 @@ pip3 install mmengine==0.10.4 ``` ### Download + Pretrained model: Dataset: to download the validation dataset. +```bash +wget https://download.openmmlab.com/mmdetection/v2.0/fsaf/fsaf_r50_fpn_1x_coco/fsaf_r50_fpn_1x_coco-94ccc51f.pth +``` ### Model Conversion + ```bash # export onnx model python3 export.py --weight fsaf_r50_fpn_1x_coco-94ccc51f.pth --cfg fsaf_r50_fpn_1x_coco.py --output fsaf.onnx @@ -55,6 +60,6 @@ bash scripts/infer_fsaf_fp16_performance.sh ## Results -| Model | BatchSize | Input Shape | Precision | FPS | mAP@0.5(%) | -| :-------: | :-------: | :---------: | :-------: | :-------: | :--------: | -| FSAF | 32 | 800x800 | FP16 | 122.35 | 0.530 | \ No newline at end of file +| Model | BatchSize | Input Shape | Precision | FPS | mAP@0.5(%) | +| :---: | :-------: | :---------: | :-------: | :-----: | :--------: | +| FSAF | 32 | 800x800 | FP16 | 178.748 | 0.530 | -- Gitee From 105a59580fcefff8a30d8b6cfc98db8ae538439a Mon Sep 17 00:00:00 2001 From: majorli Date: Wed, 24 Jul 2024 17:31:27 +0800 Subject: [PATCH 3/7] update fcos igie results Signed-off-by: majorli --- models/cv/detection/fcos/igie/README.md | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/models/cv/detection/fcos/igie/README.md b/models/cv/detection/fcos/igie/README.md index f10cc742..2aa98422 100644 --- a/models/cv/detection/fcos/igie/README.md +++ b/models/cv/detection/fcos/igie/README.md @@ -24,12 +24,17 @@ pip3 install mmengine==0.10.4 ``` ### Download + Pretrained model: Dataset: to download the validation dataset. +```bash +wget https://download.openmmlab.com/mmdetection/v2.0/fcos/fcos_r50_caffe_fpn_gn-head_1x_coco/fcos_r50_caffe_fpn_gn-head_1x_coco-821213aa.pth +``` ### Model Conversion + ```bash # export onnx model python3 export.py --weight fcos_r50_caffe_fpn_gn-head_1x_coco-821213aa.pth --cfg fcos_r50_caffe_fpn_gn-head_1x_coco.py --output fcos.onnx @@ -55,6 +60,6 @@ bash scripts/infer_fcos_fp16_performance.sh ## Results -| Model | BatchSize | Input Shape | Precision | FPS | mAP@0.5(%) | -| :-------: | :-------: | :---------: | :-------: | :-------: | :--------: | -| FCOS | 32 | 800x800 | FP16 | 83.09 | 0.522 | \ No newline at end of file +| Model | BatchSize | Input Shape | Precision | FPS | mAP@0.5(%) | +| :---: | :-------: | :---------: | :-------: | :-----: | :--------: | +| FCOS | 32 | 800x800 | FP16 | 135.019 | 0.522 | -- Gitee From 64db7772d62086abc98cea9a5f3253d0c18eaff8 Mon Sep 17 00:00:00 2001 From: majorli Date: Wed, 24 Jul 2024 17:38:40 +0800 Subject: [PATCH 4/7] update atss igie results Signed-off-by: majorli --- models/cv/detection/atss/igie/README.md | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/models/cv/detection/atss/igie/README.md b/models/cv/detection/atss/igie/README.md index c64581d4..41389251 100644 --- a/models/cv/detection/atss/igie/README.md +++ b/models/cv/detection/atss/igie/README.md @@ -24,12 +24,17 @@ pip3 install mmengine==0.10.4 ``` ### Download -Pretrained model: + +Pretrained model: Dataset: to download the validation dataset. +```bash +wget https://download.openmmlab.com/mmdetection/v2.0/atss/atss_r50_fpn_1x_coco/atss_r50_fpn_1x_coco_20200209-985f7bd0.pth +``` ### Model Conversion + ```bash # export onnx model python3 export.py --weight atss_r50_fpn_1x_coco_20200209-985f7bd0.pth --cfg atss_r50_fpn_1x_coco.py --output atss.onnx @@ -55,6 +60,6 @@ bash scripts/infer_atss_fp16_performance.sh ## Results -| Model | BatchSize | Input Shape | Precision | FPS | mAP@0.5(%) | -| :-------: | :-------: | :---------: | :-------: | :-------: | :--------: | -| ATSS | 32 | 800x800 | FP16 | 81.671 | 0.541 | \ No newline at end of file +| Model | BatchSize | Input Shape | Precision | FPS | mAP@0.5(%) | +| :---: | :-------: | :---------: | :-------: | :-----: | :--------: | +| ATSS | 32 | 800x800 | FP16 | 126.864 | 0.541 | -- Gitee From 9960c7e06b1975994cf236bb29a8348e60a838b0 Mon Sep 17 00:00:00 2001 From: majorli Date: Wed, 24 Jul 2024 17:52:05 +0800 Subject: [PATCH 5/7] update retinaface igie results Signed-off-by: majorli --- models/cv/detection/retinaface/igie/README.md | 22 +++++++++++++++---- 1 file changed, 18 insertions(+), 4 deletions(-) diff --git a/models/cv/detection/retinaface/igie/README.md b/models/cv/detection/retinaface/igie/README.md index 18da6c3c..a4eb4e4d 100755 --- a/models/cv/detection/retinaface/igie/README.md +++ b/models/cv/detection/retinaface/igie/README.md @@ -7,7 +7,14 @@ RetinaFace is an efficient single-stage face detection model that employs a mult ## Setup ### Install + ```bash +# Install libGL +## CentOS +yum install -y mesa-libGL +## Ubuntu +apt install -y libgl1-mesa-dev + pip3 install onnx pip3 install tqdm pip3 install onnxsim @@ -15,11 +22,17 @@ pip3 install opencv-python==4.6.0.66 ``` ### Download + Pretrained model: Dataset: to download the validation dataset. +```bash +wget https://github.com/biubug6/Face-Detector-1MB-with-landmark/raw/master/weights/mobilenet0.25_Final.pth +``` + ### Model Conversion + ```bash # export onnx model python3 export.py --weight mobilenet0.25_Final.pth --output retinaface.onnx @@ -35,6 +48,7 @@ export DATASETS_DIR=/Path/to/widerface/ ``` ### FP16 + ```bash # Accuracy bash scripts/infer_retinaface_fp16_accuracy.sh @@ -44,10 +58,10 @@ bash scripts/infer_retinaface_fp16_performance.sh ## Results -| Model | BatchSize | Precision | FPS | Easy AP(%) | Medium AP (%) | Hard AP(%) | -| :--------------: | :-------: | :-------: | :-----: | :--------: | :-----------: | :--------: | -| RetinaFace | 32 | FP16 | 5575.56 | 80.13 | 68.52 | 36.59 | +| Model | BatchSize | Precision | FPS | Easy AP(%) | Medium AP (%) | Hard AP(%) | +| :--------: | :-------: | :-------: | :------: | :--------: | :-----------: | :--------: | +| RetinaFace | 32 | FP16 | 8304.626 | 80.13 | 68.52 | 36.59 | ## Reference -Face-Detector-1MB-with-landmark: \ No newline at end of file +Face-Detector-1MB-with-landmark: -- Gitee From aed55b4e42bf6343ac755c5237d18e0b3af89131 Mon Sep 17 00:00:00 2001 From: majorli Date: Thu, 25 Jul 2024 11:23:16 +0800 Subject: [PATCH 6/7] update rtmpose igie results Signed-off-by: majorli --- models/cv/pose_estimation/rtmpose/igie/README.md | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/models/cv/pose_estimation/rtmpose/igie/README.md b/models/cv/pose_estimation/rtmpose/igie/README.md index 2b7fb929..02cbc5fc 100644 --- a/models/cv/pose_estimation/rtmpose/igie/README.md +++ b/models/cv/pose_estimation/rtmpose/igie/README.md @@ -25,12 +25,17 @@ pip3 install mmengine==0.10.4 ``` ### Download -Pretrained model: https://download.openmmlab.com/mmpose/v1/projects/rtmposev1/rtmpose-m_simcc-aic-coco_pt-aic-coco_420e-256x192-63eb25f7_20230126.pth + +Pretrained model: Dataset: to download the validation dataset. +```bash +wget https://download.openmmlab.com/mmpose/v1/projects/rtmposev1/rtmpose-m_simcc-aic-coco_pt-aic-coco_420e-256x192-63eb25f7_20230126.pth +``` ### Model Conversion + ```bash # export onnx model python3 export.py --weight rtmpose-m_simcc-aic-coco_pt-aic-coco_420e-256x192-63eb25f7_20230126.pth --cfg rtmpose-m_8xb256-420e_coco-256x192.py --output rtmpose.onnx @@ -56,6 +61,6 @@ bash scripts/infer_rtmpose_fp16_performance.sh ## Results -| Model | BatchSize | Input Shape | Precision | FPS | mAP@0.5(%) | -| :-------: | :-------: | :---------: | :-------: | :-------: | :--------: | -| RTMPose | 32 | 252x196 | FP16 | 2313.33 | 0.936 | +| Model | BatchSize | Input Shape | Precision | FPS | mAP@0.5(%) | +| :-----: | :-------: | :---------: | :-------: | :------: | :--------: | +| RTMPose | 32 | 256x192 | FP16 | 3459.341 | 0.936 | -- Gitee From ebc3936f250e601f251b76157d9f24713ca60ee8 Mon Sep 17 00:00:00 2001 From: majorli Date: Thu, 25 Jul 2024 11:25:24 +0800 Subject: [PATCH 7/7] update copyright Signed-off-by: majorli --- .../retinaface/igie/widerface_evaluate/box_overlaps.pyx | 2 ++ 1 file changed, 2 insertions(+) diff --git a/models/cv/detection/retinaface/igie/widerface_evaluate/box_overlaps.pyx b/models/cv/detection/retinaface/igie/widerface_evaluate/box_overlaps.pyx index 66544c1f..3cedef8f 100755 --- a/models/cv/detection/retinaface/igie/widerface_evaluate/box_overlaps.pyx +++ b/models/cv/detection/retinaface/igie/widerface_evaluate/box_overlaps.pyx @@ -1,6 +1,8 @@ # -------------------------------------------------------- # Fast R-CNN # Copyright (c) 2015 Microsoft +# Copyright (c) 2024, Shanghai Iluvatar CoreX Semiconductor Co., Ltd. +# All Rights Reserved. # Licensed under The MIT License [see LICENSE for details] # Written by Sergey Karayev # -------------------------------------------------------- -- Gitee