timm 모델 피쳐맵 채널 사이즈 반환 방법

https://huggingface.co/docs/timm/feature_extraction#query-the-feature-information

Feature Extraction

 

공식 문서에 나온대로 하면 몇몇 네트워크의 경우는 아래와 같은 아래가 발생할 것이다.

 

예를 들면 convnext_tiny 모델을 load해보면 바로 알 것이다.

 

feature_info 출력해보면 원소가 딕셔너리인 리스트이다.

 

만약 내가 마지막 레이어의 피쳐맵을 알고 싶다.

 

그러면,

 

모델.feature_info[-1]['num_chs'] 하면 된다.

 

(Pdb) self.backbone.feature_info[-1]
{'num_chs': 768, 'reduction': 32, 'module': 'stages.3'}
(Pdb) self.backbone.feature_info[-1]['num_chs']
768