open-mmlab / mmdeploy

OpenMMLab Model Deployment Framework
https://mmdeploy.readthedocs.io/en/latest/
Apache License 2.0
2.76k stars 634 forks source link

An exception occurred in the inference result of getting mask RCNN with object_detection.py #969

Open shoayi opened 2 years ago

shoayi commented 2 years ago

I use tools/deploy.py to get the TensorRT model and the configuration information is as follows: mmdeploy\configs\mmdet\instance-seg\instance-seg_tensorrt-fp16_static-800x1344.py mmdetection\configs\mask_rcnn\mask_rcnn_x101_64x4d_fpn_mstrain-poly_3x_coco.py mmdetection\checkpoints\mask_rcnn_x101_64x4d_fpn_mstrain-poly_3x_coco_20210526_120447-c376f129.pth mmdetection\demo\demo.jpg --work-dir work_dirs/mask_rcnn/ --device cuda --show --dump-info

i through object_detection.py, resulting in output_detection.png that is different from the example. The mask is displayed in the same shape as the real object, but with a much smaller area.The configuration information about object_detect. py is as follows: cuda mmdetection\mmdeploy\work_dirs\mask_rcnn\ mmdetection\demo\demo.jpg

output_detection output_detection

tpoisonooo commented 2 years ago

@lvhan028

lvhan028 commented 2 years ago

Could you please post the result of python tools/check_env.py?

shoayi commented 2 years ago

Thank you for your reply. The results are as follows: 2022-09-01 09:53:36,393 - mmdeploy - INFO -

2022-09-01 09:53:36,393 - mmdeploy - INFO - **Environmental information** 2022-09-01 09:53:38,112 - mmdeploy - INFO - sys.platform: win32 2022-09-01 09:53:38,113 - mmdeploy - INFO - Python: 3.7.13 (default, Mar 28 2022, 08:03:21) [MSC v.1916 64 bit (AMD64)] 2022-09-01 09:53:38,113 - mmdeploy - INFO - CUDA available: True 2022-09-01 09:53:38,113 - mmdeploy - INFO - GPU 0: NVIDIA GeForce RTX 3060 2022-09-01 09:53:38,113 - mmdeploy - INFO - CUDA_HOME: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3 2022-09-01 09:53:38,113 - mmdeploy - INFO - NVCC: Cuda compilation tools, release 11.3, V11.3.58 2022-09-01 09:53:38,113 - mmdeploy - INFO - MSVC: 用于 x64 的 Microsoft (R) C/C++ 优化编译器 19.29.30040 版 2022-09-01 09:53:38,113 - mmdeploy - INFO - GCC: n/a 2022-09-01 09:53:38,113 - mmdeploy - INFO - PyTorch: 1.11.0 2022-09-01 09:53:38,113 - mmdeploy - INFO - PyTorch compiling details: PyTorch built with:

2022-09-01 09:53:38,113 - mmdeploy - INFO - TorchVision: 0.12.0 2022-09-01 09:53:38,113 - mmdeploy - INFO - OpenCV: 4.6.0 2022-09-01 09:53:38,113 - mmdeploy - INFO - MMCV: 1.6.1 2022-09-01 09:53:38,113 - mmdeploy - INFO - MMCV Compiler: MSVC 192930140 2022-09-01 09:53:38,113 - mmdeploy - INFO - MMCV CUDA Compiler: 11.3 2022-09-01 09:53:38,113 - mmdeploy - INFO - MMDeploy: 0.7.0+18321ee 2022-09-01 09:53:38,113 - mmdeploy - INFO -

2022-09-01 09:53:38,113 - mmdeploy - INFO - **Backend information** 2022-09-01 09:53:38,736 - mmdeploy - INFO - onnxruntime: 1.8.1 ops_is_avaliable : False 2022-09-01 09:53:38,754 - mmdeploy - INFO - tensorrt: 8.4.3.1 ops_is_avaliable : True 2022-09-01 09:53:38,778 - mmdeploy - INFO - ncnn: None ops_is_avaliable : False 2022-09-01 09:53:38,779 - mmdeploy - INFO - pplnn_is_avaliable: False 2022-09-01 09:53:38,780 - mmdeploy - INFO - openvino_is_avaliable: False 2022-09-01 09:53:38,810 - mmdeploy - INFO - snpe_is_available: False 2022-09-01 09:53:38,810 - mmdeploy - INFO -

2022-09-01 09:53:38,810 - mmdeploy - INFO - **Codebase information** 2022-09-01 09:53:38,812 - mmdeploy - INFO - mmdet: 2.25.1 2022-09-01 09:53:38,812 - mmdeploy - INFO - mmseg: None 2022-09-01 09:53:38,812 - mmdeploy - INFO - mmcls: None 2022-09-01 09:53:38,812 - mmdeploy - INFO - mmocr: None 2022-09-01 09:53:38,812 - mmdeploy - INFO - mmedit: None 2022-09-01 09:53:38,812 - mmdeploy - INFO - mmdet3d: None 2022-09-01 09:53:38,812 - mmdeploy - INFO - mmpose: None 2022-09-01 09:53:38,812 - mmdeploy - INFO - mmrotate: None

Process finished with exit code 0