Skip to content

Latest commit

 

History

History
27 lines (18 loc) · 877 Bytes

code.md

File metadata and controls

27 lines (18 loc) · 877 Bytes

adamixer-code structure

Our code is based on mmdetection and follows the default structure of mmdetection.

AdaMixer related config files can be found in configs/adamixer/ folder.

AdaMixer related model code includes:

mmdet/models/roi_heads/adamixer_decoder.py
### AdaMixer decoder general framework

mmdet/models/roi_heads/bbox_heads/adamixer_decoder_stage.py
### AdaMixer decoder stage depicted in Figure 4

mmdet/models/roi_heads/bbox_heads/adaptive_mixing_operator.py
### implementation of adaptive channel and spatial mixing

mmdet/models/roi_heads/bbox_heads/msaq.py
### 3D sampling implementation

mmdet/models/dense_heads/query_generator.py
### initial query generator

You can resort to Makefile targets to train models.

Note that some variable names here are temporary names for developing, which might slightly differ from names in the paper.