Open m-kashani opened 4 years ago
├─layers <- custom layers e.g. deformable conv. https://github.com/facebookresearch/detectron2/tree/master/detectron2/layers
Meta Architecture:
GeneralizedRCNN (meta_arch/rcnn.py)
which has:
Backbone Network:
FPN (backbone/fpn.py)
└ ResNet (backbone/resnet.py)
Region Proposal Network:
RPN(proposal_generator/rpn.py)
├ StandardRPNHead (proposal_generator/rpn.py)
└ RPNOutput (proposal_generator/rpn_outputs.py)
ROI Heads (Box Head):
StandardROIHeads (roi_heads/roi_heads.py)
├ ROIPooler (poolers.py)
├ FastRCNNConvFCHead (roi_heads/box_heads.py)
├ FastRCNNOutputLayers (roi_heads/fast_rcnn.py)
└ FastRCNNOutputs (roi_heads/fast_rcnn.py)
print(outputs["instances"].pred_classes) print(outputs["instances"].pred_boxes)
First figure out your top 20 images and save them accordingly.
To read and understand the pipeline:
[x] Digging into Detectron Part1:
[ ] https://medium.com/@hirotoschwert/digging-into-detectron-2-part-2-dd6e8b0526e
[ ] https://medium.com/@hirotoschwert/digging-into-detectron-2-part-3-1ecc27efc0b2
[ ] Proposal Generator?
[ ] Box Head.
extra Amir