timm 모델 피쳐맵 채널 사이즈 반환 방법
https://huggingface.co/docs/timm/feature_extraction#query-the-feature-information Feature Extraction huggingface.co 공식 문서에 나온대로 하면 몇몇 네트워크의 경우는 아래와 같은 아래가 발생할 것이다. 예를 들면 convnext_tiny 모델을 load해보면 바로 알 것이다. feature_info 출력해보면 원소가 딕셔너리인 리스트이다. 만약 내가 마지막 레이어의 피쳐맵을 알고 싶다. 그러면, 모델.feature_info[-1]['num_chs'] 하면 된다. (Pdb) self.backbone.feature_info[-1]{'num_chs': 768, 'reduction': 32, 'module': 'stage..
카테고리 없음
2024. 11. 15. 22:06
공지사항
최근에 올라온 글
최근에 달린 댓글
- Total
- Today
- Yesterday
링크
TAG
- 백준 1766
- 문제집
- 백준
- 가장 긴 증가하는 부분 수열
- 인공지능을 위한 선형대수
- 백준 11437
- 자료구조
- 순열
- 위상 정렬 알고리즘
- C++ Deploy
- 백준 11053
- 이분탐색
- 백트래킹
- 조합
- 파이참
- PyCharm
- ㅂ
- MOT
- LCA
- Lowest Common Ancestor
- 단축키
- cosine
- FairMOT
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
1 | 2 | |||||
3 | 4 | 5 | 6 | 7 | 8 | 9 |
10 | 11 | 12 | 13 | 14 | 15 | 16 |
17 | 18 | 19 | 20 | 21 | 22 | 23 |
24 | 25 | 26 | 27 | 28 | 29 | 30 |
글 보관함