mmrazor/tools/visualizations/vis_configs/fpn_feature_visualization.py

9 lines
322 B
Python

# Copyright (c) OpenMMLab. All rights reserved.
recorders = dict(
neck=dict(_scope_='mmrazor', type='ModuleOutputs', source='neck'))
mappings = dict(
p3=dict(recorder='neck', data_idx=0),
p4=dict(recorder='neck', data_idx=1),
p5=dict(recorder='neck', data_idx=2),
p6=dict(recorder='neck', data_idx=3))