티스토리 뷰
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
- 파이참
- cosine
- 백준 11053
- MOT
- Lowest Common Ancestor
- 순열
- 문제집
- 이분탐색
- LCA
- 백트래킹
- 인공지능을 위한 선형대수
- ㅂ
- 자료구조
- 위상 정렬 알고리즘
- 백준 1766
- 백준 11437
- 단축키
- PyCharm
- FairMOT
- C++ Deploy
- 조합
- 가장 긴 증가하는 부분 수열
- 백준
| 일 | 월 | 화 | 수 | 목 | 금 | 토 |
|---|---|---|---|---|---|---|
| 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 |
글 보관함