mmdeploy
mmdeploy copied to clipboard
support multi-batch test in profile tool
Motivation
Support batch_size > 1 for tools/profile.py, requested in issue #861 .
Modification
- update
create_inputfor batch images in mmcls, mmdet, mmpose, mmseg, mmocr, mmrotate. - update
tools/profile.pyto support testing withbatch_size>1
BC-breaking (Optional)
None
Use cases (Optional)
If this PR introduces a new feature, it is better to list some use cases here, and update the documentation.
Checklist
- Pre-commit or other linting tools are used to fix the potential lint issues.
- The modification is covered by complete unit tests. If not, please add more unit tests to ensure the correctness.
- If the modification has a dependency on downstream projects of a newer version, this PR should be tested with all supported versions of downstream projects.
- The documentation has been modified accordingly, like docstring or example tutorials.