• Sara Beery's avatar
    Context R-CNN Updates: Added capabilities to use multi-headed or multi-layered... · 30b1c958
    Sara Beery authored
    Context R-CNN Updates: Added capabilities to use multi-headed or multi-layered attention, to place the attention heads pre- or post-second stage feature extraction, and to work with embedded features in the context feature back from pre- or post-second stage feature extraction. Added an option for RPN feature map crops to be piped through to model outputs.
    
    PiperOrigin-RevId: 345777219
    30b1c958
model_builder.py 47.9 KB