Navigation Menu

Skip to content

Instantly share code, notes, and snippets.

@BloodAxe

BloodAxe/retina.py

Last active August 19, 2020 08:51
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save BloodAxe/12acd8336cc67811887d2eb8f183d000 to your computer and use it in GitHub Desktop.
Save BloodAxe/12acd8336cc67811887d2eb8f183d000 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
retina.py
# Bad practice, don't return tuple
class RetinaNet(nn.Module):
...
def forward(self, image):
x = self.encoder(image)
x = self.decoder(x)
bboxes, scores = self.head(x)
return bboxes, scores
...
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment