Skip to content

Instantly share code, notes, and snippets.

@arunm8489

arunm8489/9. detection

Created June 3, 2020 15:03
Show Gist options
  • Save arunm8489/82c9a91baa879da4834c700a1a0ee45b to your computer and use it in GitHub Desktop.
Save arunm8489/82c9a91baa879da4834c700a1a0ee45b to your computer and use it in GitHub Desktop.
Download ZIP
Raw
9. detection
class DetectionLayer(nn.Module):
def __init__(self, anchors):
super(DetectionLayer, self).__init__()
self.anchors = anchors
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment