Skip to content

Conversation

@p-dot-max
Copy link

@WongKinYiu

Fixes an AttributeError in the feature_visualization function where a tuple was passed instead of a tensor. Added a check to handle tuple inputs by extracting the first element, ensuring compatibility with different layer outputs in YOLOv9

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant