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

Add TFDWConv() depth_multiplier arg #7858

Merged
merged 1 commit into from
May 17, 2022
Merged

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented May 17, 2022

Enables grouped non c1 == c2 convolutions in TF YOLOv5 models.

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhanced compatibility for TensorFlow depth-wise convolution layers in YOLOv5 models.

πŸ“Š Key Changes

  • Adjusted the assertion in the TFDWConv layer to allow the output channels to be a multiple of the input channels, not strictly equal.
  • Added depth_multiplier argument to the DepthwiseConv2D layer to handle the case where output channels are multiple of input channels.

🎯 Purpose & Impact

  • 🎨 Flexibility: Allows more flexibility in model design by supporting depth-wise convolutions with varied channel proportions.
  • πŸŽ“ Compatibility: Ensures broader compatibility with different model architectures that use depth-wise convolutions.
  • πŸš€ Potential Impact: Users can design and train more complex models within TensorFlow using YOLOv5, potentially improving performance on various tasks.

Enabled grouped non c1 == c2 convolutions in TF YOLOv5 models.
@glenn-jocher glenn-jocher self-assigned this May 17, 2022
@glenn-jocher glenn-jocher changed the title Add TFDWConv() depth_multiplier Add TFDWConv() depth_multiplier arg May 17, 2022
@glenn-jocher
Copy link
Member Author

Validated

Screenshot 2022-05-17 at 15 42 08

@glenn-jocher glenn-jocher merged commit 1b8e70f into master May 17, 2022
@glenn-jocher glenn-jocher deleted the update/depth_multiplier branch May 17, 2022 13:42
tdhooghe pushed a commit to tdhooghe/yolov5 that referenced this pull request Jun 10, 2022
Enabled grouped non c1 == c2 convolutions in TF YOLOv5 models.
ctjanuhowski pushed a commit to ctjanuhowski/yolov5 that referenced this pull request Sep 8, 2022
Enabled grouped non c1 == c2 convolutions in TF YOLOv5 models.
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.

None yet

1 participant