From 15b05b9b68126c497ebb4eaa1ef0bd81840ece78 Mon Sep 17 00:00:00 2001 From: shangliang Xu Date: Fri, 26 Nov 2021 12:29:01 +0800 Subject: [PATCH] [TIPC] add models, test=document_fix (#4722) --- test_tipc/README.md | 64 +------- ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 29 ++++ .../s2anet_1x_spine_train_infer_python.txt | 51 ++++++ ..._normal_amp_infer_python_linux_gpu_cpu.txt | 51 ++++++ .../dota/s2anet_alignconv_2x_spine.yml | 26 ++++ ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 29 ++++ ..._alignconv_2x_spine_train_infer_python.txt | 51 ++++++ ..._normal_amp_infer_python_linux_gpu_cpu.txt | 51 ++++++ .../configs/dota/s2anet_conv_2x_spine.yml | 31 ++++ ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 29 ++++ ...2anet_conv_2x_spine_train_infer_python.txt | 51 ++++++ ..._normal_amp_infer_python_linux_gpu_cpu.txt | 51 ++++++ ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 29 ++++ .../blazeface_1000e_train_infer_python.txt | 51 ++++++ ..._normal_amp_infer_python_linux_gpu_cpu.txt | 51 ++++++ ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 29 ++++ ...eface_fpn_ssh_1000e_train_infer_python.txt | 51 ++++++ ..._normal_amp_infer_python_linux_gpu_cpu.txt | 51 ++++++ ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 2 +- ...r_rcnn_r101_1x_coco_train_infer_python.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 4 +- ...nn_r101_fpn_1x_coco_train_infer_python.txt | 6 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 6 +- ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 2 +- ...r101_vd_fpn_1x_coco_train_infer_python.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 2 +- ...cnn_r34_fpn_1x_coco_train_infer_python.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 2 +- ..._r34_vd_fpn_1x_coco_train_infer_python.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 2 +- ...er_rcnn_r50_1x_coco_train_infer_python.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 2 +- ...cnn_r50_fpn_1x_coco_train_infer_python.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 2 +- ...rcnn_r50_vd_1x_coco_train_infer_python.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 2 +- ..._r50_vd_fpn_1x_coco_train_infer_python.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 2 +- ...in_tiny_fpn_1x_coco_train_infer_python.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 2 +- ...d_64x4d_fpn_1x_coco_train_infer_python.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 2 +- ...cos_r50_fpn_1x_coco_train_infer_python.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ...k_hrnet_w32_256x192_train_infer_python.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 2 +- ...k_hrnet_w48_256x192_train_infer_python.txt | 6 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 6 +- ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 2 +- ...hrnet_hrnet_w32_512_train_infer_python.txt | 4 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 4 +- .../hrnet_w32_256x192_train_infer_python.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- .../configs/keypoint/tinypose_128x96.yml | 147 ++++++++++++++++++ ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 8 +- .../tinypose_128x96_train_infer_python.txt | 18 +-- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 18 +-- ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 2 +- ...nn_r101_fpn_1x_coco_train_infer_python.txt | 4 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 2 +- ...r101_vd_fpn_1x_coco_train_infer_python.txt | 4 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 2 +- ...sk_rcnn_r50_1x_coco_train_infer_python.txt | 4 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 2 +- ...cnn_r50_fpn_1x_coco_train_infer_python.txt | 4 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 2 +- ..._r50_vd_fpn_1x_coco_train_infer_python.txt | 4 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 2 +- ...d_64x4d_fpn_1x_coco_train_infer_python.txt | 4 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 29 ++++ ...t_dla34_30e_576x320_train_infer_python.txt | 51 ++++++ ..._normal_amp_infer_python_linux_gpu_cpu.txt | 51 ++++++ ...fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml | 43 +++++ ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 29 ++++ ..._dlafpn_30e_576x320_train_infer_python.txt | 51 ++++++ ..._normal_amp_infer_python_linux_gpu_cpu.txt | 51 ++++++ ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 29 ++++ ...rknet53_30e_576x320_train_infer_python.txt | 51 ++++++ ..._normal_amp_infer_python_linux_gpu_cpu.txt | 51 ++++++ .../picodet_l_640_coco_train_infer_python.txt | 2 +- ...lcnet_1_5x_416_coco_train_infer_python.txt | 2 +- .../picodet_m_416_coco_train_infer_python.txt | 2 +- ...3_large_1x_416_coco_train_infer_python.txt | 2 +- ...icodet_r18_640_coco_train_infer_python.txt | 2 +- .../picodet_s_320_coco_train_infer_python.txt | 2 +- ...lenetv2_1x_416_coco_train_infer_python.txt | 2 +- ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 4 +- ...r101_vd_fpn_3x_coco_train_infer_python.txt | 4 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 4 +- ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 2 +- ...v2_r50_enhance_coco_train_infer_python.txt | 2 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 2 +- ..._normal_normal_infer_cpp_linux_gpu_cpu.txt | 4 +- ...ov2_r50_fpn_1x_coco_train_infer_python.txt | 4 +- ..._normal_amp_infer_python_linux_gpu_cpu.txt | 4 +- test_tipc/docs/install.md | 2 +- test_tipc/docs/more_models.md | 82 ++++++++++ test_tipc/prepare.sh | 77 ++++----- 115 files changed, 1544 insertions(+), 226 deletions(-) create mode 100644 test_tipc/configs/dota/s2anet_1x_spine_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt create mode 100644 test_tipc/configs/dota/s2anet_1x_spine_train_infer_python.txt create mode 100644 test_tipc/configs/dota/s2anet_1x_spine_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt create mode 100644 test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml create mode 100644 test_tipc/configs/dota/s2anet_alignconv_2x_spine_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt create mode 100644 test_tipc/configs/dota/s2anet_alignconv_2x_spine_train_infer_python.txt create mode 100644 test_tipc/configs/dota/s2anet_alignconv_2x_spine_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt create mode 100644 test_tipc/configs/dota/s2anet_conv_2x_spine.yml create mode 100644 test_tipc/configs/dota/s2anet_conv_2x_spine_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt create mode 100644 test_tipc/configs/dota/s2anet_conv_2x_spine_train_infer_python.txt create mode 100644 test_tipc/configs/dota/s2anet_conv_2x_spine_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt create mode 100644 test_tipc/configs/face_detection/blazeface_1000e_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt create mode 100644 test_tipc/configs/face_detection/blazeface_1000e_train_infer_python.txt create mode 100644 test_tipc/configs/face_detection/blazeface_1000e_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt create mode 100644 test_tipc/configs/face_detection/blazeface_fpn_ssh_1000e_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt create mode 100644 test_tipc/configs/face_detection/blazeface_fpn_ssh_1000e_train_infer_python.txt create mode 100644 test_tipc/configs/face_detection/blazeface_fpn_ssh_1000e_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt create mode 100644 test_tipc/configs/keypoint/tinypose_128x96.yml create mode 100644 test_tipc/configs/mot/fairmot_dla34_30e_576x320_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt create mode 100644 test_tipc/configs/mot/fairmot_dla34_30e_576x320_train_infer_python.txt create mode 100644 test_tipc/configs/mot/fairmot_dla34_30e_576x320_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt create mode 100644 test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml create mode 100644 test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt create mode 100644 test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320_train_infer_python.txt create mode 100644 test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt create mode 100644 test_tipc/configs/mot/jde_darknet53_30e_576x320_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt create mode 100644 test_tipc/configs/mot/jde_darknet53_30e_576x320_train_infer_python.txt create mode 100644 test_tipc/configs/mot/jde_darknet53_30e_576x320_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt create mode 100644 test_tipc/docs/more_models.md diff --git a/test_tipc/README.md b/test_tipc/README.md index 5231ae7b0..befe05f5b 100644 --- a/test_tipc/README.md +++ b/test_tipc/README.md @@ -25,73 +25,11 @@ | 算法论文 | 模型名称 | 模型类型 | 基础
训练预测 | 更多
训练方式 | 模型压缩 | 其他预测部署 | | :--- | :--- | :----: | :--------: | :---- | :---- | :---- | -| [YOLOv3](https://arxiv.org/abs/1804.02767) | [yolov3_darknet53_270e_coco](../configs/yolov3/yolov3_darknet53_270e_coco.yml) | 目标检测 | 支持 | 混合精度 | FPGM裁剪
PACT量化
离线量化 | Paddle Inference: C++ | -| YOLOv3 | [yolov3_mobilenet_v1_270e_coco](../configs/yolov3/yolov3_mobilenet_v1_270e_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| YOLOv3 | [yolov3_mobilenet_v3_large_270e_coco](../configs/yolov3/yolov3_mobilenet_v3_large_270e_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| YOLOv3 | [yolov3_r34_270e_coco](../configs/yolov3/yolov3_r34_270e_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| YOLOv3 | [yolov3_r50vd_dcn_270e_coco](../configs/yolov3/yolov3_r50vd_dcn_270e_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | | [PPYOLO](https://arxiv.org/abs/2007.12099) | [ppyolo_mbv3_large_coco](../configs/ppyolo/ppyolo_mbv3_large_coco.yml) | 目标检测 | 支持 | 混合精度 | FPGM裁剪
PACT量化
离线量化 | Paddle Inference: C++ | -| PPYOLO | [ppyolo_r50vd_dcn_1x_coco](../configs/ppyolo/ppyolo_r50vd_dcn_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | FPGM裁剪
PACT量化
离线量化 | Paddle Inference: C++ | -| PPYOLO | [ppyolo_mbv3_small_coco](../configs/ppyolo/ppyolo_mbv3_small_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| PPYOLO | [ppyolo_r18vd_coco](../configs/ppyolo/ppyolo_r18vd_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| PPYOLO-tiny | [ppyolo_tiny_650e_coco](../configs/ppyolo/ppyolo_tiny_650e_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | | [PPYOLOv2](https://arxiv.org/abs/2104.10419) | [ppyolov2_r50vd_dcn_365e_coco](../configs/ppyolo/ppyolov2_r50vd_dcn_365e_coco.yml) | 目标检测 | 支持 | 多机多卡
混合精度 | | Paddle Inference: C++ | -| PPYOLOv2 | [ppyolov2_r50vd_dcn_365e_coco](../configs/ppyolo/ppyolov2_r50vd_dcn_365e_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| PPYOLOv2 | [ppyolov2_r101vd_dcn_365e_coco](../configs/ppyolo/ppyolov2_r101vd_dcn_365e_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | | [PP-PicoDet](https://arxiv.org/abs/2111.00902) | [picodet_s_320_coco](../configs/picodet/picodet_s_320_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| PP-PicoDet | [picodet_m_416_coco](../configs/picodet/picodet_m_416_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| PP-PicoDet | [picodet_l_640_coco](../configs/picodet/picodet_l_640_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| PP-PicoDet | [picodet_lcnet_1_5x_416_coco](../configs/picodet/more_config/picodet_lcnet_1_5x_416_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| PP-PicoDet | [picodet_mobilenetv3_large_1x_416_coco](../configs/picodet/more_config/picodet_mobilenetv3_large_1x_416_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| PP-PicoDet | [picodet_r18_640_coco](../configs/picodet/more_config/picodet_r18_640_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| PP-PicoDet | [picodet_shufflenetv2_1x_416_coco](../configs/picodet/more_config/picodet_shufflenetv2_1x_416_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| [SSD](https://arxiv.org/abs/1512.02325) | [ssdlite_mobilenet_v1_300_coco](../configs/ssd/ssdlite_mobilenet_v1_300_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| [Faster R-CNN](https://arxiv.org/abs/1506.01497) | [faster_rcnn_r50_fpn_1x_coco](../configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| Faster R-CNN | [faster_rcnn_r34_fpn_1x_coco](../configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| Faster R-CNN | [faster_rcnn_r34_vd_fpn_1x_coco](../configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| Faster R-CNN | [faster_rcnn_r50_1x_coco](../configs/faster_rcnn/faster_rcnn_r50_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| Faster R-CNN | [faster_rcnn_r50_vd_1x_coco](../configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| Faster R-CNN | [faster_rcnn_r50_vd_fpn_1x_coco](../configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| Faster R-CNN | [faster_rcnn_r101_1x_coco](../configs/faster_rcnn/faster_rcnn_r101_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| Faster R-CNN | [faster_rcnn_r101_fpn_1x_coco](../configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| Faster R-CNN | [faster_rcnn_r101_vd_fpn_1x_coco](../configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| Faster R-CNN | [faster_rcnn_x101_vd_64x4d_fpn_1x_coco](../configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| Faster R-CNN | [faster_rcnn_swin_tiny_fpn_1x_coco](../configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| [Cascade Faster R-CNN](https://arxiv.org/abs/1712.00726) | [cascade_rcnn_r50_fpn_1x_coco](../configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| Cascade Faster R-CNN | [cascade_rcnn_r50_vd_fpn_ssld_1x_coco](../configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| [FCOS](https://arxiv.org/abs/1904.01355) | [fcos_r50_fpn_1x_coco](../configs/fcos/fcos_r50_fpn_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| FCOS | [fcos_dcn_r50_fpn_1x_coco](../configs/fcos/fcos_dcn_r50_fpn_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| [TTFNet](https://arxiv.org/abs/1909.00700) | [ttfnet_darknet53_1x_coco](../configs/ttfnet/ttfnet_darknet53_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| [S2ANet](https://arxiv.org/abs/2008.09397) | [s2anet_conv_2x_dota](../configs/dota/s2anet_conv_2x_dota.yml) | 目标检测 | 使用DOTA数据集,且需要编译自定义OP,接入存在风险,待接入 | -| [DETR](https://arxiv.org/abs/2005.12872) | [detr_r50_1x_coco](../configs/detr/detr_r50_1x_coco.yml) | 目标检测 | 不支持预测部署 | -| [Deformable DETR](https://arxiv.org/abs/2010.04159) | [deformable_detr_r50_1x_coco](../configs/deformable_detr/deformable_detr_r50_1x_coco.yml) | 目标检测 | 不支持预测部署 | -| [Sparse R-CNN](https://arxiv.org/abs/2011.12450) | [sparse_rcnn_r50_fpn_3x_pro100_coco](../configs/sparse_rcnn/sparse_rcnn_r50_fpn_3x_pro100_coco.yml) | 目标检测 | 不支持预测部署 | -| [BlazeFace](https://arxiv.org/abs/1907.05047) | [blazeface_1000e](../configs/face_detection/blazeface_1000e.yml) | 目标检测 | 使用WIDER-FACE数据集,接入存在风险,待接入 | -| [Faceboxes](https://arxiv.org/abs/1708.05234) | | 目标检测 | 静态图实现,暂不支持 | -| [PyramidBox](https://arxiv.org/abs/1803.07737) | 模型实现在[models](https://github.com/PaddlePaddle/models/tree/develop/PaddleCV/face_detection) repo | 目标检测 | 静态图实现,暂不支持 | -| [Libra R-CNN](https://arxiv.org/abs/1904.02701) | | 目标检测 | 静态图实现,暂不支持 | -| [RetinaNet](https://arxiv.org/abs/1708.02002) | | 目标检测 | 静态图实现,暂不支持 | -| [CornerNet](https://arxiv.org/abs/1808.01244) | | 目标检测 | 静态图实现,暂不支持 | -| [EfficientDet](https://arxiv.org/abs/1911.09070) | | 目标检测 | 静态图实现,暂不支持 | -| [Mask R-CNN](https://arxiv.org/abs/1703.06870) | [mask_rcnn_r50_fpn_1x_coco](../configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco.yml) | 实例分割 | 支持 | 混合精度 | | Paddle Inference: C++ | -| Mask R-CNN | [mask_rcnn_r50_1x_coco](../configs/mask_rcnn/mask_rcnn_r50_1x_coco.yml) | 实例分割 | 支持 | 混合精度 | | Paddle Inference: C++ | -| Mask R-CNN | [mask_rcnn_r50_vd_fpn_1x_coco](../configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco.yml) | 实例分割 | 支持 | 混合精度 | | Paddle Inference: C++ | -| Mask R-CNN | [mask_rcnn_r101_fpn_1x_coco](../configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco.yml) | 实例分割 | 支持 | 混合精度 | | Paddle Inference: C++ | -| Mask R-CNN | [mask_rcnn_r101_vd_fpn_1x_coco](../configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco.yml) | 实例分割 | 支持 | 混合精度 | | Paddle Inference: C++ | -| Mask R-CNN | [mask_rcnn_x101_vd_64x4d_fpn_1x_coco](../configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco.yml) | 实例分割 | 支持 | 混合精度 | | Paddle Inference: C++ | -| [Cascade Mask R-CNN](https://arxiv.org/abs/1906.09756) | [cascade_mask_rcnn_r50_fpn_1x_coco](../configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco.yml) | 实例分割 | 支持 | 混合精度 | | Paddle Inference: C++ | -| Cascade Mask R-CNN | [cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco](../configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco.yml) | 实例分割 | 支持 | 混合精度 | | Paddle Inference: C++ | -| [SOLOv2](https://arxiv.org/abs/2003.10152) | [solov2_r50_fpn_1x_coco](../configs/solov2/solov2_r50_fpn_1x_coco.yml) | 实例分割 | 支持 | 混合精度 | | Paddle Inference: C++ | -| SOLOv2 | [solov2_r50_enhance_coco](../configs/solov2/solov2_r50_enhance_coco.yml) | 实例分割 | 支持 | 混合精度 | | Paddle Inference: C++ | -| SOLOv2 | [solov2_r101_vd_fpn_3x_coco](../configs/solov2/solov2_r101_vd_fpn_3x_coco.yml) | 实例分割 | 支持 | 混合精度 | | Paddle Inference: C++ | -| [PP-Tinypose] | [tinypose_128x96](../configs/keypoint/tiny_pose/tinypose_128x96.yml) | 关键点检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| [HRNet](https://arxiv.org/abs/1902.09212) | [hrnet_w32_256x192](../configs/keypoint/hrnet/hrnet_w32_256x192.yml) | 关键点检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| HRNet | [dark_hrnet_w32_256x192](../configs/keypoint/hrnet/dark_hrnet_w32_256x192.yml) | 关键点检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| HRNet | [dark_hrnet_w48_256x192](../configs/keypoint/hrnet/dark_hrnet_w48_256x192.yml) | 关键点检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| [HigherHRNet](https://arxiv.org/abs/1908.10357) | [higherhrnet_hrnet_w32_512](../configs/keypoint/higherhrnet/higherhrnet_hrnet_w32_512.yml) | 关键点检测 | 支持 | 混合精度 | | Paddle Inference: C++ | -| [FairMot](https://arxiv.org/abs/2004.01888) | [fairmot_dla34_30e_576x320](../configs/mot/fairmot/fairmot_dla34_30e_576x320.yml) | 目标跟踪 | 支持,待接入 | -| [JDE](https://arxiv.org/abs/1909.12605) | [jde_darknet53_30e_576x320](../configs/mot/jde/jde_darknet53_30e_576x320.yml) | 目标跟踪 | 支持,待接入 | +更详细的汇总信息可以查看[更多模型](docs/more_models.md) ## 3. 测试工具简介 ### 目录介绍 diff --git a/test_tipc/configs/dota/s2anet_1x_spine_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/dota/s2anet_1x_spine_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt new file mode 100644 index 000000000..fb25bd8ea --- /dev/null +++ b/test_tipc/configs/dota/s2anet_1x_spine_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -0,0 +1,29 @@ +===========================cpp_infer_params=========================== +model_name:s2anet_1x_spine +python:python3.7 +filename:s2anet_1x_spine +## +--output_dir:./output_inference +weights:https://paddledet.bj.bcebos.com/models/s2anet_1x_spine.pdparams +norm_export:tools/export_model.py -c configs/dota/s2anet_1x_spine.yml -o +quant_export:tools/export_model.py -c configs/dota/s2anet_1x_spine.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c configs/dota/s2anet_1x_spine.yml --slim_config _template_fpgm -o +distill_export:null +export1:null +export2:null +kl_quant_export:tools/post_quant.py -c configs/dota/s2anet_1x_spine.yml --slim_config _template_kl_quant -o +## +opencv_dir:default +infer_mode:norm +infer_quant:False +inference:./deploy/cpp/build/main +--device:gpu|cpu +--use_mkldnn:True|False +--cpu_threads:1|6 +--batch_size:1|8 +--use_tensorrt:null +--run_mode:fluid|trt_fp32|trt_fp16 +--model_dir: +--image_dir:./dataset/spine_coco/test/ +--run_benchmark:True +null:null \ No newline at end of file diff --git a/test_tipc/configs/dota/s2anet_1x_spine_train_infer_python.txt b/test_tipc/configs/dota/s2anet_1x_spine_train_infer_python.txt new file mode 100644 index 000000000..9737fec11 --- /dev/null +++ b/test_tipc/configs/dota/s2anet_1x_spine_train_infer_python.txt @@ -0,0 +1,51 @@ +===========================train_params=========================== +model_name:s2anet_1x_spine +python:python3.7 +gpu_list:0|0,1 +use_gpu:True +auto_cast:null +epoch:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=12 +save_dir:null +TrainReader.batch_size:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=1 +pretrain_weights:https://paddledet.bj.bcebos.com/models/s2anet_1x_spine.pdparams +trained_model_name:model_final.pdparams +train_infer_img_dir:./dataset/spine_coco/test/ +filename:null +## +trainer:norm_train +norm_train:tools/train.py -c configs/dota/s2anet_1x_spine.yml -o +pact_train:tools/train.py -c configs/dota/s2anet_1x_spine.yml --slim_config _template_pact -o +fpgm_train:tools/train.py -c configs/dota/s2anet_1x_spine.yml --slim_config _template_fpgm -o +distill_train:null +null:null +null:null +## +===========================eval_params=========================== +eval:tools/eval.py -c configs/dota/s2anet_1x_spine.yml -o +null:null +## +===========================infer_params=========================== +--output_dir:./output_inference +weights:https://paddledet.bj.bcebos.com/models/s2anet_1x_spine.pdparams +norm_export:tools/export_model.py -c configs/dota/s2anet_1x_spine.yml -o +pact_export:tools/export_model.py -c configs/dota/s2anet_1x_spine.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c configs/dota/s2anet_1x_spine.yml --slim_config _template_fpgm -o +distill_export:null +export1:null +export2:null +kl_quant_export:tools/post_quant.py -c configs/dota/s2anet_1x_spine.yml --slim_config _template_kl_quant -o +## +infer_mode:norm +infer_quant:False +inference:./deploy/python/infer.py +--device:gpu|cpu +--enable_mkldnn:True|False +--cpu_threads:1|6 +--batch_size:1|8 +--use_tensorrt:null +--run_mode:fluid|trt_fp32|trt_fp16 +--model_dir: +--image_dir:./dataset/spine_coco/test/ +--save_log_path:null +--run_benchmark:True +null:null \ No newline at end of file diff --git a/test_tipc/configs/dota/s2anet_1x_spine_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/dota/s2anet_1x_spine_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt new file mode 100644 index 000000000..ea7296398 --- /dev/null +++ b/test_tipc/configs/dota/s2anet_1x_spine_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -0,0 +1,51 @@ +===========================train_params=========================== +model_name:s2anet_1x_spine +python:python3.7 +gpu_list:0|0,1 +use_gpu:True +auto_cast:amp +epoch:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=12 +save_dir:null +TrainReader.batch_size:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=1 +pretrain_weights:https://paddledet.bj.bcebos.com/models/s2anet_1x_spine.pdparams +trained_model_name:s2anet_1x_spine/model_final.pdparams +train_infer_img_dir:./dataset/spine_coco/test/ +filename:null +## +trainer:norm_train +norm_train:tools/train.py -c configs/dota/s2anet_1x_spine.yml -o +pact_train:tools/train.py -c configs/dota/s2anet_1x_spine.yml --slim_config _template_pact -o +fpgm_train:tools/train.py -c configs/dota/s2anet_1x_spine.yml --slim_config _template_fpgm -o +distill_train:null +null:null +null:null +## +===========================eval_params=========================== +eval:tools/eval.py -c configs/dota/s2anet_1x_spine.yml -o +null:null +## +===========================infer_params=========================== +--output_dir:./output_inference +weights:https://paddledet.bj.bcebos.com/models/s2anet_1x_spine.pdparams +norm_export:tools/export_model.py -c configs/dota/s2anet_1x_spine.yml -o +pact_export:tools/export_model.py -c configs/dota/s2anet_1x_spine.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c configs/dota/s2anet_1x_spine.yml --slim_config _template_fpgm -o +distill_export:null +export1:null +export2:null +kl_quant_export:tools/post_quant.py -c configs/dota/s2anet_1x_spine.yml --slim_config _template_kl_quant -o +## +infer_mode:norm +infer_quant:False +inference:./deploy/python/infer.py +--device:gpu|cpu +--enable_mkldnn:True|False +--cpu_threads:1|6 +--batch_size:1|8 +--use_tensorrt:null +--run_mode:fluid|trt_fp16 +--model_dir: +--image_dir:./dataset/spine_coco/test/ +--save_log_path:null +--run_benchmark:True +null:null \ No newline at end of file diff --git a/test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml b/test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml new file mode 100644 index 000000000..89f3b5f8a --- /dev/null +++ b/test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml @@ -0,0 +1,26 @@ +_BASE_: [ + '../../../configs/datasets/spine_coco.yml', + '../../../configs/runtime.yml', + '../../../configs/dota/_base_/s2anet_optimizer_2x.yml', + '../../../configs/dota/_base_/s2anet.yml', + '../../../configs/dota/_base_/s2anet_reader.yml', +] +pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_vd_ssld_v2_pretrained.pdparams + +weights: output/s2anet_alignconv_2x_dota/model_final + +S2ANetHead: + anchor_strides: [8, 16, 32, 64, 128] + anchor_scales: [4] + anchor_ratios: [1.0] + anchor_assign: RBoxAssigner + stacked_convs: 2 + feat_in: 256 + feat_out: 256 + num_classes: 9 + align_conv_type: 'AlignConv' # AlignConv Conv + align_conv_size: 3 + use_sigmoid_cls: True + reg_loss_weight: [1.0, 1.0, 1.0, 1.0, 1.1] + cls_loss_weight: [1.1, 1.05] + reg_loss_type: 'l1' diff --git a/test_tipc/configs/dota/s2anet_alignconv_2x_spine_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/dota/s2anet_alignconv_2x_spine_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt new file mode 100644 index 000000000..1d6426439 --- /dev/null +++ b/test_tipc/configs/dota/s2anet_alignconv_2x_spine_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -0,0 +1,29 @@ +===========================cpp_infer_params=========================== +model_name:s2anet_alignconv_2x_spine +python:python3.7 +filename:s2anet_alignconv_2x_spine +## +--output_dir:./output_inference +weights:https://paddledet.bj.bcebos.com/models/s2anet_alignconv_2x_dota.pdparams +norm_export:tools/export_model.py -c test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml -o +quant_export:tools/export_model.py -c test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml --slim_config _template_fpgm -o +distill_export:null +export1:null +export2:null +kl_quant_export:tools/post_quant.py -c test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml --slim_config _template_kl_quant -o +## +opencv_dir:default +infer_mode:norm +infer_quant:False +inference:./deploy/cpp/build/main +--device:gpu|cpu +--use_mkldnn:True|False +--cpu_threads:1|6 +--batch_size:1|8 +--use_tensorrt:null +--run_mode:fluid|trt_fp32|trt_fp16 +--model_dir: +--image_dir:./dataset/spine_coco/test/ +--run_benchmark:True +null:null \ No newline at end of file diff --git a/test_tipc/configs/dota/s2anet_alignconv_2x_spine_train_infer_python.txt b/test_tipc/configs/dota/s2anet_alignconv_2x_spine_train_infer_python.txt new file mode 100644 index 000000000..1007ccf57 --- /dev/null +++ b/test_tipc/configs/dota/s2anet_alignconv_2x_spine_train_infer_python.txt @@ -0,0 +1,51 @@ +===========================train_params=========================== +model_name:s2anet_alignconv_2x_spine +python:python3.7 +gpu_list:0|0,1 +use_gpu:True +auto_cast:null +epoch:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=24 +save_dir:null +TrainReader.batch_size:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=1 +pretrain_weights:https://paddledet.bj.bcebos.com/models/s2anet_alignconv_2x_dota.pdparams +trained_model_name:model_final.pdparams +train_infer_img_dir:./dataset/spine_coco/test/ +filename:null +## +trainer:norm_train +norm_train:tools/train.py -c test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml -o +pact_train:tools/train.py -c test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml --slim_config _template_pact -o +fpgm_train:tools/train.py -c test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml --slim_config _template_fpgm -o +distill_train:null +null:null +null:null +## +===========================eval_params=========================== +eval:tools/eval.py -c test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml -o +null:null +## +===========================infer_params=========================== +--output_dir:./output_inference +weights:https://paddledet.bj.bcebos.com/models/s2anet_alignconv_2x_dota.pdparams +norm_export:tools/export_model.py -c test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml -o +pact_export:tools/export_model.py -c test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml --slim_config _template_fpgm -o +distill_export:null +export1:null +export2:null +kl_quant_export:tools/post_quant.py -c test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml --slim_config _template_kl_quant -o +## +infer_mode:norm +infer_quant:False +inference:./deploy/python/infer.py +--device:gpu|cpu +--enable_mkldnn:True|False +--cpu_threads:1|6 +--batch_size:1|8 +--use_tensorrt:null +--run_mode:fluid|trt_fp32|trt_fp16 +--model_dir: +--image_dir:./dataset/spine_coco/test/ +--save_log_path:null +--run_benchmark:True +null:null \ No newline at end of file diff --git a/test_tipc/configs/dota/s2anet_alignconv_2x_spine_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/dota/s2anet_alignconv_2x_spine_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt new file mode 100644 index 000000000..6b1ea1605 --- /dev/null +++ b/test_tipc/configs/dota/s2anet_alignconv_2x_spine_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -0,0 +1,51 @@ +===========================train_params=========================== +model_name:s2anet_alignconv_2x_spine +python:python3.7 +gpu_list:0|0,1 +use_gpu:True +auto_cast:amp +epoch:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=24 +save_dir:null +TrainReader.batch_size:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=1 +pretrain_weights:https://paddledet.bj.bcebos.com/models/s2anet_alignconv_2x_dota.pdparams +trained_model_name:s2anet_alignconv_2x_spine/model_final.pdparams +train_infer_img_dir:./dataset/spine_coco/test/ +filename:null +## +trainer:norm_train +norm_train:tools/train.py -c test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml -o +pact_train:tools/train.py -c test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml --slim_config _template_pact -o +fpgm_train:tools/train.py -c test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml --slim_config _template_fpgm -o +distill_train:null +null:null +null:null +## +===========================eval_params=========================== +eval:tools/eval.py -c test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml -o +null:null +## +===========================infer_params=========================== +--output_dir:./output_inference +weights:https://paddledet.bj.bcebos.com/models/s2anet_alignconv_2x_dota.pdparams +norm_export:tools/export_model.py -c test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml -o +pact_export:tools/export_model.py -c test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml --slim_config _template_fpgm -o +distill_export:null +export1:null +export2:null +kl_quant_export:tools/post_quant.py -c test_tipc/configs/dota/s2anet_alignconv_2x_spine.yml --slim_config _template_kl_quant -o +## +infer_mode:norm +infer_quant:False +inference:./deploy/python/infer.py +--device:gpu|cpu +--enable_mkldnn:True|False +--cpu_threads:1|6 +--batch_size:1|8 +--use_tensorrt:null +--run_mode:fluid|trt_fp16 +--model_dir: +--image_dir:./dataset/spine_coco/test/ +--save_log_path:null +--run_benchmark:True +null:null \ No newline at end of file diff --git a/test_tipc/configs/dota/s2anet_conv_2x_spine.yml b/test_tipc/configs/dota/s2anet_conv_2x_spine.yml new file mode 100644 index 000000000..746ef0cc9 --- /dev/null +++ b/test_tipc/configs/dota/s2anet_conv_2x_spine.yml @@ -0,0 +1,31 @@ +_BASE_: [ + '../../../configs/datasets/spine_coco.yml', + '../../../configs/runtime.yml', + '../../../configs/dota/_base_/s2anet_optimizer_2x.yml', + '../../../configs/dota/_base_/s2anet.yml', + '../../../configs/dota/_base_/s2anet_reader.yml', +] +weights: output/s2anet_conv_1x_dota/model_final +pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_cos_pretrained.pdparams + +ResNet: + depth: 50 + variant: b + norm_type: bn + return_idx: [1,2,3] + num_stages: 4 + +S2ANetHead: + anchor_strides: [8, 16, 32, 64, 128] + anchor_scales: [4] + anchor_ratios: [1.0] + anchor_assign: RBoxAssigner + stacked_convs: 2 + feat_in: 256 + feat_out: 256 + num_classes: 9 + align_conv_type: 'Conv' # AlignConv Conv + align_conv_size: 3 + use_sigmoid_cls: True + reg_loss_weight: [1.0, 1.0, 1.0, 1.0, 1.1] + cls_loss_weight: [1.1, 1.05] diff --git a/test_tipc/configs/dota/s2anet_conv_2x_spine_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/dota/s2anet_conv_2x_spine_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt new file mode 100644 index 000000000..e90e4c892 --- /dev/null +++ b/test_tipc/configs/dota/s2anet_conv_2x_spine_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -0,0 +1,29 @@ +===========================cpp_infer_params=========================== +model_name:s2anet_conv_2x_spine +python:python3.7 +filename:s2anet_conv_2x_spine +## +--output_dir:./output_inference +weights:https://paddledet.bj.bcebos.com/models/s2anet_conv_2x_dota.pdparams +norm_export:tools/export_model.py -c test_tipc/configs/dota/s2anet_conv_2x_spine.yml -o +quant_export:tools/export_model.py -c test_tipc/configs/dota/s2anet_conv_2x_spine.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c test_tipc/configs/dota/s2anet_conv_2x_spine.yml --slim_config _template_fpgm -o +distill_export:null +export1:null +export2:null +kl_quant_export:tools/post_quant.py -c test_tipc/configs/dota/s2anet_conv_2x_spine.yml --slim_config _template_kl_quant -o +## +opencv_dir:default +infer_mode:norm +infer_quant:False +inference:./deploy/cpp/build/main +--device:gpu|cpu +--use_mkldnn:True|False +--cpu_threads:1|6 +--batch_size:1|8 +--use_tensorrt:null +--run_mode:fluid|trt_fp32|trt_fp16 +--model_dir: +--image_dir:./dataset/spine_coco/test/ +--run_benchmark:True +null:null \ No newline at end of file diff --git a/test_tipc/configs/dota/s2anet_conv_2x_spine_train_infer_python.txt b/test_tipc/configs/dota/s2anet_conv_2x_spine_train_infer_python.txt new file mode 100644 index 000000000..9b26a809a --- /dev/null +++ b/test_tipc/configs/dota/s2anet_conv_2x_spine_train_infer_python.txt @@ -0,0 +1,51 @@ +===========================train_params=========================== +model_name:s2anet_conv_2x_spine +python:python3.7 +gpu_list:0|0,1 +use_gpu:True +auto_cast:null +epoch:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=24 +save_dir:null +TrainReader.batch_size:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=1 +pretrain_weights:https://paddledet.bj.bcebos.com/models/s2anet_conv_2x_dota.pdparams +trained_model_name:model_final.pdparams +train_infer_img_dir:./dataset/spine_coco/test/ +filename:null +## +trainer:norm_train +norm_train:tools/train.py -c test_tipc/configs/dota/s2anet_conv_2x_spine.yml -o +pact_train:tools/train.py -c test_tipc/configs/dota/s2anet_conv_2x_spine.yml --slim_config _template_pact -o +fpgm_train:tools/train.py -c test_tipc/configs/dota/s2anet_conv_2x_spine.yml --slim_config _template_fpgm -o +distill_train:null +null:null +null:null +## +===========================eval_params=========================== +eval:tools/eval.py -c test_tipc/configs/dota/s2anet_conv_2x_spine.yml -o +null:null +## +===========================infer_params=========================== +--output_dir:./output_inference +weights:https://paddledet.bj.bcebos.com/models/s2anet_conv_2x_dota.pdparams +norm_export:tools/export_model.py -c test_tipc/configs/dota/s2anet_conv_2x_spine.yml -o +pact_export:tools/export_model.py -c test_tipc/configs/dota/s2anet_conv_2x_spine.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c test_tipc/configs/dota/s2anet_conv_2x_spine.yml --slim_config _template_fpgm -o +distill_export:null +export1:null +export2:null +kl_quant_export:tools/post_quant.py -c test_tipc/configs/dota/s2anet_conv_2x_spine.yml --slim_config _template_kl_quant -o +## +infer_mode:norm +infer_quant:False +inference:./deploy/python/infer.py +--device:gpu|cpu +--enable_mkldnn:True|False +--cpu_threads:1|6 +--batch_size:1|8 +--use_tensorrt:null +--run_mode:fluid|trt_fp32|trt_fp16 +--model_dir: +--image_dir:./dataset/spine_coco/test/ +--save_log_path:null +--run_benchmark:True +null:null \ No newline at end of file diff --git a/test_tipc/configs/dota/s2anet_conv_2x_spine_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/dota/s2anet_conv_2x_spine_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt new file mode 100644 index 000000000..9ca4184e4 --- /dev/null +++ b/test_tipc/configs/dota/s2anet_conv_2x_spine_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -0,0 +1,51 @@ +===========================train_params=========================== +model_name:s2anet_conv_2x_spine +python:python3.7 +gpu_list:0|0,1 +use_gpu:True +auto_cast:amp +epoch:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=24 +save_dir:null +TrainReader.batch_size:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=1 +pretrain_weights:https://paddledet.bj.bcebos.com/models/s2anet_conv_2x_dota.pdparams +trained_model_name:s2anet_conv_2x_spine/model_final.pdparams +train_infer_img_dir:./dataset/spine_coco/test/ +filename:null +## +trainer:norm_train +norm_train:tools/train.py -c test_tipc/configs/dota/s2anet_conv_2x_spine.yml -o +pact_train:tools/train.py -c test_tipc/configs/dota/s2anet_conv_2x_spine.yml --slim_config _template_pact -o +fpgm_train:tools/train.py -c test_tipc/configs/dota/s2anet_conv_2x_spine.yml --slim_config _template_fpgm -o +distill_train:null +null:null +null:null +## +===========================eval_params=========================== +eval:tools/eval.py -c test_tipc/configs/dota/s2anet_conv_2x_spine.yml -o +null:null +## +===========================infer_params=========================== +--output_dir:./output_inference +weights:https://paddledet.bj.bcebos.com/models/s2anet_conv_2x_dota.pdparams +norm_export:tools/export_model.py -c test_tipc/configs/dota/s2anet_conv_2x_spine.yml -o +pact_export:tools/export_model.py -c test_tipc/configs/dota/s2anet_conv_2x_spine.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c test_tipc/configs/dota/s2anet_conv_2x_spine.yml --slim_config _template_fpgm -o +distill_export:null +export1:null +export2:null +kl_quant_export:tools/post_quant.py -c test_tipc/configs/dota/s2anet_conv_2x_spine.yml --slim_config _template_kl_quant -o +## +infer_mode:norm +infer_quant:False +inference:./deploy/python/infer.py +--device:gpu|cpu +--enable_mkldnn:True|False +--cpu_threads:1|6 +--batch_size:1|8 +--use_tensorrt:null +--run_mode:fluid|trt_fp16 +--model_dir: +--image_dir:./dataset/spine_coco/test/ +--save_log_path:null +--run_benchmark:True +null:null \ No newline at end of file diff --git a/test_tipc/configs/face_detection/blazeface_1000e_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/face_detection/blazeface_1000e_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt new file mode 100644 index 000000000..829766393 --- /dev/null +++ b/test_tipc/configs/face_detection/blazeface_1000e_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -0,0 +1,29 @@ +===========================cpp_infer_params=========================== +model_name:blazeface_1000e +python:python3.7 +filename:blazeface_1000e +## +--output_dir:./output_inference +weights:https://paddledet.bj.bcebos.com/models/blazeface_1000e.pdparams +norm_export:tools/export_model.py -c configs/face_detection/blazeface_1000e.yml -o +quant_export:tools/export_model.py -c configs/face_detection/blazeface_1000e.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c configs/face_detection/blazeface_1000e.yml --slim_config _template_fpgm -o +distill_export:null +export1:null +export2:null +kl_quant_export:tools/post_quant.py -c configs/face_detection/blazeface_1000e.yml --slim_config _template_kl_quant -o +## +opencv_dir:default +infer_mode:norm +infer_quant:False +inference:./deploy/cpp/build/main +--device:gpu|cpu +--use_mkldnn:True|False +--cpu_threads:1|6 +--batch_size:1|8 +--use_tensorrt:null +--run_mode:fluid|trt_fp32|trt_fp16 +--model_dir: +--image_dir:./dataset/wider_face/WIDER_val/images/0--Parade/ +--run_benchmark:True +null:null \ No newline at end of file diff --git a/test_tipc/configs/face_detection/blazeface_1000e_train_infer_python.txt b/test_tipc/configs/face_detection/blazeface_1000e_train_infer_python.txt new file mode 100644 index 000000000..124843c83 --- /dev/null +++ b/test_tipc/configs/face_detection/blazeface_1000e_train_infer_python.txt @@ -0,0 +1,51 @@ +===========================train_params=========================== +model_name:blazeface_1000e +python:python3.7 +gpu_list:0|0,1 +use_gpu:True +auto_cast:null +epoch:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=1000 +save_dir:null +TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=8 +pretrain_weights:https://paddledet.bj.bcebos.com/models/blazeface_1000e.pdparams +trained_model_name:model_final.pdparams +train_infer_img_dir:./dataset/wider_face/WIDER_val/images/0--Parade/ +filename:null +## +trainer:norm_train +norm_train:tools/train.py -c configs/face_detection/blazeface_1000e.yml -o +pact_train:tools/train.py -c configs/face_detection/blazeface_1000e.yml --slim_config _template_pact -o +fpgm_train:tools/train.py -c configs/face_detection/blazeface_1000e.yml --slim_config _template_fpgm -o +distill_train:null +null:null +null:null +## +===========================eval_params=========================== +eval:tools/eval.py -c configs/face_detection/blazeface_1000e.yml -o +null:null +## +===========================infer_params=========================== +--output_dir:./output_inference +weights:https://paddledet.bj.bcebos.com/models/blazeface_1000e.pdparams +norm_export:tools/export_model.py -c configs/face_detection/blazeface_1000e.yml -o +pact_export:tools/export_model.py -c configs/face_detection/blazeface_1000e.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c configs/face_detection/blazeface_1000e.yml --slim_config _template_fpgm -o +distill_export:null +export1:null +export2:null +kl_quant_export:tools/post_quant.py -c configs/face_detection/blazeface_1000e.yml --slim_config _template_kl_quant -o +## +infer_mode:norm +infer_quant:False +inference:./deploy/python/infer.py +--device:gpu|cpu +--enable_mkldnn:True|False +--cpu_threads:1|6 +--batch_size:1|8 +--use_tensorrt:null +--run_mode:fluid|trt_fp32|trt_fp16 +--model_dir: +--image_dir:./dataset/wider_face/WIDER_val/images/0--Parade/ +--save_log_path:null +--run_benchmark:True +null:null \ No newline at end of file diff --git a/test_tipc/configs/face_detection/blazeface_1000e_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/face_detection/blazeface_1000e_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt new file mode 100644 index 000000000..9586ed007 --- /dev/null +++ b/test_tipc/configs/face_detection/blazeface_1000e_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -0,0 +1,51 @@ +===========================train_params=========================== +model_name:blazeface_1000e +python:python3.7 +gpu_list:0|0,1 +use_gpu:True +auto_cast:amp +epoch:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=1000 +save_dir:null +TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=8 +pretrain_weights:https://paddledet.bj.bcebos.com/models/blazeface_1000e.pdparams +trained_model_name:blazeface_1000e/model_final.pdparams +train_infer_img_dir:./dataset/wider_face/WIDER_val/images/0--Parade/ +filename:null +## +trainer:norm_train +norm_train:tools/train.py -c configs/face_detection/blazeface_1000e.yml -o +pact_train:tools/train.py -c configs/face_detection/blazeface_1000e.yml --slim_config _template_pact -o +fpgm_train:tools/train.py -c configs/face_detection/blazeface_1000e.yml --slim_config _template_fpgm -o +distill_train:null +null:null +null:null +## +===========================eval_params=========================== +eval:tools/eval.py -c configs/face_detection/blazeface_1000e.yml -o +null:null +## +===========================infer_params=========================== +--output_dir:./output_inference +weights:https://paddledet.bj.bcebos.com/models/blazeface_1000e.pdparams +norm_export:tools/export_model.py -c configs/face_detection/blazeface_1000e.yml -o +pact_export:tools/export_model.py -c configs/face_detection/blazeface_1000e.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c configs/face_detection/blazeface_1000e.yml --slim_config _template_fpgm -o +distill_export:null +export1:null +export2:null +kl_quant_export:tools/post_quant.py -c configs/face_detection/blazeface_1000e.yml --slim_config _template_kl_quant -o +## +infer_mode:norm +infer_quant:False +inference:./deploy/python/infer.py +--device:gpu|cpu +--enable_mkldnn:True|False +--cpu_threads:1|6 +--batch_size:1|8 +--use_tensorrt:null +--run_mode:fluid|trt_fp16 +--model_dir: +--image_dir:./dataset/wider_face/WIDER_val/images/0--Parade/ +--save_log_path:null +--run_benchmark:True +null:null \ No newline at end of file diff --git a/test_tipc/configs/face_detection/blazeface_fpn_ssh_1000e_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/face_detection/blazeface_fpn_ssh_1000e_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt new file mode 100644 index 000000000..79142a753 --- /dev/null +++ b/test_tipc/configs/face_detection/blazeface_fpn_ssh_1000e_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -0,0 +1,29 @@ +===========================cpp_infer_params=========================== +model_name:blazeface_fpn_ssh_1000e +python:python3.7 +filename:blazeface_fpn_ssh_1000e +## +--output_dir:./output_inference +weights:https://paddledet.bj.bcebos.com/models/blazeface_fpn_ssh_1000e.pdparams +norm_export:tools/export_model.py -c configs/face_detection/blazeface_fpn_ssh_1000e.yml -o +quant_export:tools/export_model.py -c configs/face_detection/blazeface_fpn_ssh_1000e.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c configs/face_detection/blazeface_fpn_ssh_1000e.yml --slim_config _template_fpgm -o +distill_export:null +export1:null +export2:null +kl_quant_export:tools/post_quant.py -c configs/face_detection/blazeface_fpn_ssh_1000e.yml --slim_config _template_kl_quant -o +## +opencv_dir:default +infer_mode:norm +infer_quant:False +inference:./deploy/cpp/build/main +--device:gpu|cpu +--use_mkldnn:True|False +--cpu_threads:1|6 +--batch_size:1|8 +--use_tensorrt:null +--run_mode:fluid|trt_fp32|trt_fp16 +--model_dir: +--image_dir:./dataset/wider_face/WIDER_val/images/0--Parade/ +--run_benchmark:True +null:null \ No newline at end of file diff --git a/test_tipc/configs/face_detection/blazeface_fpn_ssh_1000e_train_infer_python.txt b/test_tipc/configs/face_detection/blazeface_fpn_ssh_1000e_train_infer_python.txt new file mode 100644 index 000000000..bb92515c5 --- /dev/null +++ b/test_tipc/configs/face_detection/blazeface_fpn_ssh_1000e_train_infer_python.txt @@ -0,0 +1,51 @@ +===========================train_params=========================== +model_name:blazeface_fpn_ssh_1000e +python:python3.7 +gpu_list:0|0,1 +use_gpu:True +auto_cast:null +epoch:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=1000 +save_dir:null +TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=8 +pretrain_weights:https://paddledet.bj.bcebos.com/models/blazeface_fpn_ssh_1000e.pdparams +trained_model_name:model_final.pdparams +train_infer_img_dir:./dataset/wider_face/WIDER_val/images/0--Parade/ +filename:null +## +trainer:norm_train +norm_train:tools/train.py -c configs/face_detection/blazeface_fpn_ssh_1000e.yml -o +pact_train:tools/train.py -c configs/face_detection/blazeface_fpn_ssh_1000e.yml --slim_config _template_pact -o +fpgm_train:tools/train.py -c configs/face_detection/blazeface_fpn_ssh_1000e.yml --slim_config _template_fpgm -o +distill_train:null +null:null +null:null +## +===========================eval_params=========================== +eval:tools/eval.py -c configs/face_detection/blazeface_fpn_ssh_1000e.yml -o +null:null +## +===========================infer_params=========================== +--output_dir:./output_inference +weights:https://paddledet.bj.bcebos.com/models/blazeface_fpn_ssh_1000e.pdparams +norm_export:tools/export_model.py -c configs/face_detection/blazeface_fpn_ssh_1000e.yml -o +pact_export:tools/export_model.py -c configs/face_detection/blazeface_fpn_ssh_1000e.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c configs/face_detection/blazeface_fpn_ssh_1000e.yml --slim_config _template_fpgm -o +distill_export:null +export1:null +export2:null +kl_quant_export:tools/post_quant.py -c configs/face_detection/blazeface_fpn_ssh_1000e.yml --slim_config _template_kl_quant -o +## +infer_mode:norm +infer_quant:False +inference:./deploy/python/infer.py +--device:gpu|cpu +--enable_mkldnn:True|False +--cpu_threads:1|6 +--batch_size:1|8 +--use_tensorrt:null +--run_mode:fluid|trt_fp32|trt_fp16 +--model_dir: +--image_dir:./dataset/wider_face/WIDER_val/images/0--Parade/ +--save_log_path:null +--run_benchmark:True +null:null \ No newline at end of file diff --git a/test_tipc/configs/face_detection/blazeface_fpn_ssh_1000e_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/face_detection/blazeface_fpn_ssh_1000e_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt new file mode 100644 index 000000000..d09667e58 --- /dev/null +++ b/test_tipc/configs/face_detection/blazeface_fpn_ssh_1000e_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -0,0 +1,51 @@ +===========================train_params=========================== +model_name:blazeface_fpn_ssh_1000e +python:python3.7 +gpu_list:0|0,1 +use_gpu:True +auto_cast:amp +epoch:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=1000 +save_dir:null +TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=8 +pretrain_weights:https://paddledet.bj.bcebos.com/models/blazeface_fpn_ssh_1000e.pdparams +trained_model_name:blazeface_fpn_ssh_1000e/model_final.pdparams +train_infer_img_dir:./dataset/wider_face/WIDER_val/images/0--Parade/ +filename:null +## +trainer:norm_train +norm_train:tools/train.py -c configs/face_detection/blazeface_fpn_ssh_1000e.yml -o +pact_train:tools/train.py -c configs/face_detection/blazeface_fpn_ssh_1000e.yml --slim_config _template_pact -o +fpgm_train:tools/train.py -c configs/face_detection/blazeface_fpn_ssh_1000e.yml --slim_config _template_fpgm -o +distill_train:null +null:null +null:null +## +===========================eval_params=========================== +eval:tools/eval.py -c configs/face_detection/blazeface_fpn_ssh_1000e.yml -o +null:null +## +===========================infer_params=========================== +--output_dir:./output_inference +weights:https://paddledet.bj.bcebos.com/models/blazeface_fpn_ssh_1000e.pdparams +norm_export:tools/export_model.py -c configs/face_detection/blazeface_fpn_ssh_1000e.yml -o +pact_export:tools/export_model.py -c configs/face_detection/blazeface_fpn_ssh_1000e.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c configs/face_detection/blazeface_fpn_ssh_1000e.yml --slim_config _template_fpgm -o +distill_export:null +export1:null +export2:null +kl_quant_export:tools/post_quant.py -c configs/face_detection/blazeface_fpn_ssh_1000e.yml --slim_config _template_kl_quant -o +## +infer_mode:norm +infer_quant:False +inference:./deploy/python/infer.py +--device:gpu|cpu +--enable_mkldnn:True|False +--cpu_threads:1|6 +--batch_size:1|8 +--use_tensorrt:null +--run_mode:fluid|trt_fp16 +--model_dir: +--image_dir:./dataset/wider_face/WIDER_val/images/0--Parade/ +--save_log_path:null +--run_benchmark:True +null:null \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index 2c11c0ac3..bc0262a0c 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -26,4 +26,4 @@ inference:./deploy/cpp/build/main --model_dir: --image_dir:./dataset/coco/test2017/ --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_1x_coco_train_infer_python.txt index a67e9deee..b10141d94 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_1x_coco_train_infer_python.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 9c03d2b90..2d00cae15 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index a1f877da1..a06fbc00d 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -4,7 +4,7 @@ python:python3.7 filename:faster_rcnn_r101_fpn_1x_coco ## --output_dir:./output_inference -weights:https://paddledet.bj.bcebos.com/models/faster_rcnn_r101_fpn_1x_coco.pdparams +weights:https://paddledet.bj.bcebos.com/models/faster_rcnn_r101_fpn_2x_coco.pdparams norm_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco.yml -o quant_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco.yml --slim_config _template_pact -o fpgm_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco.yml --slim_config _template_fpgm -o @@ -26,4 +26,4 @@ inference:./deploy/cpp/build/main --model_dir: --image_dir:./dataset/coco/test2017/ --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_train_infer_python.txt index c018669c1..bcf36f97d 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_train_infer_python.txt @@ -7,7 +7,7 @@ auto_cast:null epoch:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=12 save_dir:null TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=1 -pretrain_weights:https://paddledet.bj.bcebos.com/models/faster_rcnn_r101_fpn_1x_coco.pdparams +pretrain_weights:https://paddledet.bj.bcebos.com/models/faster_rcnn_r101_fpn_2x_coco.pdparams trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ filename:null @@ -26,7 +26,7 @@ null:null ## ===========================infer_params=========================== --output_dir:./output_inference -weights:https://paddledet.bj.bcebos.com/models/faster_rcnn_r101_fpn_1x_coco.pdparams +weights:https://paddledet.bj.bcebos.com/models/faster_rcnn_r101_fpn_2x_coco.pdparams norm_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco.yml -o pact_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco.yml --slim_config _template_pact -o fpgm_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco.yml --slim_config _template_fpgm -o @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 5a78be3a5..755b63a8b 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -7,7 +7,7 @@ auto_cast:amp epoch:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=12 save_dir:null TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=1 -pretrain_weights:https://paddledet.bj.bcebos.com/models/faster_rcnn_r101_fpn_1x_coco.pdparams +pretrain_weights:https://paddledet.bj.bcebos.com/models/faster_rcnn_r101_fpn_2x_coco.pdparams trained_model_name:faster_rcnn_r101_fpn_1x_coco/model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ filename:null @@ -26,7 +26,7 @@ null:null ## ===========================infer_params=========================== --output_dir:./output_inference -weights:https://paddledet.bj.bcebos.com/models/faster_rcnn_r101_fpn_1x_coco.pdparams +weights:https://paddledet.bj.bcebos.com/models/faster_rcnn_r101_fpn_2x_coco.pdparams norm_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco.yml -o pact_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco.yml --slim_config _template_pact -o fpgm_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco.yml --slim_config _template_fpgm -o @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index 0035fd912..a5c37d7e5 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -26,4 +26,4 @@ inference:./deploy/cpp/build/main --model_dir: --image_dir:./dataset/coco/test2017/ --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt index 8129670fd..53f4b23a9 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 4d2fc2210..e9bab4d47 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index 40f226447..f325c0c97 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -26,4 +26,4 @@ inference:./deploy/cpp/build/main --model_dir: --image_dir:./dataset/coco/test2017/ --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco_train_infer_python.txt index 29d37fc29..74f8bb423 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco_train_infer_python.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 5bf1d49f2..bac2e78aa 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index 2d3d54d54..46287f8b8 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -26,4 +26,4 @@ inference:./deploy/cpp/build/main --model_dir: --image_dir:./dataset/coco/test2017/ --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco_train_infer_python.txt index 01a622f48..499daea42 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco_train_infer_python.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 56a9ac380..844e5cacd 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index 579b81970..1fac1ba3f 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -26,4 +26,4 @@ inference:./deploy/cpp/build/main --model_dir: --image_dir:./dataset/coco/test2017/ --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_1x_coco_train_infer_python.txt index c37e497d5..2f7035bf0 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_1x_coco_train_infer_python.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 01f1121aa..653de7c5d 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index 9b9489d08..f4f436485 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -26,4 +26,4 @@ inference:./deploy/cpp/build/main --model_dir: --image_dir:./dataset/coco/test2017/ --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco_train_infer_python.txt index 85ccca940..8e9e841e0 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco_train_infer_python.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 7d52cf5e0..c798610fa 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index af2d871f0..77bebb1f0 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -26,4 +26,4 @@ inference:./deploy/cpp/build/main --model_dir: --image_dir:./dataset/coco/test2017/ --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco_train_infer_python.txt index 378e3dc53..22c169c99 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco_train_infer_python.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 2a6e623f8..7f85facea 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index ce38015cf..53baebba2 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -26,4 +26,4 @@ inference:./deploy/cpp/build/main --model_dir: --image_dir:./dataset/coco/test2017/ --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt index 808999ad1..a0c6e64e5 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 901566ca2..dc2483f59 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index 73583ca09..29fcc1b63 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -26,4 +26,4 @@ inference:./deploy/cpp/build/main --model_dir: --image_dir:./dataset/coco/test2017/ --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco_train_infer_python.txt index c3827db4b..aae1c3dfa 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco_train_infer_python.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index f9b24c34c..8ded89c00 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index fa03514e6..a448a8187 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -26,4 +26,4 @@ inference:./deploy/cpp/build/main --model_dir: --image_dir:./dataset/coco/test2017/ --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt index db9508873..1e560b78e 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index f0be1d47a..b721de6c4 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/fcos/fcos_r50_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/fcos/fcos_r50_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index 77d0508b1..6958e88bd 100644 --- a/test_tipc/configs/fcos/fcos_r50_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/fcos/fcos_r50_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -26,4 +26,4 @@ inference:./deploy/cpp/build/main --model_dir: --image_dir:./dataset/coco/test2017/ --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/fcos/fcos_r50_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/fcos/fcos_r50_fpn_1x_coco_train_infer_python.txt index 185d23dbf..1ce27f461 100644 --- a/test_tipc/configs/fcos/fcos_r50_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/fcos/fcos_r50_fpn_1x_coco_train_infer_python.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/fcos/fcos_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/fcos/fcos_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 2ed634265..9a4ad4ce0 100644 --- a/test_tipc/configs/fcos/fcos_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/fcos/fcos_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/keypoint/dark_hrnet_w32_256x192_train_infer_python.txt b/test_tipc/configs/keypoint/dark_hrnet_w32_256x192_train_infer_python.txt index 471ca1a5b..501c47e85 100644 --- a/test_tipc/configs/keypoint/dark_hrnet_w32_256x192_train_infer_python.txt +++ b/test_tipc/configs/keypoint/dark_hrnet_w32_256x192_train_infer_python.txt @@ -37,7 +37,7 @@ kl_quant_export:tools/post_quant.py -c configs/keypoint/hrnet/dark_hrnet_w32_256 ## infer_mode:norm infer_quant:False -inference:./deploy/python/infer.py +inference:./deploy/python/keypoint_infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 diff --git a/test_tipc/configs/keypoint/dark_hrnet_w32_256x192_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/keypoint/dark_hrnet_w32_256x192_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index d652b36c1..576b38e2d 100644 --- a/test_tipc/configs/keypoint/dark_hrnet_w32_256x192_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/keypoint/dark_hrnet_w32_256x192_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -37,7 +37,7 @@ kl_quant_export:tools/post_quant.py -c configs/keypoint/hrnet/dark_hrnet_w32_256 ## infer_mode:norm infer_quant:False -inference:./deploy/python/infer.py +inference:./deploy/python/keypoint_infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 diff --git a/test_tipc/configs/keypoint/dark_hrnet_w48_256x192_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/keypoint/dark_hrnet_w48_256x192_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index 80ee22e76..5e19aade2 100644 --- a/test_tipc/configs/keypoint/dark_hrnet_w48_256x192_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/keypoint/dark_hrnet_w48_256x192_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -4,7 +4,7 @@ python:python3.7 filename:dark_hrnet_w48_256x192 ## --output_dir:./output_inference -weights:https://paddledet.bj.bcebos.com/models/keypoint/dark_hrnet_w48_256x192.pdparams +weights:null norm_export:tools/export_model.py -c configs/keypoint/hrnet/dark_hrnet_w48_256x192.yml -o quant_export:tools/export_model.py -c configs/keypoint/hrnet/dark_hrnet_w48_256x192.yml --slim_config _template_pact -o fpgm_export:tools/export_model.py -c configs/keypoint/hrnet/dark_hrnet_w48_256x192.yml --slim_config _template_fpgm -o diff --git a/test_tipc/configs/keypoint/dark_hrnet_w48_256x192_train_infer_python.txt b/test_tipc/configs/keypoint/dark_hrnet_w48_256x192_train_infer_python.txt index ca3f6556c..ff4c88675 100644 --- a/test_tipc/configs/keypoint/dark_hrnet_w48_256x192_train_infer_python.txt +++ b/test_tipc/configs/keypoint/dark_hrnet_w48_256x192_train_infer_python.txt @@ -7,7 +7,7 @@ auto_cast:null epoch:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=210 save_dir:null TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=64 -pretrain_weights:https://paddledet.bj.bcebos.com/models/keypoint/dark_hrnet_w48_256x192.pdparams +pretrain_weights:null trained_model_name:model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ filename:null @@ -26,7 +26,7 @@ null:null ## ===========================infer_params=========================== --output_dir:./output_inference -weights:https://paddledet.bj.bcebos.com/models/keypoint/dark_hrnet_w48_256x192.pdparams +weights:null norm_export:tools/export_model.py -c configs/keypoint/hrnet/dark_hrnet_w48_256x192.yml -o pact_export:tools/export_model.py -c configs/keypoint/hrnet/dark_hrnet_w48_256x192.yml --slim_config _template_pact -o fpgm_export:tools/export_model.py -c configs/keypoint/hrnet/dark_hrnet_w48_256x192.yml --slim_config _template_fpgm -o @@ -37,7 +37,7 @@ kl_quant_export:tools/post_quant.py -c configs/keypoint/hrnet/dark_hrnet_w48_256 ## infer_mode:norm infer_quant:False -inference:./deploy/python/infer.py +inference:./deploy/python/keypoint_infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 diff --git a/test_tipc/configs/keypoint/dark_hrnet_w48_256x192_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/keypoint/dark_hrnet_w48_256x192_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 6ffcdba6a..b1c05ed49 100644 --- a/test_tipc/configs/keypoint/dark_hrnet_w48_256x192_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/keypoint/dark_hrnet_w48_256x192_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -7,7 +7,7 @@ auto_cast:amp epoch:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=210 save_dir:null TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=64 -pretrain_weights:https://paddledet.bj.bcebos.com/models/keypoint/dark_hrnet_w48_256x192.pdparams +pretrain_weights:null trained_model_name:dark_hrnet_w48_256x192/model_final.pdparams train_infer_img_dir:./dataset/coco/test2017/ filename:null @@ -26,7 +26,7 @@ null:null ## ===========================infer_params=========================== --output_dir:./output_inference -weights:https://paddledet.bj.bcebos.com/models/keypoint/dark_hrnet_w48_256x192.pdparams +weights:null norm_export:tools/export_model.py -c configs/keypoint/hrnet/dark_hrnet_w48_256x192.yml -o pact_export:tools/export_model.py -c configs/keypoint/hrnet/dark_hrnet_w48_256x192.yml --slim_config _template_pact -o fpgm_export:tools/export_model.py -c configs/keypoint/hrnet/dark_hrnet_w48_256x192.yml --slim_config _template_fpgm -o @@ -37,7 +37,7 @@ kl_quant_export:tools/post_quant.py -c configs/keypoint/hrnet/dark_hrnet_w48_256 ## infer_mode:norm infer_quant:False -inference:./deploy/python/infer.py +inference:./deploy/python/keypoint_infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 diff --git a/test_tipc/configs/keypoint/higherhrnet_hrnet_w32_512_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/keypoint/higherhrnet_hrnet_w32_512_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index 854ee6ca0..f91f3ddfe 100644 --- a/test_tipc/configs/keypoint/higherhrnet_hrnet_w32_512_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/keypoint/higherhrnet_hrnet_w32_512_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -26,4 +26,4 @@ inference:./deploy/cpp/build/main --model_dir: --image_dir:./dataset/coco/test2017/ --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/keypoint/higherhrnet_hrnet_w32_512_train_infer_python.txt b/test_tipc/configs/keypoint/higherhrnet_hrnet_w32_512_train_infer_python.txt index 5b5d85c07..eee491ee0 100644 --- a/test_tipc/configs/keypoint/higherhrnet_hrnet_w32_512_train_infer_python.txt +++ b/test_tipc/configs/keypoint/higherhrnet_hrnet_w32_512_train_infer_python.txt @@ -37,7 +37,7 @@ kl_quant_export:tools/post_quant.py -c configs/keypoint/higherhrnet/higherhrnet_ ## infer_mode:norm infer_quant:False -inference:./deploy/python/infer.py +inference:./deploy/python/keypoint_infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/keypoint/higherhrnet_hrnet_w32_512_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/keypoint/higherhrnet_hrnet_w32_512_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 70364dc11..0e6ea46d3 100644 --- a/test_tipc/configs/keypoint/higherhrnet_hrnet_w32_512_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/keypoint/higherhrnet_hrnet_w32_512_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -37,7 +37,7 @@ kl_quant_export:tools/post_quant.py -c configs/keypoint/higherhrnet/higherhrnet_ ## infer_mode:norm infer_quant:False -inference:./deploy/python/infer.py +inference:./deploy/python/keypoint_infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/keypoint/hrnet_w32_256x192_train_infer_python.txt b/test_tipc/configs/keypoint/hrnet_w32_256x192_train_infer_python.txt index 640606035..ab09671f7 100644 --- a/test_tipc/configs/keypoint/hrnet_w32_256x192_train_infer_python.txt +++ b/test_tipc/configs/keypoint/hrnet_w32_256x192_train_infer_python.txt @@ -37,7 +37,7 @@ kl_quant_export:tools/post_quant.py -c configs/keypoint/hrnet/hrnet_w32_256x192. ## infer_mode:norm infer_quant:False -inference:./deploy/python/infer.py +inference:./deploy/python/keypoint_infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 diff --git a/test_tipc/configs/keypoint/hrnet_w32_256x192_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/keypoint/hrnet_w32_256x192_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 54bba3b9e..68e5d4253 100644 --- a/test_tipc/configs/keypoint/hrnet_w32_256x192_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/keypoint/hrnet_w32_256x192_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -37,7 +37,7 @@ kl_quant_export:tools/post_quant.py -c configs/keypoint/hrnet/hrnet_w32_256x192. ## infer_mode:norm infer_quant:False -inference:./deploy/python/infer.py +inference:./deploy/python/keypoint_infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 diff --git a/test_tipc/configs/keypoint/tinypose_128x96.yml b/test_tipc/configs/keypoint/tinypose_128x96.yml new file mode 100644 index 000000000..338d9793c --- /dev/null +++ b/test_tipc/configs/keypoint/tinypose_128x96.yml @@ -0,0 +1,147 @@ +use_gpu: true +log_iter: 5 +save_dir: output +snapshot_epoch: 10 +weights: output/tinypose_128x96/model_final +epoch: 420 +num_joints: &num_joints 17 +pixel_std: &pixel_std 200 +metric: KeyPointTopDownCOCOEval +num_classes: 1 +train_height: &train_height 128 +train_width: &train_width 96 +trainsize: &trainsize [*train_width, *train_height] +hmsize: &hmsize [24, 32] +flip_perm: &flip_perm [[1, 2], [3, 4], [5, 6], [7, 8], [9, 10], [11, 12], [13, 14], [15, 16]] + + +#####model +architecture: TopDownHRNet + +TopDownHRNet: + backbone: LiteHRNet + post_process: HRNetPostProcess + flip_perm: *flip_perm + num_joints: *num_joints + width: &width 40 + loss: KeyPointMSELoss + use_dark: true + +LiteHRNet: + network_type: wider_naive + freeze_at: -1 + freeze_norm: false + return_idx: [0] + +KeyPointMSELoss: + use_target_weight: true + loss_scale: 1.0 + +#####optimizer +LearningRate: + base_lr: 0.008 + schedulers: + - !PiecewiseDecay + milestones: [380, 410] + gamma: 0.1 + - !LinearWarmup + start_factor: 0.001 + steps: 500 + +OptimizerBuilder: + optimizer: + type: Adam + regularizer: + factor: 0.0 + type: L2 + + +#####data +TrainDataset: + !KeypointTopDownCocoDataset + image_dir: train2017 + anno_path: annotations/person_keypoints_train2017.json + dataset_dir: dataset/coco + num_joints: *num_joints + trainsize: *trainsize + pixel_std: *pixel_std + use_gt_bbox: True + + +EvalDataset: + !KeypointTopDownCocoDataset + image_dir: val2017 + anno_path: annotations/person_keypoints_val2017.json + dataset_dir: dataset/coco + num_joints: *num_joints + trainsize: *trainsize + pixel_std: *pixel_std + use_gt_bbox: True + image_thre: 0.5 + +TestDataset: + !ImageFolder + anno_path: dataset/coco/keypoint_imagelist.txt + +worker_num: 2 +global_mean: &global_mean [0.485, 0.456, 0.406] +global_std: &global_std [0.229, 0.224, 0.225] +TrainReader: + sample_transforms: + - RandomFlipHalfBodyTransform: + scale: 0.25 + rot: 30 + num_joints_half_body: 8 + prob_half_body: 0.3 + pixel_std: *pixel_std + trainsize: *trainsize + upper_body_ids: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10] + flip_pairs: *flip_perm + - AugmentationbyInformantionDropping: + prob_cutout: 0.5 + offset_factor: 0.05 + num_patch: 1 + trainsize: *trainsize + - TopDownAffine: + trainsize: *trainsize + use_udp: true + - ToHeatmapsTopDown_DARK: + hmsize: *hmsize + sigma: 1 + batch_transforms: + - NormalizeImage: + mean: *global_mean + std: *global_std + is_scale: true + - Permute: {} + batch_size: 512 + shuffle: true + drop_last: false + +EvalReader: + sample_transforms: + - TopDownAffine: + trainsize: *trainsize + use_udp: true + batch_transforms: + - NormalizeImage: + mean: *global_mean + std: *global_std + is_scale: true + - Permute: {} + batch_size: 16 + +TestReader: + inputs_def: + image_shape: [3, *train_height, *train_width] + sample_transforms: + - Decode: {} + - TopDownEvalAffine: + trainsize: *trainsize + - NormalizeImage: + mean: *global_mean + std: *global_std + is_scale: true + - Permute: {} + batch_size: 1 + fuse_normalize: false diff --git a/test_tipc/configs/keypoint/tinypose_128x96_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/keypoint/tinypose_128x96_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index a5a2beab9..469718dce 100644 --- a/test_tipc/configs/keypoint/tinypose_128x96_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/keypoint/tinypose_128x96_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -5,13 +5,13 @@ filename:tinypose_128x96 ## --output_dir:./output_inference weights:https://paddledet.bj.bcebos.com/models/keypoint/tinypose_128x96.pdparams -norm_export:tools/export_model.py -c configs/keypoint/tiny_pose/tinypose_128x96.yml -o -quant_export:tools/export_model.py -c configs/keypoint/tiny_pose/tinypose_128x96.yml --slim_config _template_pact -o -fpgm_export:tools/export_model.py -c configs/keypoint/tiny_pose/tinypose_128x96.yml --slim_config _template_fpgm -o +norm_export:tools/export_model.py -c test_tipc/configs/keypoint/tinypose_128x96.yml -o +quant_export:tools/export_model.py -c test_tipc/configs/keypoint/tinypose_128x96.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c test_tipc/configs/keypoint/tinypose_128x96.yml --slim_config _template_fpgm -o distill_export:null export1:null export2:null -kl_quant_export:tools/post_quant.py -c configs/keypoint/tiny_pose/tinypose_128x96.yml --slim_config _template_kl_quant -o +kl_quant_export:tools/post_quant.py -c test_tipc/configs/keypoint/tinypose_128x96.yml --slim_config _template_kl_quant -o ## opencv_dir:default infer_mode:norm diff --git a/test_tipc/configs/keypoint/tinypose_128x96_train_infer_python.txt b/test_tipc/configs/keypoint/tinypose_128x96_train_infer_python.txt index cf2b21bc6..8a6513c8b 100644 --- a/test_tipc/configs/keypoint/tinypose_128x96_train_infer_python.txt +++ b/test_tipc/configs/keypoint/tinypose_128x96_train_infer_python.txt @@ -13,31 +13,31 @@ train_infer_img_dir:./dataset/coco/test2017/ filename:null ## trainer:norm_train -norm_train:tools/train.py -c configs/keypoint/tiny_pose/tinypose_128x96.yml -o -pact_train:tools/train.py -c configs/keypoint/tiny_pose/tinypose_128x96.yml --slim_config _template_pact -o -fpgm_train:tools/train.py -c configs/keypoint/tiny_pose/tinypose_128x96.yml --slim_config _template_fpgm -o +norm_train:tools/train.py -c test_tipc/configs/keypoint/tinypose_128x96.yml -o +pact_train:tools/train.py -c test_tipc/configs/keypoint/tinypose_128x96.yml --slim_config _template_pact -o +fpgm_train:tools/train.py -c test_tipc/configs/keypoint/tinypose_128x96.yml --slim_config _template_fpgm -o distill_train:null null:null null:null ## ===========================eval_params=========================== -eval:tools/eval.py -c configs/keypoint/tiny_pose/tinypose_128x96.yml -o +eval:tools/eval.py -c test_tipc/configs/keypoint/tinypose_128x96.yml -o null:null ## ===========================infer_params=========================== --output_dir:./output_inference weights:https://paddledet.bj.bcebos.com/models/keypoint/tinypose_128x96.pdparams -norm_export:tools/export_model.py -c configs/keypoint/tiny_pose/tinypose_128x96.yml -o -pact_export:tools/export_model.py -c configs/keypoint/tiny_pose/tinypose_128x96.yml --slim_config _template_pact -o -fpgm_export:tools/export_model.py -c configs/keypoint/tiny_pose/tinypose_128x96.yml --slim_config _template_fpgm -o +norm_export:tools/export_model.py -c test_tipc/configs/keypoint/tinypose_128x96.yml -o +pact_export:tools/export_model.py -c test_tipc/configs/keypoint/tinypose_128x96.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c test_tipc/configs/keypoint/tinypose_128x96.yml --slim_config _template_fpgm -o distill_export:null export1:null export2:null -kl_quant_export:tools/post_quant.py -c configs/keypoint/tiny_pose/tinypose_128x96.yml --slim_config _template_kl_quant -o +kl_quant_export:tools/post_quant.py -c test_tipc/configs/keypoint/tinypose_128x96.yml --slim_config _template_kl_quant -o ## infer_mode:norm infer_quant:False -inference:./deploy/python/infer.py +inference:./deploy/python/keypoint_infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 diff --git a/test_tipc/configs/keypoint/tinypose_128x96_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/keypoint/tinypose_128x96_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 877d068dc..11a67ccaa 100644 --- a/test_tipc/configs/keypoint/tinypose_128x96_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/keypoint/tinypose_128x96_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -13,31 +13,31 @@ train_infer_img_dir:./dataset/coco/test2017/ filename:null ## trainer:norm_train -norm_train:tools/train.py -c configs/keypoint/tiny_pose/tinypose_128x96.yml -o -pact_train:tools/train.py -c configs/keypoint/tiny_pose/tinypose_128x96.yml --slim_config _template_pact -o -fpgm_train:tools/train.py -c configs/keypoint/tiny_pose/tinypose_128x96.yml --slim_config _template_fpgm -o +norm_train:tools/train.py -c test_tipc/configs/keypoint/tinypose_128x96.yml -o +pact_train:tools/train.py -c test_tipc/configs/keypoint/tinypose_128x96.yml --slim_config _template_pact -o +fpgm_train:tools/train.py -c test_tipc/configs/keypoint/tinypose_128x96.yml --slim_config _template_fpgm -o distill_train:null null:null null:null ## ===========================eval_params=========================== -eval:tools/eval.py -c configs/keypoint/tiny_pose/tinypose_128x96.yml -o +eval:tools/eval.py -c test_tipc/configs/keypoint/tinypose_128x96.yml -o null:null ## ===========================infer_params=========================== --output_dir:./output_inference weights:https://paddledet.bj.bcebos.com/models/keypoint/tinypose_128x96.pdparams -norm_export:tools/export_model.py -c configs/keypoint/tiny_pose/tinypose_128x96.yml -o -pact_export:tools/export_model.py -c configs/keypoint/tiny_pose/tinypose_128x96.yml --slim_config _template_pact -o -fpgm_export:tools/export_model.py -c configs/keypoint/tiny_pose/tinypose_128x96.yml --slim_config _template_fpgm -o +norm_export:tools/export_model.py -c test_tipc/configs/keypoint/tinypose_128x96.yml -o +pact_export:tools/export_model.py -c test_tipc/configs/keypoint/tinypose_128x96.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c test_tipc/configs/keypoint/tinypose_128x96.yml --slim_config _template_fpgm -o distill_export:null export1:null export2:null -kl_quant_export:tools/post_quant.py -c configs/keypoint/tiny_pose/tinypose_128x96.yml --slim_config _template_kl_quant -o +kl_quant_export:tools/post_quant.py -c test_tipc/configs/keypoint/tinypose_128x96.yml --slim_config _template_kl_quant -o ## infer_mode:norm infer_quant:False -inference:./deploy/python/infer.py +inference:./deploy/python/keypoint_infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index 07fc66c79..7631e34b8 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -26,4 +26,4 @@ inference:./deploy/cpp/build/main --model_dir: --image_dir:./dataset/coco/test2017/ --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco_train_infer_python.txt index 695bfda79..637207ef0 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco_train_infer_python.txt @@ -41,11 +41,11 @@ inference:./deploy/python/infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp32|trt_fp16 --model_dir: --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 80803495a..39387130c 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index 415494e3d..89ffd5ebd 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -26,4 +26,4 @@ inference:./deploy/cpp/build/main --model_dir: --image_dir:./dataset/coco/test2017/ --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt index bc3b9cccf..1014d1f1d 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt @@ -41,11 +41,11 @@ inference:./deploy/python/infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp32|trt_fp16 --model_dir: --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 9b76ccaf1..cd8009610 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index a6067d69d..7a5890340 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -26,4 +26,4 @@ inference:./deploy/cpp/build/main --model_dir: --image_dir:./dataset/coco/test2017/ --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_1x_coco_train_infer_python.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_1x_coco_train_infer_python.txt index effa300f3..39814f3e7 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_1x_coco_train_infer_python.txt @@ -41,11 +41,11 @@ inference:./deploy/python/infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp32|trt_fp16 --model_dir: --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index d11dbe204..b3afb274c 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index a6fa63b4e..8cd1375bf 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -26,4 +26,4 @@ inference:./deploy/cpp/build/main --model_dir: --image_dir:./dataset/coco/test2017/ --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt index 0e606873f..a2fef0698 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt @@ -41,11 +41,11 @@ inference:./deploy/python/infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp32|trt_fp16 --model_dir: --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 364ac88e5..e29cad76c 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index 9ac863ad1..0794af25e 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -26,4 +26,4 @@ inference:./deploy/cpp/build/main --model_dir: --image_dir:./dataset/coco/test2017/ --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt index 52c2d3440..c02b26d32 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt @@ -41,11 +41,11 @@ inference:./deploy/python/infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp32|trt_fp16 --model_dir: --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index bfb332c25..91b2663cf 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index 9954f6cc4..87f118bd3 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -26,4 +26,4 @@ inference:./deploy/cpp/build/main --model_dir: --image_dir:./dataset/coco/test2017/ --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt index 77daea639..949259f8b 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt @@ -41,11 +41,11 @@ inference:./deploy/python/infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp32|trt_fp16 --model_dir: --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index bd6d80fb3..f055782f1 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -48,4 +48,4 @@ inference:./deploy/python/infer.py --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/mot/fairmot_dla34_30e_576x320_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/mot/fairmot_dla34_30e_576x320_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt new file mode 100644 index 000000000..d26e86428 --- /dev/null +++ b/test_tipc/configs/mot/fairmot_dla34_30e_576x320_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -0,0 +1,29 @@ +===========================cpp_infer_params=========================== +model_name:fairmot_dla34_30e_576x320 +python:python3.7 +filename:fairmot_dla34_30e_576x320 +## +--output_dir:./output_inference +weights:https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_576x320.pdparams +norm_export:tools/export_model.py -c configs/mot/fairmot/fairmot_dla34_30e_576x320.yml -o +quant_export:tools/export_model.py -c configs/mot/fairmot/fairmot_dla34_30e_576x320.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c configs/mot/fairmot/fairmot_dla34_30e_576x320.yml --slim_config _template_fpgm -o +distill_export:null +export1:null +export2:null +kl_quant_export:tools/post_quant.py -c configs/mot/fairmot/fairmot_dla34_30e_576x320.yml --slim_config _template_kl_quant -o +## +opencv_dir:default +infer_mode:norm +infer_quant:False +inference:./deploy/cpp/build/main +--device:gpu|cpu +--use_mkldnn:True|False +--cpu_threads:1|6 +--batch_size:1|8 +--use_tensorrt:null +--run_mode:fluid|trt_fp32|trt_fp16 +--model_dir: +--video_file:./dataset/mot/test.mp4 +--run_benchmark:True +null:null \ No newline at end of file diff --git a/test_tipc/configs/mot/fairmot_dla34_30e_576x320_train_infer_python.txt b/test_tipc/configs/mot/fairmot_dla34_30e_576x320_train_infer_python.txt new file mode 100644 index 000000000..efdf9b5a1 --- /dev/null +++ b/test_tipc/configs/mot/fairmot_dla34_30e_576x320_train_infer_python.txt @@ -0,0 +1,51 @@ +===========================train_params=========================== +model_name:fairmot_dla34_30e_576x320 +python:python3.7 +gpu_list:0|0,1 +use_gpu:True +auto_cast:null +epoch:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=30 +save_dir:null +TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=6 +pretrain_weights:https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_576x320.pdparams +trained_model_name:model_final.pdparams +train_infer_img_dir:./dataset/mot/test.mp4 +filename:null +## +trainer:norm_train +norm_train:tools/train.py -c configs/mot/fairmot/fairmot_dla34_30e_576x320.yml -o +pact_train:tools/train.py -c configs/mot/fairmot/fairmot_dla34_30e_576x320.yml --slim_config _template_pact -o +fpgm_train:tools/train.py -c configs/mot/fairmot/fairmot_dla34_30e_576x320.yml --slim_config _template_fpgm -o +distill_train:null +null:null +null:null +## +===========================eval_params=========================== +eval:tools/eval_mot.py -c configs/mot/fairmot/fairmot_dla34_30e_576x320.yml -o +null:null +## +===========================infer_params=========================== +--output_dir:./output_inference +weights:https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_576x320.pdparams +norm_export:tools/export_model.py -c configs/mot/fairmot/fairmot_dla34_30e_576x320.yml -o +pact_export:tools/export_model.py -c configs/mot/fairmot/fairmot_dla34_30e_576x320.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c configs/mot/fairmot/fairmot_dla34_30e_576x320.yml --slim_config _template_fpgm -o +distill_export:null +export1:null +export2:null +kl_quant_export:tools/post_quant.py -c configs/mot/fairmot/fairmot_dla34_30e_576x320.yml --slim_config _template_kl_quant -o +## +infer_mode:norm +infer_quant:False +inference:./deploy/pptracking/python/mot_jde_infer.py +--device:gpu|cpu +--enable_mkldnn:True|False +--cpu_threads:1|6 +--batch_size:1|8 +--use_tensorrt:null +--run_mode:fluid|trt_fp32|trt_fp16 +--model_dir: +--video_file:./dataset/mot/test.mp4 +--save_log_path:null +--run_benchmark:False +null:null \ No newline at end of file diff --git a/test_tipc/configs/mot/fairmot_dla34_30e_576x320_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/mot/fairmot_dla34_30e_576x320_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt new file mode 100644 index 000000000..2875baa0b --- /dev/null +++ b/test_tipc/configs/mot/fairmot_dla34_30e_576x320_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -0,0 +1,51 @@ +===========================train_params=========================== +model_name:fairmot_dla34_30e_576x320 +python:python3.7 +gpu_list:0|0,1 +use_gpu:True +auto_cast:amp +epoch:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=30 +save_dir:null +TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=6 +pretrain_weights:https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_576x320.pdparams +trained_model_name:fairmot_dla34_30e_576x320/model_final.pdparams +train_infer_img_dir:./dataset/mot/test.mp4 +filename:null +## +trainer:norm_train +norm_train:tools/train.py -c configs/mot/fairmot/fairmot_dla34_30e_576x320.yml -o +pact_train:tools/train.py -c configs/mot/fairmot/fairmot_dla34_30e_576x320.yml --slim_config _template_pact -o +fpgm_train:tools/train.py -c configs/mot/fairmot/fairmot_dla34_30e_576x320.yml --slim_config _template_fpgm -o +distill_train:null +null:null +null:null +## +===========================eval_params=========================== +eval:tools/eval_mot.py -c configs/mot/fairmot/fairmot_dla34_30e_576x320.yml -o +null:null +## +===========================infer_params=========================== +--output_dir:./output_inference +weights:https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_576x320.pdparams +norm_export:tools/export_model.py -c configs/mot/fairmot/fairmot_dla34_30e_576x320.yml -o +pact_export:tools/export_model.py -c configs/mot/fairmot/fairmot_dla34_30e_576x320.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c configs/mot/fairmot/fairmot_dla34_30e_576x320.yml --slim_config _template_fpgm -o +distill_export:null +export1:null +export2:null +kl_quant_export:tools/post_quant.py -c configs/mot/fairmot/fairmot_dla34_30e_576x320.yml --slim_config _template_kl_quant -o +## +infer_mode:norm +infer_quant:False +inference:./deploy/pptracking/python/mot_jde_infer.py +--device:gpu|cpu +--enable_mkldnn:True|False +--cpu_threads:1|6 +--batch_size:1|8 +--use_tensorrt:null +--run_mode:fluid|trt_fp16 +--model_dir: +--video_file:./dataset/mot/test.mp4 +--save_log_path:null +--run_benchmark:False +null:null \ No newline at end of file diff --git a/test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml b/test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml new file mode 100644 index 000000000..b010c2dc1 --- /dev/null +++ b/test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml @@ -0,0 +1,43 @@ +_BASE_: [ + '../../../configs/datasets/mot.yml', + '../../../configs/runtime.yml', + '../../../configs/mot/fairmot/_base_/optimizer_30e_momentum.yml', + '../../../configs/mot/fairmot/_base_/fairmot_hrnetv2_w18_dlafpn.yml', + '../../../configs/mot/fairmot/_base_/fairmot_reader_576x320.yml', +] + +norm_type: sync_bn +use_ema: true +ema_decay: 0.9998 + +# add crowdhuman +TrainDataset: + !MOTDataSet + dataset_dir: dataset/mot + image_lists: ['mot17.train', 'caltech.all', 'cuhksysu.train', 'prw.train', 'citypersons.train', 'eth.train'] + data_fields: ['image', 'gt_bbox', 'gt_class', 'gt_ide'] + +worker_num: 4 +TrainReader: + inputs_def: + image_shape: [3, 320, 576] + sample_transforms: + - Decode: {} + - RGBReverse: {} + - AugmentHSV: {} + - LetterBoxResize: {target_size: [320, 576]} + - MOTRandomAffine: {reject_outside: False} + - RandomFlip: {} + - BboxXYXY2XYWH: {} + - NormalizeBox: {} + - NormalizeImage: {mean: [0, 0, 0], std: [1, 1, 1]} + - RGBReverse: {} + - Permute: {} + batch_transforms: + - Gt2FairMOTTarget: {} + batch_size: 4 + shuffle: True + drop_last: True + use_shared_memory: True + +weights: output/fairmot_hrnetv2_w18_dlafpn_30e_576x320/model_final diff --git a/test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt new file mode 100644 index 000000000..1972cf0c4 --- /dev/null +++ b/test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -0,0 +1,29 @@ +===========================cpp_infer_params=========================== +model_name:fairmot_hrnetv2_w18_dlafpn_30e_576x320 +python:python3.7 +filename:fairmot_hrnetv2_w18_dlafpn_30e_576x320 +## +--output_dir:./output_inference +weights:https://paddledet.bj.bcebos.com/models/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.pdparams +norm_export:tools/export_model.py -c test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml -o +quant_export:tools/export_model.py -c test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml --slim_config _template_fpgm -o +distill_export:null +export1:null +export2:null +kl_quant_export:tools/post_quant.py -c test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml --slim_config _template_kl_quant -o +## +opencv_dir:default +infer_mode:norm +infer_quant:False +inference:./deploy/cpp/build/main +--device:gpu|cpu +--use_mkldnn:True|False +--cpu_threads:1|6 +--batch_size:1|8 +--use_tensorrt:null +--run_mode:fluid|trt_fp32|trt_fp16 +--model_dir: +--video_file:./dataset/mot/test.mp4 +--run_benchmark:True +null:null \ No newline at end of file diff --git a/test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320_train_infer_python.txt b/test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320_train_infer_python.txt new file mode 100644 index 000000000..3851ac9cc --- /dev/null +++ b/test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320_train_infer_python.txt @@ -0,0 +1,51 @@ +===========================train_params=========================== +model_name:fairmot_hrnetv2_w18_dlafpn_30e_576x320 +python:python3.7 +gpu_list:0|0,1 +use_gpu:True +auto_cast:null +epoch:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=30 +save_dir:null +TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=6 +pretrain_weights:https://paddledet.bj.bcebos.com/models/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.pdparams +trained_model_name:model_final.pdparams +train_infer_img_dir:./dataset/mot/test.mp4 +filename:null +## +trainer:norm_train +norm_train:tools/train.py -c test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml -o +pact_train:tools/train.py -c test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml --slim_config _template_pact -o +fpgm_train:tools/train.py -c test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml --slim_config _template_fpgm -o +distill_train:null +null:null +null:null +## +===========================eval_params=========================== +eval:tools/eval_mot.py -c test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml -o +null:null +## +===========================infer_params=========================== +--output_dir:./output_inference +weights:https://paddledet.bj.bcebos.com/models/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.pdparams +norm_export:tools/export_model.py -c test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml -o +pact_export:tools/export_model.py -c test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml --slim_config _template_fpgm -o +distill_export:null +export1:null +export2:null +kl_quant_export:tools/post_quant.py -c test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml --slim_config _template_kl_quant -o +## +infer_mode:norm +infer_quant:False +inference:./deploy/pptracking/python/mot_jde_infer.py +--device:gpu|cpu +--enable_mkldnn:True|False +--cpu_threads:1|6 +--batch_size:1|8 +--use_tensorrt:null +--run_mode:fluid|trt_fp32|trt_fp16 +--model_dir: +--video_file:./dataset/mot/test.mp4 +--save_log_path:null +--run_benchmark:False +null:null \ No newline at end of file diff --git a/test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt new file mode 100644 index 000000000..abe9c6524 --- /dev/null +++ b/test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -0,0 +1,51 @@ +===========================train_params=========================== +model_name:fairmot_hrnetv2_w18_dlafpn_30e_576x320 +python:python3.7 +gpu_list:0|0,1 +use_gpu:True +auto_cast:amp +epoch:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=30 +save_dir:null +TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=6 +pretrain_weights:https://paddledet.bj.bcebos.com/models/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.pdparams +trained_model_name:fairmot_hrnetv2_w18_dlafpn_30e_576x320/model_final.pdparams +train_infer_img_dir:./dataset/mot/test.mp4 +filename:null +## +trainer:norm_train +norm_train:tools/train.py -c test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml -o +pact_train:tools/train.py -c test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml --slim_config _template_pact -o +fpgm_train:tools/train.py -c test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml --slim_config _template_fpgm -o +distill_train:null +null:null +null:null +## +===========================eval_params=========================== +eval:tools/eval_mot.py -c test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml -o +null:null +## +===========================infer_params=========================== +--output_dir:./output_inference +weights:https://paddledet.bj.bcebos.com/models/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.pdparams +norm_export:tools/export_model.py -c test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml -o +pact_export:tools/export_model.py -c test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml --slim_config _template_fpgm -o +distill_export:null +export1:null +export2:null +kl_quant_export:tools/post_quant.py -c test_tipc/configs/mot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml --slim_config _template_kl_quant -o +## +infer_mode:norm +infer_quant:False +inference:./deploy/pptracking/python/mot_jde_infer.py +--device:gpu|cpu +--enable_mkldnn:True|False +--cpu_threads:1|6 +--batch_size:1|8 +--use_tensorrt:null +--run_mode:fluid|trt_fp16 +--model_dir: +--video_file:./dataset/mot/test.mp4 +--save_log_path:null +--run_benchmark:False +null:null \ No newline at end of file diff --git a/test_tipc/configs/mot/jde_darknet53_30e_576x320_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/mot/jde_darknet53_30e_576x320_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt new file mode 100644 index 000000000..5f61e6826 --- /dev/null +++ b/test_tipc/configs/mot/jde_darknet53_30e_576x320_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -0,0 +1,29 @@ +===========================cpp_infer_params=========================== +model_name:jde_darknet53_30e_576x320 +python:python3.7 +filename:jde_darknet53_30e_576x320 +## +--output_dir:./output_inference +weights:https://paddledet.bj.bcebos.com/models/mot/jde_darknet53_30e_576x320.pdparams +norm_export:tools/export_model.py -c configs/mot/jde/jde_darknet53_30e_576x320.yml -o +quant_export:tools/export_model.py -c configs/mot/jde/jde_darknet53_30e_576x320.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c configs/mot/jde/jde_darknet53_30e_576x320.yml --slim_config _template_fpgm -o +distill_export:null +export1:null +export2:null +kl_quant_export:tools/post_quant.py -c configs/mot/jde/jde_darknet53_30e_576x320.yml --slim_config _template_kl_quant -o +## +opencv_dir:default +infer_mode:norm +infer_quant:False +inference:./deploy/cpp/build/main +--device:gpu|cpu +--use_mkldnn:True|False +--cpu_threads:1|6 +--batch_size:1|8 +--use_tensorrt:null +--run_mode:fluid|trt_fp32|trt_fp16 +--model_dir: +--video_file:./dataset/mot/test.mp4 +--run_benchmark:True +null:null \ No newline at end of file diff --git a/test_tipc/configs/mot/jde_darknet53_30e_576x320_train_infer_python.txt b/test_tipc/configs/mot/jde_darknet53_30e_576x320_train_infer_python.txt new file mode 100644 index 000000000..8af93e8bd --- /dev/null +++ b/test_tipc/configs/mot/jde_darknet53_30e_576x320_train_infer_python.txt @@ -0,0 +1,51 @@ +===========================train_params=========================== +model_name:jde_darknet53_30e_576x320 +python:python3.7 +gpu_list:0|0,1 +use_gpu:True +auto_cast:null +epoch:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=30 +save_dir:null +TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=4 +pretrain_weights:https://paddledet.bj.bcebos.com/models/mot/jde_darknet53_30e_576x320.pdparams +trained_model_name:model_final.pdparams +train_infer_img_dir:./dataset/mot/test.mp4 +filename:null +## +trainer:norm_train +norm_train:tools/train.py -c configs/mot/jde/jde_darknet53_30e_576x320.yml -o +pact_train:tools/train.py -c configs/mot/jde/jde_darknet53_30e_576x320.yml --slim_config _template_pact -o +fpgm_train:tools/train.py -c configs/mot/jde/jde_darknet53_30e_576x320.yml --slim_config _template_fpgm -o +distill_train:null +null:null +null:null +## +===========================eval_params=========================== +eval:tools/eval_mot.py -c configs/mot/jde/jde_darknet53_30e_576x320.yml -o +null:null +## +===========================infer_params=========================== +--output_dir:./output_inference +weights:https://paddledet.bj.bcebos.com/models/mot/jde_darknet53_30e_576x320.pdparams +norm_export:tools/export_model.py -c configs/mot/jde/jde_darknet53_30e_576x320.yml -o +pact_export:tools/export_model.py -c configs/mot/jde/jde_darknet53_30e_576x320.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c configs/mot/jde/jde_darknet53_30e_576x320.yml --slim_config _template_fpgm -o +distill_export:null +export1:null +export2:null +kl_quant_export:tools/post_quant.py -c configs/mot/jde/jde_darknet53_30e_576x320.yml --slim_config _template_kl_quant -o +## +infer_mode:norm +infer_quant:False +inference:./deploy/pptracking/python/mot_jde_infer.py +--device:gpu|cpu +--enable_mkldnn:True|False +--cpu_threads:1|6 +--batch_size:1|8 +--use_tensorrt:null +--run_mode:fluid|trt_fp32|trt_fp16 +--model_dir: +--video_file:./dataset/mot/test.mp4 +--save_log_path:null +--run_benchmark:False +null:null \ No newline at end of file diff --git a/test_tipc/configs/mot/jde_darknet53_30e_576x320_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/mot/jde_darknet53_30e_576x320_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt new file mode 100644 index 000000000..93234c2f7 --- /dev/null +++ b/test_tipc/configs/mot/jde_darknet53_30e_576x320_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -0,0 +1,51 @@ +===========================train_params=========================== +model_name:jde_darknet53_30e_576x320 +python:python3.7 +gpu_list:0|0,1 +use_gpu:True +auto_cast:amp +epoch:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=30 +save_dir:null +TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=4 +pretrain_weights:https://paddledet.bj.bcebos.com/models/mot/jde_darknet53_30e_576x320.pdparams +trained_model_name:jde_darknet53_30e_576x320/model_final.pdparams +train_infer_img_dir:./dataset/mot/test.mp4 +filename:null +## +trainer:norm_train +norm_train:tools/train.py -c configs/mot/jde/jde_darknet53_30e_576x320.yml -o +pact_train:tools/train.py -c configs/mot/jde/jde_darknet53_30e_576x320.yml --slim_config _template_pact -o +fpgm_train:tools/train.py -c configs/mot/jde/jde_darknet53_30e_576x320.yml --slim_config _template_fpgm -o +distill_train:null +null:null +null:null +## +===========================eval_params=========================== +eval:tools/eval_mot.py -c configs/mot/jde/jde_darknet53_30e_576x320.yml -o +null:null +## +===========================infer_params=========================== +--output_dir:./output_inference +weights:https://paddledet.bj.bcebos.com/models/mot/jde_darknet53_30e_576x320.pdparams +norm_export:tools/export_model.py -c configs/mot/jde/jde_darknet53_30e_576x320.yml -o +pact_export:tools/export_model.py -c configs/mot/jde/jde_darknet53_30e_576x320.yml --slim_config _template_pact -o +fpgm_export:tools/export_model.py -c configs/mot/jde/jde_darknet53_30e_576x320.yml --slim_config _template_fpgm -o +distill_export:null +export1:null +export2:null +kl_quant_export:tools/post_quant.py -c configs/mot/jde/jde_darknet53_30e_576x320.yml --slim_config _template_kl_quant -o +## +infer_mode:norm +infer_quant:False +inference:./deploy/pptracking/python/mot_jde_infer.py +--device:gpu|cpu +--enable_mkldnn:True|False +--cpu_threads:1|6 +--batch_size:1|8 +--use_tensorrt:null +--run_mode:fluid|trt_fp16 +--model_dir: +--video_file:./dataset/mot/test.mp4 +--save_log_path:null +--run_benchmark:False +null:null \ No newline at end of file diff --git a/test_tipc/configs/picodet/picodet_l_640_coco_train_infer_python.txt b/test_tipc/configs/picodet/picodet_l_640_coco_train_infer_python.txt index fa493777b..167e2a059 100644 --- a/test_tipc/configs/picodet/picodet_l_640_coco_train_infer_python.txt +++ b/test_tipc/configs/picodet/picodet_l_640_coco_train_infer_python.txt @@ -41,7 +41,7 @@ inference:./deploy/python/infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp32|trt_fp16 --model_dir: diff --git a/test_tipc/configs/picodet/picodet_lcnet_1_5x_416_coco_train_infer_python.txt b/test_tipc/configs/picodet/picodet_lcnet_1_5x_416_coco_train_infer_python.txt index 91e2868dc..b20d8d1d8 100644 --- a/test_tipc/configs/picodet/picodet_lcnet_1_5x_416_coco_train_infer_python.txt +++ b/test_tipc/configs/picodet/picodet_lcnet_1_5x_416_coco_train_infer_python.txt @@ -41,7 +41,7 @@ inference:./deploy/python/infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp32|trt_fp16 --model_dir: diff --git a/test_tipc/configs/picodet/picodet_m_416_coco_train_infer_python.txt b/test_tipc/configs/picodet/picodet_m_416_coco_train_infer_python.txt index 6b78b6b7c..81208aca8 100644 --- a/test_tipc/configs/picodet/picodet_m_416_coco_train_infer_python.txt +++ b/test_tipc/configs/picodet/picodet_m_416_coco_train_infer_python.txt @@ -41,7 +41,7 @@ inference:./deploy/python/infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp32|trt_fp16 --model_dir: diff --git a/test_tipc/configs/picodet/picodet_mobilenetv3_large_1x_416_coco_train_infer_python.txt b/test_tipc/configs/picodet/picodet_mobilenetv3_large_1x_416_coco_train_infer_python.txt index e0ae18135..b896ef725 100644 --- a/test_tipc/configs/picodet/picodet_mobilenetv3_large_1x_416_coco_train_infer_python.txt +++ b/test_tipc/configs/picodet/picodet_mobilenetv3_large_1x_416_coco_train_infer_python.txt @@ -41,7 +41,7 @@ inference:./deploy/python/infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp32|trt_fp16 --model_dir: diff --git a/test_tipc/configs/picodet/picodet_r18_640_coco_train_infer_python.txt b/test_tipc/configs/picodet/picodet_r18_640_coco_train_infer_python.txt index b7c735636..43dd6887a 100644 --- a/test_tipc/configs/picodet/picodet_r18_640_coco_train_infer_python.txt +++ b/test_tipc/configs/picodet/picodet_r18_640_coco_train_infer_python.txt @@ -41,7 +41,7 @@ inference:./deploy/python/infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp32|trt_fp16 --model_dir: diff --git a/test_tipc/configs/picodet/picodet_s_320_coco_train_infer_python.txt b/test_tipc/configs/picodet/picodet_s_320_coco_train_infer_python.txt index c95d18074..ed667e39b 100644 --- a/test_tipc/configs/picodet/picodet_s_320_coco_train_infer_python.txt +++ b/test_tipc/configs/picodet/picodet_s_320_coco_train_infer_python.txt @@ -41,7 +41,7 @@ inference:./deploy/python/infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp32|trt_fp16 --model_dir: diff --git a/test_tipc/configs/picodet/picodet_shufflenetv2_1x_416_coco_train_infer_python.txt b/test_tipc/configs/picodet/picodet_shufflenetv2_1x_416_coco_train_infer_python.txt index 92886d1ae..cc281988f 100644 --- a/test_tipc/configs/picodet/picodet_shufflenetv2_1x_416_coco_train_infer_python.txt +++ b/test_tipc/configs/picodet/picodet_shufflenetv2_1x_416_coco_train_infer_python.txt @@ -41,7 +41,7 @@ inference:./deploy/python/infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp32|trt_fp16 --model_dir: diff --git a/test_tipc/configs/solov2/solov2_r101_vd_fpn_3x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/solov2/solov2_r101_vd_fpn_3x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index 762ee7e4a..6f0fc4b59 100644 --- a/test_tipc/configs/solov2/solov2_r101_vd_fpn_3x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/solov2/solov2_r101_vd_fpn_3x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -20,10 +20,10 @@ inference:./deploy/cpp/build/main --device:gpu|cpu --use_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp32|trt_fp16 --model_dir: --image_dir:./dataset/coco/test2017/ ---run_benchmark:True +--run_benchmark:False null:null \ No newline at end of file diff --git a/test_tipc/configs/solov2/solov2_r101_vd_fpn_3x_coco_train_infer_python.txt b/test_tipc/configs/solov2/solov2_r101_vd_fpn_3x_coco_train_infer_python.txt index b3152d414..f65e3ac35 100644 --- a/test_tipc/configs/solov2/solov2_r101_vd_fpn_3x_coco_train_infer_python.txt +++ b/test_tipc/configs/solov2/solov2_r101_vd_fpn_3x_coco_train_infer_python.txt @@ -41,11 +41,11 @@ inference:./deploy/python/infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp32|trt_fp16 --model_dir: --image_dir:./dataset/coco/test2017/ --save_log_path:null ---run_benchmark:True +--run_benchmark:False null:null \ No newline at end of file diff --git a/test_tipc/configs/solov2/solov2_r101_vd_fpn_3x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/solov2/solov2_r101_vd_fpn_3x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 8b9ecca75..5315ffb91 100644 --- a/test_tipc/configs/solov2/solov2_r101_vd_fpn_3x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/solov2/solov2_r101_vd_fpn_3x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -41,11 +41,11 @@ inference:./deploy/python/infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp16 --model_dir: --image_dir:./dataset/coco/test2017/ --save_log_path:null ---run_benchmark:True +--run_benchmark:False null:null \ No newline at end of file diff --git a/test_tipc/configs/solov2/solov2_r50_enhance_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/solov2/solov2_r50_enhance_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index f8a020ff5..dafb24ff4 100644 --- a/test_tipc/configs/solov2/solov2_r50_enhance_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/solov2/solov2_r50_enhance_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -20,7 +20,7 @@ inference:./deploy/cpp/build/main --device:gpu|cpu --use_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp32|trt_fp16 --model_dir: diff --git a/test_tipc/configs/solov2/solov2_r50_enhance_coco_train_infer_python.txt b/test_tipc/configs/solov2/solov2_r50_enhance_coco_train_infer_python.txt index 02c2e82f4..6931134ba 100644 --- a/test_tipc/configs/solov2/solov2_r50_enhance_coco_train_infer_python.txt +++ b/test_tipc/configs/solov2/solov2_r50_enhance_coco_train_infer_python.txt @@ -41,7 +41,7 @@ inference:./deploy/python/infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp32|trt_fp16 --model_dir: diff --git a/test_tipc/configs/solov2/solov2_r50_enhance_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/solov2/solov2_r50_enhance_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 63be0b362..6c09bb1d0 100644 --- a/test_tipc/configs/solov2/solov2_r50_enhance_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/solov2/solov2_r50_enhance_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -41,7 +41,7 @@ inference:./deploy/python/infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp16 --model_dir: diff --git a/test_tipc/configs/solov2/solov2_r50_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt b/test_tipc/configs/solov2/solov2_r50_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt index e6c80a5e2..454f4fa79 100644 --- a/test_tipc/configs/solov2/solov2_r50_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt +++ b/test_tipc/configs/solov2/solov2_r50_fpn_1x_coco_model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt @@ -20,10 +20,10 @@ inference:./deploy/cpp/build/main --device:gpu|cpu --use_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp32|trt_fp16 --model_dir: --image_dir:./dataset/coco/test2017/ --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/solov2/solov2_r50_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/solov2/solov2_r50_fpn_1x_coco_train_infer_python.txt index 5f32ef5ba..6d8901f9b 100644 --- a/test_tipc/configs/solov2/solov2_r50_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/solov2/solov2_r50_fpn_1x_coco_train_infer_python.txt @@ -41,11 +41,11 @@ inference:./deploy/python/infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp32|trt_fp16 --model_dir: --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/configs/solov2/solov2_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt b/test_tipc/configs/solov2/solov2_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt index 71e73048e..29688f569 100644 --- a/test_tipc/configs/solov2/solov2_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt +++ b/test_tipc/configs/solov2/solov2_r50_fpn_1x_coco_train_linux_gpu_normal_amp_infer_python_linux_gpu_cpu.txt @@ -41,11 +41,11 @@ inference:./deploy/python/infer.py --device:gpu|cpu --enable_mkldnn:True|False --cpu_threads:1|6 ---batch_size:1|8 +--batch_size:1 --use_tensorrt:null --run_mode:fluid|trt_fp16 --model_dir: --image_dir:./dataset/coco/test2017/ --save_log_path:null --run_benchmark:True -null:null \ No newline at end of file +--trt_max_shape:1600 \ No newline at end of file diff --git a/test_tipc/docs/install.md b/test_tipc/docs/install.md index 360050656..eaac6908d 100644 --- a/test_tipc/docs/install.md +++ b/test_tipc/docs/install.md @@ -110,7 +110,7 @@ cd TensorRT-${version}/graphsurgeon ### 3.3 安装PaddlePaddle -下载支持TensorRT版本的Paddle安装包,注意安装包的TensorRT版本需要与本地TensorRT一致,下载[链接](https://paddleinference.paddlepaddle.org.cn/user_guides/download_lib.html#python) +下载支持TensorRT版本的Paddle安装包,注意安装包的TensorRT版本需要与本地TensorRT一致,下载[链接](https://paddleinference.paddlepaddle.org.cn/master/user_guides/download_lib.html#python) 选择下载 linux-cuda10.2-trt7-gcc8.2 Python3.7版本的Paddle: ``` # 从下载链接中可以看到是paddle2.1.1-cuda10.2-cudnn8.1版本 diff --git a/test_tipc/docs/more_models.md b/test_tipc/docs/more_models.md new file mode 100644 index 000000000..f559dd85a --- /dev/null +++ b/test_tipc/docs/more_models.md @@ -0,0 +1,82 @@ +## 汇总信息 + +已填写的部分表示可以使用本工具进行一键测试,未填写的表示正在支持中。 + +**字段说明:** +- 基础训练预测:包括模型训练、Paddle Inference Python预测。 +- 更多训练方式:包括多机多卡、混合精度。 +- 模型压缩:包括裁剪、离线/在线量化、蒸馏。 +- 其他预测部署:包括Paddle Inference C++预测、Paddle Serving部署、Paddle-Lite部署等。 + +| 算法论文 | 模型名称 | 模型类型 | 基础
训练预测 | 更多
训练方式 | 模型压缩 | 其他预测部署 | +| :--- | :--- | :----: | :--------: | :---- | :---- | :---- | +| [YOLOv3](https://arxiv.org/abs/1804.02767) | [yolov3_darknet53_270e_coco](../configs/yolov3/yolov3_darknet53_270e_coco.yml) | 目标检测 | 支持 | 混合精度 | FPGM裁剪
PACT量化
离线量化 | Paddle Inference: C++ | +| YOLOv3 | [yolov3_mobilenet_v1_270e_coco](../configs/yolov3/yolov3_mobilenet_v1_270e_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| YOLOv3 | [yolov3_mobilenet_v3_large_270e_coco](../configs/yolov3/yolov3_mobilenet_v3_large_270e_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| YOLOv3 | [yolov3_r34_270e_coco](../configs/yolov3/yolov3_r34_270e_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| YOLOv3 | [yolov3_r50vd_dcn_270e_coco](../configs/yolov3/yolov3_r50vd_dcn_270e_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| [PPYOLO](https://arxiv.org/abs/2007.12099) | [ppyolo_mbv3_large_coco](../configs/ppyolo/ppyolo_mbv3_large_coco.yml) | 目标检测 | 支持 | 混合精度 | FPGM裁剪
PACT量化
离线量化 | Paddle Inference: C++ | +| PPYOLO | [ppyolo_r50vd_dcn_1x_coco](../configs/ppyolo/ppyolo_r50vd_dcn_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | FPGM裁剪
PACT量化
离线量化 | Paddle Inference: C++ | +| PPYOLO | [ppyolo_mbv3_small_coco](../configs/ppyolo/ppyolo_mbv3_small_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| PPYOLO | [ppyolo_r18vd_coco](../configs/ppyolo/ppyolo_r18vd_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| PPYOLO-tiny | [ppyolo_tiny_650e_coco](../configs/ppyolo/ppyolo_tiny_650e_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| [PPYOLOv2](https://arxiv.org/abs/2104.10419) | [ppyolov2_r50vd_dcn_365e_coco](../configs/ppyolo/ppyolov2_r50vd_dcn_365e_coco.yml) | 目标检测 | 支持 | 多机多卡
混合精度 | | Paddle Inference: C++ | +| PPYOLOv2 | [ppyolov2_r50vd_dcn_365e_coco](../configs/ppyolo/ppyolov2_r50vd_dcn_365e_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| PPYOLOv2 | [ppyolov2_r101vd_dcn_365e_coco](../configs/ppyolo/ppyolov2_r101vd_dcn_365e_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| [PP-PicoDet](https://arxiv.org/abs/2111.00902) | [picodet_s_320_coco](../configs/picodet/picodet_s_320_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| PP-PicoDet | [picodet_m_416_coco](../configs/picodet/picodet_m_416_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| PP-PicoDet | [picodet_l_640_coco](../configs/picodet/picodet_l_640_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| PP-PicoDet | [picodet_lcnet_1_5x_416_coco](../configs/picodet/more_config/picodet_lcnet_1_5x_416_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| PP-PicoDet | [picodet_mobilenetv3_large_1x_416_coco](../configs/picodet/more_config/picodet_mobilenetv3_large_1x_416_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| PP-PicoDet | [picodet_r18_640_coco](../configs/picodet/more_config/picodet_r18_640_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| PP-PicoDet | [picodet_shufflenetv2_1x_416_coco](../configs/picodet/more_config/picodet_shufflenetv2_1x_416_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| [SSD](https://arxiv.org/abs/1512.02325) | [ssdlite_mobilenet_v1_300_coco](../configs/ssd/ssdlite_mobilenet_v1_300_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| [Faster R-CNN](https://arxiv.org/abs/1506.01497) | [faster_rcnn_r50_fpn_1x_coco](../configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| Faster R-CNN | [faster_rcnn_r34_fpn_1x_coco](../configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| Faster R-CNN | [faster_rcnn_r34_vd_fpn_1x_coco](../configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| Faster R-CNN | [faster_rcnn_r50_1x_coco](../configs/faster_rcnn/faster_rcnn_r50_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| Faster R-CNN | [faster_rcnn_r50_vd_1x_coco](../configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| Faster R-CNN | [faster_rcnn_r50_vd_fpn_1x_coco](../configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| Faster R-CNN | [faster_rcnn_r101_1x_coco](../configs/faster_rcnn/faster_rcnn_r101_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| Faster R-CNN | [faster_rcnn_r101_fpn_1x_coco](../configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| Faster R-CNN | [faster_rcnn_r101_vd_fpn_1x_coco](../configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| Faster R-CNN | [faster_rcnn_x101_vd_64x4d_fpn_1x_coco](../configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| Faster R-CNN | [faster_rcnn_swin_tiny_fpn_1x_coco](../configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| [Cascade Faster R-CNN](https://arxiv.org/abs/1712.00726) | [cascade_rcnn_r50_fpn_1x_coco](../configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| Cascade Faster R-CNN | [cascade_rcnn_r50_vd_fpn_ssld_1x_coco](../configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| [FCOS](https://arxiv.org/abs/1904.01355) | [fcos_r50_fpn_1x_coco](../configs/fcos/fcos_r50_fpn_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| FCOS | [fcos_dcn_r50_fpn_1x_coco](../configs/fcos/fcos_dcn_r50_fpn_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| [TTFNet](https://arxiv.org/abs/1909.00700) | [ttfnet_darknet53_1x_coco](../configs/ttfnet/ttfnet_darknet53_1x_coco.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| [S2ANet](https://arxiv.org/abs/2008.09397) | [s2anet_conv_2x_dota](../configs/dota/s2anet_conv_2x_dota.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| S2ANet | [s2anet_1x_spine](../configs/dota/s2anet_1x_spine.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| S2ANet | [s2anet_alignconv_2x_dota](../configs/dota/s2anet_alignconv_2x_dota.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| [DETR](https://arxiv.org/abs/2005.12872) | [detr_r50_1x_coco](../configs/detr/detr_r50_1x_coco.yml) | 目标检测 | 不支持预测部署 | +| [Deformable DETR](https://arxiv.org/abs/2010.04159) | [deformable_detr_r50_1x_coco](../configs/deformable_detr/deformable_detr_r50_1x_coco.yml) | 目标检测 | 不支持预测部署 | +| [Sparse R-CNN](https://arxiv.org/abs/2011.12450) | [sparse_rcnn_r50_fpn_3x_pro100_coco](../configs/sparse_rcnn/sparse_rcnn_r50_fpn_3x_pro100_coco.yml) | 目标检测 | 不支持预测部署 | +| [BlazeFace](https://arxiv.org/abs/1907.05047) | [blazeface_1000e](../configs/face_detection/blazeface_1000e.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| BlazeFace | [blazeface_fpn_ssh_1000e](../configs/face_detection/blazeface_fpn_ssh_1000e.yml) | 目标检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| [Faceboxes](https://arxiv.org/abs/1708.05234) | | 目标检测 | 静态图实现,暂不支持 | +| [PyramidBox](https://arxiv.org/abs/1803.07737) | 模型实现在[models](https://github.com/PaddlePaddle/models/tree/develop/PaddleCV/face_detection) repo | 目标检测 | 静态图实现,暂不支持 | +| [Libra R-CNN](https://arxiv.org/abs/1904.02701) | | 目标检测 | 静态图实现,暂不支持 | +| [RetinaNet](https://arxiv.org/abs/1708.02002) | | 目标检测 | 静态图实现,暂不支持 | +| [CornerNet](https://arxiv.org/abs/1808.01244) | | 目标检测 | 静态图实现,暂不支持 | +| [EfficientDet](https://arxiv.org/abs/1911.09070) | | 目标检测 | 静态图实现,暂不支持 | +| [Mask R-CNN](https://arxiv.org/abs/1703.06870) | [mask_rcnn_r50_fpn_1x_coco](../configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco.yml) | 实例分割 | 支持 | 混合精度 | | Paddle Inference: C++ | +| Mask R-CNN | [mask_rcnn_r50_1x_coco](../configs/mask_rcnn/mask_rcnn_r50_1x_coco.yml) | 实例分割 | 支持 | 混合精度 | | Paddle Inference: C++ | +| Mask R-CNN | [mask_rcnn_r50_vd_fpn_1x_coco](../configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco.yml) | 实例分割 | 支持 | 混合精度 | | Paddle Inference: C++ | +| Mask R-CNN | [mask_rcnn_r101_fpn_1x_coco](../configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco.yml) | 实例分割 | 支持 | 混合精度 | | Paddle Inference: C++ | +| Mask R-CNN | [mask_rcnn_r101_vd_fpn_1x_coco](../configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco.yml) | 实例分割 | 支持 | 混合精度 | | Paddle Inference: C++ | +| Mask R-CNN | [mask_rcnn_x101_vd_64x4d_fpn_1x_coco](../configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco.yml) | 实例分割 | 支持 | 混合精度 | | Paddle Inference: C++ | +| [Cascade Mask R-CNN](https://arxiv.org/abs/1906.09756) | [cascade_mask_rcnn_r50_fpn_1x_coco](../configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco.yml) | 实例分割 | 支持 | 混合精度 | | Paddle Inference: C++ | +| Cascade Mask R-CNN | [cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco](../configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco.yml) | 实例分割 | 支持 | 混合精度 | | Paddle Inference: C++ | +| [SOLOv2](https://arxiv.org/abs/2003.10152) | [solov2_r50_fpn_1x_coco](../configs/solov2/solov2_r50_fpn_1x_coco.yml) | 实例分割 | 支持 | 混合精度 | | Paddle Inference: C++ | +| SOLOv2 | [solov2_r50_enhance_coco](../configs/solov2/solov2_r50_enhance_coco.yml) | 实例分割 | 支持 | 混合精度 | | Paddle Inference: C++ | +| SOLOv2 | [solov2_r101_vd_fpn_3x_coco](../configs/solov2/solov2_r101_vd_fpn_3x_coco.yml) | 实例分割 | 支持 | 混合精度 | | Paddle Inference: C++ | +| [PP-Tinypose] | [tinypose_128x96](../configs/keypoint/tiny_pose/tinypose_128x96.yml) | 关键点检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| [HRNet](https://arxiv.org/abs/1902.09212) | [hrnet_w32_256x192](../configs/keypoint/hrnet/hrnet_w32_256x192.yml) | 关键点检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| HRNet | [dark_hrnet_w32_256x192](../configs/keypoint/hrnet/dark_hrnet_w32_256x192.yml) | 关键点检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| HRNet | [dark_hrnet_w48_256x192](../configs/keypoint/hrnet/dark_hrnet_w48_256x192.yml) | 关键点检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| [HigherHRNet](https://arxiv.org/abs/1908.10357) | [higherhrnet_hrnet_w32_512](../configs/keypoint/higherhrnet/higherhrnet_hrnet_w32_512.yml) | 关键点检测 | 支持 | 混合精度 | | Paddle Inference: C++ | +| [FairMot](https://arxiv.org/abs/2004.01888) | [fairmot_dla34_30e_576x320](../configs/mot/fairmot/fairmot_dla34_30e_576x320.yml) | 目标跟踪 | 支持 | 混合精度 | | Paddle Inference: C++ | +| FairMot | [fairmot_hrnetv2_w18_dlafpn_30e_576x320](../configs/mot/fairmot/fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml) | 目标跟踪 | 支持 | 混合精度 | | Paddle Inference: C++ | +| [JDE](https://arxiv.org/abs/1909.12605) | [jde_darknet53_30e_576x320](../configs/mot/jde/jde_darknet53_30e_576x320.yml) | 目标跟踪 | 支持 | 混合精度 | | Paddle Inference: C++ | diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh index fd007593b..ca3328825 100644 --- a/test_tipc/prepare.sh +++ b/test_tipc/prepare.sh @@ -16,23 +16,32 @@ lines=(${dataline}) model_name=$(func_parser_value "${lines[1]}") python=$(func_parser_value "${lines[2]}") -if [ ${MODE} = "lite_train_lite_infer" ] || [ ${MODE} = "lite_train_whole_infer" ];then - mv ./dataset/coco/download_coco.py . && rm -rf ./dataset/coco/* && mv ./download_coco.py ./dataset/coco/ - # download data - wget -nc -P ./dataset/coco/ https://paddledet.bj.bcebos.com/data/coco_tipc.tar - cd ./dataset/coco/ && tar -xvf coco_tipc.tar && mv -u coco_tipc/* . - rm -rf coco_tipc/ && cd ../../ -elif [ ${MODE} = "whole_train_whole_infer" ];then +if [ ${MODE} = "whole_train_whole_infer" ];then mv ./dataset/coco/download_coco.py . && rm -rf ./dataset/coco/* && mv ./download_coco.py ./dataset/coco/ # prepare whole training data eval "${python} ./dataset/coco/download_coco.py" -elif [ ${MODE} = "whole_infer" ] || [ ${MODE} = "klquant_whole_infer" ];then - mv ./dataset/coco/download_coco.py . && rm -rf ./dataset/coco/* && mv ./download_coco.py ./dataset/coco/ - # download data - wget -nc -P ./dataset/coco/ https://paddledet.bj.bcebos.com/data/coco_tipc.tar - cd ./dataset/coco/ && tar -xvf coco_tipc.tar && mv -u coco_tipc/* . - rm -rf coco_tipc/ && cd ../../ elif [ ${MODE} = "cpp_infer" ];then + # download coco lite data + wget -nc -P ./dataset/coco/ https://paddledet.bj.bcebos.com/data/tipc/coco_tipc.tar + cd ./dataset/coco/ && tar -xvf coco_tipc.tar && mv -n coco_tipc/* . + rm -rf coco_tipc/ && cd ../../ + # download wider_face lite data + wget -nc -P ./dataset/wider_face/ https://paddledet.bj.bcebos.com/data/tipc/wider_tipc.tar + cd ./dataset/wider_face/ && tar -xvf wider_tipc.tar && mv -n wider_tipc/* . + rm -rf wider_tipc/ && cd ../../ + # download spine lite data + wget -nc -P ./dataset/spine_coco/ https://paddledet.bj.bcebos.com/data/tipc/spine_tipc.tar + cd ./dataset/spine_coco/ && tar -xvf spine_tipc.tar && mv -n spine_tipc/* . + rm -rf spine_tipc/ && cd ../../ + if [[ ${model_name} =~ "s2anet" ]]; then + cd ./ppdet/ext_op && eval "${python} setup.py install" + cd ../../ + fi + # download mot lite data + wget -nc -P ./dataset/mot/ https://paddledet.bj.bcebos.com/data/tipc/mot_tipc.tar + cd ./dataset/mot/ && tar -xvf mot_tipc.tar && mv -n mot_tipc/* . + rm -rf mot_tipc/ && cd ../../ + opencv_dir=$(func_parser_value "${lines[15]}") # prepare opencv cd ./deploy/cpp @@ -48,30 +57,24 @@ elif [ ${MODE} = "cpp_infer" ];then fi cd ../../ else - mv ./dataset/coco/download_coco.py . && rm -rf ./dataset/coco/* && mv ./download_coco.py ./dataset/coco/ - # prepare infer data - wget -nc -P ./dataset/coco/ https://paddledet.bj.bcebos.com/data/coco_tipc.tar - cd ./dataset/coco/ && tar -xvf coco_tipc.tar && mv -u coco_tipc/* . + # download coco lite data + wget -nc -P ./dataset/coco/ https://paddledet.bj.bcebos.com/data/tipc/coco_tipc.tar + cd ./dataset/coco/ && tar -xvf coco_tipc.tar && mv -n coco_tipc/* . rm -rf coco_tipc/ && cd ../../ - if [[ ${model_name} = "yolov3_darknet53_270e_coco" ]]; then - wget -nc -P ./tests/weights/ https://paddledet.bj.bcebos.com/models/yolov3_darknet53_270e_coco.pdparams - wget -nc -P ./tests/weights/ https://paddledet.bj.bcebos.com/models/slim/yolov3_darknet_coco_qat.pdparams - wget -nc -P ./tests/weights/ https://paddledet.bj.bcebos.com/models/slim/yolov3_darknet_prune_fpgm.pdparams - elif [[ ${model_name} = "ppyolo_r50vd_dcn_1x_coco" ]]; then - wget -nc -P ./tests/weights/ https://paddledet.bj.bcebos.com/models/ppyolo_r50vd_dcn_1x_coco.pdparams - wget -nc -P ./tests/weights/ https://paddledet.bj.bcebos.com/models/slim/ppyolo_r50vd_qat_pact.pdparams - wget -nc -P ./tests/weights/ https://paddledet.bj.bcebos.com/models/slim/ppyolo_r50vd_prune_fpgm.pdparams - elif [[ ${model_name} = "ppyolo_mbv3_large_coco" ]]; then - wget -nc -P ./tests/weights/ https://paddledet.bj.bcebos.com/models/ppyolo_mbv3_large_coco.pdparams - wget -nc -P ./tests/weights/ https://paddledet.bj.bcebos.com/models/slim/ppyolo_mbv3_large_qat.pdparams - wget -nc -P ./tests/weights/ https://paddledet.bj.bcebos.com/models/slim/ppyolo_mbv3_large_prune_fpgm.pdparams - elif [[ ${model_name} = "ppyolov2_r50vd_dcn_365e_coco" ]]; then - wget -nc -P ./tests/weights/ https://paddledet.bj.bcebos.com/models/ppyolov2_r50vd_dcn_365e_coco.pdparams - elif [[ ${model_name} = "mask_rcnn_r50_fpn_1x_coco" ]]; then - wget -nc -P ./tests/weights/ https://paddledet.bj.bcebos.com/models/mask_rcnn_r50_fpn_1x_coco.pdparams - elif [[ ${model_name} = "solov2_r50_fpn_1x_coco" ]]; then - wget -nc -P ./tests/weights/ https://paddledet.bj.bcebos.com/models/solov2_r50_fpn_1x_coco.pdparams - else - sleep 1 + # download wider_face lite data + wget -nc -P ./dataset/wider_face/ https://paddledet.bj.bcebos.com/data/tipc/wider_tipc.tar + cd ./dataset/wider_face/ && tar -xvf wider_tipc.tar && mv -n wider_tipc/* . + rm -rf wider_tipc/ && cd ../../ + # download spine_coco lite data + wget -nc -P ./dataset/spine_coco/ https://paddledet.bj.bcebos.com/data/tipc/spine_tipc.tar + cd ./dataset/spine_coco/ && tar -xvf spine_tipc.tar && mv -n spine_tipc/* . + rm -rf spine_tipc/ && cd ../../ + if [[ ${model_name} =~ "s2anet" ]]; then + cd ./ppdet/ext_op && eval "${python} setup.py install" + cd ../../ fi + # download mot lite data + wget -nc -P ./dataset/mot/ https://paddledet.bj.bcebos.com/data/tipc/mot_tipc.tar + cd ./dataset/mot/ && tar -xvf mot_tipc.tar && mv -n mot_tipc/* . + rm -rf mot_tipc/ && cd ../../ fi -- GitLab