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
This issue relates to the implementation in downstream/object_detection/dino/models/dino /fastervit.py. I'm using this FasterViT implementation as a backbone for my own object detector, so I only needed the forward_raw() method. However this method leads to an error because the output format is incorrect, due to an unnecessary permutation.
This line I'm guessing was supposed to permute NHWC-> NCHW, but the output from the layers is already in NCHW format so this actually permutes it from NCHW -> NWCH.
# Bug: x_out already NCHW, so this permutes to NWCHouts.append(x_out.permute(0, 3, 1, 2).contiguous())
These lines should simply be deleted and it will work. I don't see this function actually being used anywhere, hence why it wasn't caught.
The text was updated successfully, but these errors were encountered:
collinmccarthy
changed the title
Bug in downstream forward_raw() methods (incorrect permutation)
[Bug]: Downstream forward_raw() methods have incorrect permutation
Jan 22, 2025
This issue relates to the implementation in
downstream/object_detection/dino/models/dino /fastervit.py
. I'm using this FasterViT implementation as a backbone for my own object detector, so I only needed theforward_raw()
method. However this method leads to an error because the output format is incorrect, due to an unnecessary permutation.This line I'm guessing was supposed to permute NHWC-> NCHW, but the output from the layers is already in NCHW format so this actually permutes it from NCHW -> NWCH.
Similarly here in the
motrv2
models.These lines should simply be deleted and it will work. I don't see this function actually being used anywhere, hence why it wasn't caught.
The text was updated successfully, but these errors were encountered: