티스토리 뷰
import timm
model = timm.create_model('vit_small_patch32_384', pretrained=pretrained, num_classes=0, img_size=[640, 640])
img_size 를 정의해주면 된다.
https://github.com/huggingface/pytorch-image-models/blob/main/timm/models/vision_transformer.py#L389
GitHub - huggingface/pytorch-image-models: PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, E
PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more - GitHub - hugg...
github.com
'Deep Learning' 카테고리의 다른 글
| ultralytics(YOLOv8) 8.0.114 흥미로운 패치 (0) | 2023.06.09 |
|---|---|
| CLIP 비교 벤치마크 자료 (0) | 2023.06.07 |
| Hockey Fight Detection Dataset (0) | 2023.05.21 |
| [Deep Learning] Vanilla DETR의 수렴 속도를 확실히 체감할 수 있는 그래프 (1) | 2023.05.08 |
| [Deep Learning] DETR 전처리 과정(padding, bounding box normalization)에 생긴 의문점 (1) | 2023.05.06 |
공지사항
최근에 올라온 글
최근에 달린 댓글
- Total
- Today
- Yesterday
링크
TAG
- 가장 긴 증가하는 부분 수열
- 이분탐색
- LCA
- 인공지능을 위한 선형대수
- 순열
- 단축키
- 문제집
- PyCharm
- cosine
- 파이참
- 백준
- 백준 1766
- FairMOT
- C++ Deploy
- Lowest Common Ancestor
- ㅂ
- 백준 11437
- 자료구조
- MOT
- 조합
- 위상 정렬 알고리즘
- 백준 11053
- 백트래킹
| 일 | 월 | 화 | 수 | 목 | 금 | 토 |
|---|---|---|---|---|---|---|
| 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 |
| 31 |
글 보관함