diff --git a/test_tipc/README.md b/test_tipc/README.md index 5231ae7b0451c660c27cd1d2378a0593ddd28ead..befe05f5b03b821c15dff849d5635f36a451f6ac 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 0000000000000000000000000000000000000000..fb25bd8eab27b3073aedf1d79b2b19206188e967 --- /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 0000000000000000000000000000000000000000..9737fec11eb1070321f6c832e2e0ffa73d5b2845 --- /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 0000000000000000000000000000000000000000..ea7296398ba2c9613f00b64dcc415c5100fbf5bb --- /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 0000000000000000000000000000000000000000..89f3b5f8aa44be7140f26416b92a7a47b7ec81b7 --- /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 0000000000000000000000000000000000000000..1d64264390a958a79a66b101967a68cc47c50cde --- /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 0000000000000000000000000000000000000000..1007ccf579c5e6d055207bc04ba512fcde05568f --- /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 0000000000000000000000000000000000000000..6b1ea1605c77ac1b09f4d162f369cd86550967e1 --- /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 0000000000000000000000000000000000000000..746ef0cc90a79e08033c48267f3e3118167fd938 --- /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 0000000000000000000000000000000000000000..e90e4c8929c0c4b2d0c19ce9004729ab7667978c --- /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 0000000000000000000000000000000000000000..9b26a809a0d205eeb547b20615448a1d516dd1ff --- /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 0000000000000000000000000000000000000000..9ca4184e49e52cc1cdc6f72a4fc066f514ba0a5b --- /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 0000000000000000000000000000000000000000..829766393f943c4a213873612fda26456e249389 --- /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 0000000000000000000000000000000000000000..124843c838447600d9a610ff84ce5949516e0abb --- /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 0000000000000000000000000000000000000000..9586ed00792b828dfb6d09833b36617ea4ed7939 --- /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 0000000000000000000000000000000000000000..79142a7530defa1a1d4030a68905454c56b75d85 --- /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 0000000000000000000000000000000000000000..bb92515c548079c43cc586e8df309bd7b4b9f971 --- /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 0000000000000000000000000000000000000000..d09667e58ca2304b57fb883d80622ee25b233b1c --- /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 2c11c0ac32583b227525ce388a590a7bc500d8e6..bc0262a0cfb148b125be8d5ae67314762f25897e 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 a67e9deeeeff6eb16f0418cbe947cea2290995ef..b10141d9414394c446dd9e5da792dbd3504f32e8 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 9c03d2b90322e4cccf937790dd22d7bc8c96fb47..2d00cae156dfe84032fb0f2ac032c50bdbb6bcf3 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 a1f877da1e1963859d46ff2a197e7fcad2b22e95..a06fbc00d2dc4cbe2fb0a986de5cc10cb69dc433 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 c018669c117510d2524524a9bfa424435da92620..bcf36f97d915e72c05d9177bc04f91665dc612d4 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 5a78be3a549bd6fdb2c9fcb79b52bbee0583135d..755b63a8b9e5a8336a0ca06655d6ee6ef4c11c57 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 0035fd912c362fe8e3329946e1ba713d9c043604..a5c37d7e526529dc6ec8f172ca8c6d240bbbcd2e 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 8129670fd6dc8361c32a527bda46856c4cdcdf09..53f4b23a97fc909934337aebfff53b4df787e975 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 4d2fc2210f2d5dec30a9b3e7ecfbb5032ded6d72..e9bab4d472f20328c9d55f38530fbbc41e05b5bb 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 40f226447f3c1751c4b6bf1f193dc9ddc5455d29..f325c0c972b8c69b872818170358f7d5a06169d7 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 29d37fc2956f8e36150b0a60d38791e608fed70d..74f8bb423c55a78a40de11590c78cee5e35dd00f 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 5bf1d49f276c083b76f8428dd0b186eb463db10c..bac2e78aa88b47c8a1641c6f1f58ce68710deebf 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 2d3d54d54f1a354d750a94a45b24bff947dbab6d..46287f8b84a4cae45e28f4ddbe778ac13abf9849 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 01a622f484947a1d5cc242606eae18ff519a9bcb..499daea42d7885727bc64ca2f9892663525879a5 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 56a9ac380cb66b621763c92e55224968e9620991..844e5cacd6207f0d02bb325fc59d0e239b66c29a 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 579b81970efa365cc1d0361afaea74a927735899..1fac1ba3f8cc2dc66c80abf61069a58d52590165 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 c37e497d5e4e012bbdf5f0d6ee508842495cbb0c..2f7035bf09651f74c900305a5fde5cd3071dbb97 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 01f1121aa7df4d47322d081a5bda90c5e65da0be..653de7c5d8fd84b698e208c66a3e5cb148a06083 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 9b9489d0849b853767ec94f45aec6a56c067f551..f4f436485df4982c0bf65211eea88bcb608b4156 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 85ccca9405cdad514bfc881bd0813a42bdd47725..8e9e841e00afddf64d60540de96a0f944d23bc95 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 7d52cf5e0dd9a11b161b9359b96282ae2200a1c4..c798610fa6958abe856f7976a8f830ebbb0c18bd 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 af2d871f0b958483ed0f66313dd44a68c0c6fba4..77bebb1f03115cddb5eb19e8cc3e27ffa53e25e0 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 378e3dc53bc6965868b0e283d8961281d5048464..22c169c99f30c488830873e3ad8a1dc2de0a5e68 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 2a6e623f8e6cec9e8de23b93308fdc5385e0ae65..7f85facea797cd6f1a62cc3db5c729f1a0a1c500 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 ce38015cfafbda515a1aa3f3a470463663052ef5..53baebba2a9fccef51beb74b3c743fa8394b2372 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 808999ad19ebd333864f8027ece74411ad928014..a0c6e64e5a14c3ca73f9834f554e73ab8b7e2074 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 901566ca2da93f1d733e5660d8a0e341db42e8bc..dc2483f591d4bb078fd5efcdb853a1a1b53e4edc 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 73583ca09fe34d9871f5e7314ba8d528291a9c67..29fcc1b63dd117a820527e0292f0215384cd4f53 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 c3827db4baa843e004cac7a3497e459dc1866073..aae1c3dfae37bd4196c76e6cbb200efe2fe92191 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 f9b24c34cd64f7610a33980ebf10574f8f5b12ab..8ded89c00dada9cdb8923890726ad2366dcaba1e 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 fa03514e665801332682659f5bd971d5c3742609..a448a8187269a3c85d154b6304a1a0c9a7dd8f2f 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 db95088739526b3ff5af2fd70012956b94a7fb60..1e560b78e3d0ccfaa682b7a332e35e5ac9f51c4d 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 f0be1d47a50f8158e504e946edc0affd0e510379..b721de6c45e305559115c1d3756d15fb556cca8f 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 77d0508b162843e29dd5c81de4d99f88cdd18f17..6958e88bd129c5f84b43b226d960dd8d470e8069 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 185d23dbfb3e42d37d55fb26e027d2021a5e765a..1ce27f46136e4f3b4a7aeefe49ff0dcfbaba515d 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 2ed634265039044f2bab6576aa5b48d9eda64796..9a4ad4ce09f3222329467b7fc48e8aaa0e504db0 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 471ca1a5bd0b50073e37d5f1adc4684dd7a0bef2..501c47e853e110c839cd21cc35ba78519e2d67d2 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 d652b36c1afc6bb5ea7d7af878a414f745091777..576b38e2d6497e4ed47d5ba0bbce740339ca9088 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 80ee22e76e6bc44799dd35b16890bee417ee3caa..5e19aade21981199f44a08352ef39d4f11e755e0 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 ca3f6556cf7bc04b3fcb50244dbf89c14e4852b2..ff4c88675d960c62e82b2c2c08bd21437a4db5eb 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 6ffcdba6a6ed306adf39d41b24f9b20c9f94af03..b1c05ed49facb066de3d31f1159696a609a3f110 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 854ee6ca018e4c87ee79f55a838fbc0225bd6478..f91f3ddfe0607b42b4bbb34f181032a97b468b39 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 5b5d85c07ea415ebb259705f872a6462cf1a2c5c..eee491ee067a1f88da9b2b5863d15ba7dbd58cdd 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 70364dc1116f8fb4fc53159ca0049f688de0cd62..0e6ea46d3aa182b81a882be957bd4a0bfdd90121 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 6406060359aa99986216f70f2938439e96785db1..ab09671f7cd803d9ff20312a41d3f2c30224445b 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 54bba3b9e4f030c07644ec3cbf94bc06e5375a80..68e5d425358020802d9cf99b997083a880b3ae63 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 0000000000000000000000000000000000000000..338d9793c47550fbae8f79c62efca9f62edef2d9 --- /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 a5a2beab97f22aed3b14538bc5a430ab841001aa..469718dce762b9d459d88c1485d9332cda73d85c 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 cf2b21bc6f74110ffe498dac6f12b0ac06b126b0..8a6513c8b7e4d81d1e08e2796eae1da303444653 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 877d068dc64cfd5dd7552bf95c7e1bc54c203837..11a67ccaae2d76f13740aa684b938a266d03c690 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 07fc66c79388db70f568b70d7c31ca106f97e9c0..7631e34b82c20197a4eabed823649a4d6f661712 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 695bfda790adb0a5b919fe3e43bc52e1f44aea7e..637207ef0d1c8278cf7154205cded42102d12324 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 80803495aa1ebd4d56d054f632e0f436fb3cb1ec..39387130c4170d074f2929f4d5c7bd42ba0fd781 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 415494e3d60baa98a1704295c274229f994f86a9..89ffd5ebd9e539cfd224e10b10a088797ca586d7 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 bc3b9cccf7a8916c3bb4e987b426aac78dee7cb8..1014d1f1ddcfb78f457722b3112a40396e9e2c8c 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 9b76ccaf1c8338bd131e16f48ee74718f894b565..cd800961057e6bef5bdecb8d01fba27228024bf5 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 a6067d69dd3de840e0f0241d17f840a3f62665c0..7a5890340cbb27bd599d63ba951b42faf23464a1 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 effa300f3953faac3398ffa914386e96c033123a..39814f3e7fbc9fad02cfa97bd4c5d844859974f8 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 d11dbe2047de0523243d172be86ffc5700c97e9c..b3afb274c33d00497b626daec8873b378acf76c8 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 a6fa63b4e14af739e101ad9f090628f5ea8ca336..8cd1375bf5d8dc251f596dcac8eb4c5858bcd25b 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 0e606873f10b2d0904b863ca3331d651e6693d57..a2fef06980cf580d006e8520481fbc4e583092ca 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 364ac88e5059e9b56d54f929d35c28a69b472949..e29cad76c225d43de1c05b0ad2a01362d5c024a2 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 9ac863ad144c854a79624df935a22fbd3756b097..0794af25e31cdb2c5be99d043837c3a8d1c8633c 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 52c2d3440032a02dc605498ed2a3d0b062fd50fd..c02b26d32f3ca7ec26907d5648ee53bdc282d32a 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 bfb332c258f837f1dbcd8435294151c2b22f7346..91b2663cff81b316be63a8cdade84922f3c2e23f 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 9954f6cc403dc1dca48eb0c85c5ac4a63255c569..87f118bd327af3c9b81250a93c7507d73a2db3f0 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 77daea639cb94aef2688d634124dbe638b551cca..949259f8b99b44920477a8981a0671681300aed1 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 bd6d80fb33f3eabc684829e1c0dbc9ac8df0b3f9..f055782f1037e63bfc3fd0494fd9cfe33609e123 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 0000000000000000000000000000000000000000..d26e86428dc167144d2cbcb85f2c5d656642a2b3 --- /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 0000000000000000000000000000000000000000..efdf9b5a173259951d1b40cd835cb7a3f1f1f24a --- /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 0000000000000000000000000000000000000000..2875baa0b24ee7f90340a686842b2bec3aef92eb --- /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 0000000000000000000000000000000000000000..b010c2dc15bb06faf222afd86c91624dea41c26a --- /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 0000000000000000000000000000000000000000..1972cf0c42592493cd7f25d816f34279962170e1 --- /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 0000000000000000000000000000000000000000..3851ac9cca3a29c79508c0f344afda15987c6d52 --- /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 0000000000000000000000000000000000000000..abe9c6524cdf22a595ce67a6a219dfb01058a831 --- /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 0000000000000000000000000000000000000000..5f61e68267e1b79ae8645023187b1e2db9b46da1 --- /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 0000000000000000000000000000000000000000..8af93e8bda9ee73d08e70c5daba81e141dd0ea1e --- /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 0000000000000000000000000000000000000000..93234c2f71c1422984a669b83954782b61be1bba --- /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 fa493777bb03407b8c4ce8e57b89f847347fc0a8..167e2a059977c6a92eb4437bd160052dc807b170 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 91e2868dc81595bb5ac6adb59eed5eb869a8a0e1..b20d8d1d81943a8ce908d3757bd2a69705d64ae9 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 6b78b6b7c2ce752619b710b21f8e7a69979fc255..81208aca8a6dcd5e176924cd879607171bf39322 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 e0ae18135cd56dcefbe475761076a46cf5b30c7e..b896ef725fc23880fffbfefb829e2690087f2ce0 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 b7c7356366ed050fe588f9700bd3e07d88807a8b..43dd6887af0c0c5627e94e975db0260aa88d1b0b 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 c95d1807418a1b81faadd510305b99c43a51f7ee..ed667e39bb1e8a4655be6eec0c7e1e71eeea52e9 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 92886d1ae309aec7948bce02bfdbe2a45143f37a..cc281988f90e271bc16aa98f093e8daf8cc46bcc 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 762ee7e4adc42af67692c710fd264997e3eeb0e6..6f0fc4b590fbbc25360203b16e64e0b88f76dc76 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 b3152d414af63d3e02f65ac0f5d8560d37d74d29..f65e3ac3500fafef5082ec2badad8233a07bf741 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 8b9ecca756ba04730fadcb6d4e726f1ebb492130..5315ffb914a7e8c8d0a4e9adb358d8b7d5c11d75 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 f8a020ff5ea56599363d59d8494ad85325c64f09..dafb24ff4d2255a838f9d1f7e36b7ce70e015ea1 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 02c2e82f4f21af97446a7dc88fa5d8b6c9d9bd64..6931134bae41369c6217e5cd40e40f5f1c1ce22d 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 63be0b36218ae64b6a085dfa30a92720ec23aabb..6c09bb1d080b9c96a1d0bdabd8f5a35ada50688a 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 e6c80a5e29a05dbdf2e7cf20fd60e64e48166975..454f4fa797734193e237a8f3fa196657c46a6249 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 5f32ef5baaacba6551732b5256ac91bb5a5cb1f7..6d8901f9ba8afd0fe85fa97f7f1bdbd2b90ecd80 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 71e73048e1a0c38e586a9f5855816075f807a0cc..29688f569399ad0532b0410d8bb4ed7995f4e0f9 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 360050656471ed35a37742f1fecf7fd4e91431c3..eaac6908dbb9f5c6f394644da438c0a4f8ca60f3 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 0000000000000000000000000000000000000000..f559dd85a7ed2a35cb32b2560e476ae16a799040 --- /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 fd007593be6b47490c8fd09210122071eab0e923..ca33288250addaf4cb1ae2de505508e3dbeef0ad 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