You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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
The text was updated successfully, but these errors were encountered:
Is your feature request related to a problem? Please describe.
PyTorch models expect inputs to be provided as channels-first. The
PyTorchObjectDetector
,PyTorchFasterRCNN
andPyTorchYolo
models have achannels_first
parameter which will permute the input accordingly such that it is accepted by the model. ThePyTorchObjectDetector
andPyTorchFasterRCNN
models default this toFalse
which is not the expected behavior.Describe the solution you'd like
Change the
channels_first
parameter inPyTorchObjectDetector
andPyTorchFasterRCNN
to default toTrue
instead ofFalse
. This is the expected behavior if following torchvision tutorials. Additionally, theinput_shape
parameter (although unused) should be adjusted accordingly.Describe alternatives you've considered
N/A
Additional context
N/A
The text was updated successfully, but these errors were encountered: