fix: add missing device argument in image_demo #1947
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Motivation
The motivation for this PR is to fix a missing
device
argument in the initialization ofImageClassificationInferencer
in theimage_demo.py
script. This fix ensures that the correct device (e.g., CPU or GPU) is specified when running the image classification inference, improving the flexibility and functionality of the script.Modification
This PR adds the
device
argument to theImageClassificationInferencer
initialization inimage_demo.py
. The code change ensures that the model can be run on the specified device provided by the user through the command line arguments.BC-breaking (Optional)
This modification does not introduce any backward compatibility issues. The change is additive, providing additional functionality without altering existing behavior.
Use cases (Optional)
No new features are introduced in this PR, so no specific use cases are added. However, this fix allows users to specify the computation device for inference, which can be crucial for running models on different hardware setups.
Checklist
Before PR:
After PR: