Skip to content

Instantly share code, notes, and snippets.

@BloodAxe

BloodAxe/retina.py

Last active Aug 19, 2020
Embed
What would you like to do?
# 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