Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Change the channels_first parameter in PyTorchObjectDetector and PyTorchFasterRCNN to True #2264

Closed
f4str opened this issue Aug 30, 2023 · 0 comments · Fixed by #2265
Closed
Assignees
Labels
bug Something isn't working
Milestone

Comments

@f4str
Copy link
Collaborator

f4str commented Aug 30, 2023

Is your feature request related to a problem? Please describe.
PyTorch models expect inputs to be provided as channels-first. The PyTorchObjectDetector, PyTorchFasterRCNN and PyTorchYolo models have a channels_first parameter which will permute the input accordingly such that it is accepted by the model. The PyTorchObjectDetector and PyTorchFasterRCNN models default this to False which is not the expected behavior.

Describe the solution you'd like
Change the channels_first parameter in PyTorchObjectDetector and PyTorchFasterRCNN to default to True instead of False. This is the expected behavior if following torchvision tutorials. Additionally, the input_shape parameter (although unused) should be adjusted accordingly.

Describe alternatives you've considered
N/A

Additional context
N/A

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
2 participants