4paradigm-CV / SE-STAD

8 stars 1 forks source link

A question about classification head #6

Open nichorians opened 6 months ago

nichorians commented 6 months ago

Hi,I wounder why the input chaennel is 4608 in your model's classification head. Hope for your respond, thanks. /SE-STAD-main/mmaction2/configs/detection/sestad/e2e/slowfast_k400_r50_fpn_fcos.py roi_head=dict( type='AVARoIHead', bbox_roi_extractor=dict( type='SingleRoIExtractor3D', roi_layer_type='RoIAlign', output_size=8, with_temporal_pool=True, with_global=True), bbox_head=dict( type='BBoxHeadAVA', in_channels=4608, num_classes=81, multilabel=True, dropout_ratio=0.5, loss_weight=10.0), ),