Skip to content

Instantly share code, notes, and snippets.

@AlessandroMondin

AlessandroMondin/backbone.py

Last active December 20, 2022 20:01
Show Gist options
  • Save AlessandroMondin/b67e84b0ce9eed0426c10d46ae1aea07 to your computer and use it in GitHub Desktop.
Save AlessandroMondin/b67e84b0ce9eed0426c10d46ae1aea07 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
backbone.py
self.backbone += [
CBL(in_channels=3, out_channels=first_out, kernel_size=6, stride=2, padding=2),
CBL(in_channels=first_out, out_channels=first_out*2, kernel_size=3, stride=2, padding=1),
C3(in_channels=first_out*2, out_channels=first_out*2, width_multiple=0.5, depth=2),
CBL(in_channels=first_out*2, out_channels=first_out*4, kernel_size=3, stride=2, padding=1),
C3(in_channels=first_out*4, out_channels=first_out*4, width_multiple=0.5, depth=4),
CBL(in_channels=first_out*4, out_channels=first_out*8, kernel_size=3, stride=2, padding=1),
C3(in_channels=first_out*8, out_channels=first_out*8, width_multiple=0.5, depth=6),
CBL(in_channels=first_out*8, out_channels=first_out*16, kernel_size=3, stride=2, padding=1),
C3(in_channels=first_out*16, out_channels=first_out*16, width_multiple=0.5, depth=2),
SPPF(in_channels=first_out*16, out_channels=first_out*16)
]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment