timm documentation
ConvNeXt
ConvNeXt
Kategori: CNN · Kode: timm/models/convnext.py · Arsitektur: 31 · Bobot pralatih: 101 (61 dengan hasil ImageNet-1k)
ConvNeXt adalah jaringan konvolusional murni yang mengadopsi pilihan desain dari vision transformer, seperti kernel depthwise besar, inverted bottleneck, dan LayerNorm. ConvNeXt-V2 menambahkan Global Response Normalization (GRN) dan dipretrain sebagai fully convolutional masked autoencoder (FCMAE).
Makalah
- ConvNet untuk 2020-an
- ConvNeXt V2: Merancang dan Mengembangkan ConvNet dengan Masked Autoencoders
- Hukum penskalaan yang dapat direproduksi untuk pembelajaran bahasa-gambar kontrastif
- DINOv3
- Encoder Persepsi Universal yang Efisien
Penggunaan
Lihat Quickstart untuk memuat bobot terlatih, prapemrosesan gambar, dan penyesuaian halus, serta Ekstraksi Fitur untuk menggunakan model sebagai tulang punggung.
>>> import timm
>>> model = timm.create_model('convnext_tiny.in12k_ft_in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)Dukungan API Model
✓ didukung, ✓ (persegi) hanya berfungsi untuk ukuran persegi, ✗ tidak didukung, – tidak diperlukan (model sudah menerima ukuran input apa pun). Diperiksa dengan menjalankan setiap fitur pada model yang diinisialisasi secara acak (ConvNeXt dengan convnext_zepto_rms pada 224), arsitektur lain dari kelas yang sama diharapkan berperilaku sama. Lihat halaman Model API untuk detailnya.
Arsitektur
| Arsitektur | Parameter (M) | GMACs | Ukuran input | Bobot terlatih |
|---|---|---|---|---|
convnext_atto | 3.70 | 0.55 | 224 | 1 |
convnext_atto_ols | 3.70 | 0.58 | 224 | 1 |
convnext_atto_rms | 3.69 | 0.55 | 224 | 0 |
convnext_base | 88.59 | 15.38 | 224 | 16 |
convnext_femto | 5.22 | 0.79 | 224 | 1 |
convnext_femto_ols | 5.23 | 0.82 | 224 | 1 |
convnext_large | 197.77 | 34.40 | 224 | 5 |
convnext_large_mlp | 200.13 | 70.21 | 320 | 10 |
convnext_nano | 15.59 | 2.46 | 224 | 6 |
convnext_nano_ols | 15.65 | 2.65 | 224 | 1 |
convnext_pico | 9.05 | 1.37 | 224 | 1 |
convnext_pico_ols | 9.06 | 1.43 | 224 | 1 |
convnext_small | 50.22 | 8.71 | 224 | 9 |
convnext_tiny | 28.59 | 4.47 | 224 | 9 |
convnext_tiny_hnf | 28.59 | 4.47 | 224 | 1 |
convnext_xlarge | 350.20 | 60.98 | 224 | 3 |
convnext_xxlarge | 846.47 | 198.09 | 256 | 4 |
convnext_zepto_rms | 2.16 | 0.30 | 224 | 1 |
convnext_zepto_rms_ols | 2.16 | 0.34 | 224 | 1 |
convnextv2_atto | 3.71 | 0.55 | 224 | 2 |
convnextv2_base | 88.72 | 15.38 | 224 | 4 |
convnextv2_femto | 5.23 | 0.79 | 224 | 2 |
convnextv2_huge | 660.29 | 337.96 | 384 | 4 |
convnextv2_large | 197.96 | 34.40 | 224 | 4 |
convnextv2_nano | 15.62 | 2.46 | 224 | 4 |
convnextv2_pico | 9.07 | 1.37 | 224 | 2 |
convnextv2_small | 50.32 | 8.71 | 224 | 0 |
convnextv2_tiny | 28.64 | 4.47 | 224 | 4 |
test_convnext | 0.27 | 0.03 | 160 | 1 |
test_convnext2 | 0.48 | 0.05 | 160 | 1 |
test_convnext3 | 0.47 | 0.05 | 160 | 1 |
GMACs dan jumlah parameter adalah untuk ukuran input dan klasifer default dari setiap arsitektur, dari hasil benchmark.
Bobot terlatih
Top-1 adalah akurasi validasi ImageNet-1k dari results-imagenet.csv pada ukuran eval yang tercantum, - jika bobot tidak memiliki hasil ImageNet-1k (misalnya bobot pretraining atau image-text). Classes adalah ukuran dari classifier pretrained, none untuk bobot yang tidak memilikinya.
Bobot dikelompokkan berdasarkan sumber: paper atau asal yang diberikan dalam konfigurasi pretrained, jika tidak ada maka organisasi Hub selain timm, jika tidak ada maka bagian pertama dari tag pretrained (architecture.tag). Sumber dengan satu bobot terdaftar di bawah
other.
ConvNeXt V2
Konfigurasi pretrained merujuk pada ConvNeXt V2: Co-designing and Scaling ConvNets with Masked Autoencoders.
| Bobot | Ukuran input | Top-1 (%) | Ukuran evaluasi | Kelas | Lisensi | Hub |
|---|---|---|---|---|---|---|
convnextv2_atto.fcmae | 224 | - | - | none | cc-by-nc-4.0 | timm |
convnextv2_atto.fcmae_ft_in1k | 224 (uji 288) | 77.794 | 288 | 1000 | cc-by-nc-4.0 | timm |
convnextv2_base.fcmae | 224 | - | - | none | cc-by-nc-4.0 | timm |
convnextv2_base.fcmae_ft_in1k | 224 (uji 288) | 85.490 | 288 | 1000 | cc-by-nc-4.0 | timm |
convnextv2_base.fcmae_ft_in22k_in1k | 224 (uji 288) | 86.994 | 288 | 1000 | cc-by-nc-4.0 | timm |
convnextv2_base.fcmae_ft_in22k_in1k_384 | 384 | 87.638 | 384 | 1000 | cc-by-nc-4.0 | timm |
convnextv2_femto.fcmae | 224 | - | - | none | cc-by-nc-4.0 | timm |
convnextv2_femto.fcmae_ft_in1k | 224 (uji 288) | 79.342 | 288 | 1000 | cc-by-nc-4.0 | timm |
convnextv2_huge.fcmae | 224 | - | - | none | cc-by-nc-4.0 | timm |
convnextv2_huge.fcmae_ft_in1k | 224 (uji 288) | 86.618 | 288 | 1000 | cc-by-nc-4.0 | timm |
convnextv2_huge.fcmae_ft_in22k_in1k_384 | 384 | 88.666 | 384 | 1000 | cc-by-nc-4.0 | timm |
convnextv2_huge.fcmae_ft_in22k_in1k_512 | 512 | 88.860 | 512 | 1000 | cc-by-nc-4.0 | timm |
convnextv2_large.fcmae | 224 | - | - | none | cc-by-nc-4.0 | timm |
convnextv2_large.fcmae_ft_in1k | 224 (uji 288) | 86.132 | 288 | 1000 | cc-by-nc-4.0 | timm |
convnextv2_large.fcmae_ft_in22k_in1k | 224 (uji 288) | 87.494 | 288 | 1000 | cc-by-nc-4.0 | timm |
convnextv2_large.fcmae_ft_in22k_in1k_384 | 384 | 88.180 | 384 | 1000 | cc-by-nc-4.0 | timm |
convnextv2_nano.fcmae | 224 | - | - | none | cc-by-nc-4.0 | timm |
convnextv2_nano.fcmae_ft_in1k | 224 (uji 288) | 82.490 | 288 | 1000 | cc-by-nc-4.0 | timm |
convnextv2_nano.fcmae_ft_in22k_in1k | 224 (uji 288) | 82.666 | 288 | 1000 | cc-by-nc-4.0 | timm |
convnextv2_nano.fcmae_ft_in22k_in1k_384 | 384 | 83.366 | 384 | 1000 | cc-by-nc-4.0 | timm |
convnextv2_pico.fcmae | 224 | - | - | none | cc-by-nc-4.0 | timm |
convnextv2_pico.fcmae_ft_in1k | 224 (uji 288) | 81.062 | 288 | 1000 | cc-by-nc-4.0 | timm |
convnextv2_tiny.fcmae | 224 | - | - | none | cc-by-nc-4.0 | timm |
convnextv2_tiny.fcmae_ft_in1k | 224 (uji 288) | 83.478 | 288 | 1000 | cc-by-nc-4.0 | timm |
convnextv2_tiny.fcmae_ft_in22k_in1k | 224 (uji 288) | 84.428 | 288 | 1000 | cc-by-nc-4.0 | timm |
convnextv2_tiny.fcmae_ft_in22k_in1k_384 | 384 | 85.098 | 384 | 1000 | cc-by-nc-4.0 | timm |
tag klip
| Bobot | Ukuran input | Top-1 (%) | Ukuran evaluasi | Kelas | Lisensi | Hub |
|---|---|---|---|---|---|---|
convnext_base.clip_laion2b | 256 | - | - | 640 | apache-2.0 | timm |
convnext_base.clip_laion2b_augreg | 256 | - | - | 640 | apache-2.0 | timm |
convnext_base.clip_laion2b_augreg_ft_in12k | 256 | - | - | 11821 | apache-2.0 | timm |
convnext_base.clip_laion2b_augreg_ft_in12k_in1k | 256 | 86.378 | 256 | 1000 | apache-2.0 | timm |
convnext_base.clip_laion2b_augreg_ft_in12k_in1k_384 | 384 | 87.144 | 384 | 1000 | apache-2.0 | timm |
convnext_base.clip_laion2b_augreg_ft_in1k | 256 | 86.184 | 256 | 1000 | apache-2.0 | timm |
convnext_base.clip_laiona | 256 | - | - | 640 | apache-2.0 | timm |
convnext_base.clip_laiona_320 | 320 | - | - | 640 | apache-2.0 | timm |
convnext_base.clip_laiona_augreg_320 | 320 | - | - | 640 | apache-2.0 | timm |
convnext_base.clip_laiona_augreg_ft_in1k_384 | 384 | 86.490 | 384 | 1000 | apache-2.0 | timm |
convnext_large_mlp.clip_laion2b_augreg | 256 | - | - | 768 | apache-2.0 | timm |
convnext_large_mlp.clip_laion2b_augreg_ft_in12k_384 | 384 | - | - | 11821 | apache-2.0 | timm |
convnext_large_mlp.clip_laion2b_augreg_ft_in1k | 256 | 87.344 | 256 | 1000 | apache-2.0 | timm |
convnext_large_mlp.clip_laion2b_augreg_ft_in1k_384 | 384 | 87.890 | 384 | 1000 | apache-2.0 | timm |
convnext_large_mlp.clip_laion2b_ft_320 | 320 | - | - | 768 | apache-2.0 | timm |
convnext_large_mlp.clip_laion2b_ft_soup_320 | 320 | - | - | 768 | apache-2.0 | timm |
convnext_large_mlp.clip_laion2b_soup_ft_in12k_320 | 320 | - | - | 11821 | apache-2.0 | timm |
convnext_large_mlp.clip_laion2b_soup_ft_in12k_384 | 384 | - | - | 11821 | apache-2.0 | timm |
convnext_large_mlp.clip_laion2b_soup_ft_in12k_in1k_320 | 320 | 87.970 | 320 | 1000 | apache-2.0 | timm |
convnext_large_mlp.clip_laion2b_soup_ft_in12k_in1k_384 | 384 | 88.334 | 384 | 1000 | apache-2.0 | timm |
convnext_xxlarge.clip_laion2b_rewind | 256 | - | - | 1024 | apache-2.0 | timm |
convnext_xxlarge.clip_laion2b_soup | 256 | - | - | 1024 | apache-2.0 | timm |
convnext_xxlarge.clip_laion2b_soup_ft_in12k | 256 | - | - | 11821 | apache-2.0 | timm |
convnext_xxlarge.clip_laion2b_soup_ft_in1k | 256 | 88.622 | 256 | 1000 | apache-2.0 | timm |
tag fb
| Bobot | Ukuran input | Top-1 (%) | Ukuran evaluasi | Kelas | Lisensi | Hub |
|---|---|---|---|---|---|---|
convnext_base.fb_in1k | 224 (uji 288) | 84.466 | 288 | 1000 | apache-2.0 | timm |
convnext_base.fb_in22k | 224 | - | - | 21841 | apache-2.0 | timm |
convnext_base.fb_in22k_ft_in1k | 224 (uji 288) | 86.302 | 288 | 1000 | apache-2.0 | timm |
convnext_base.fb_in22k_ft_in1k_384 | 384 | 86.826 | 384 | 1000 | apache-2.0 | timm |
convnext_large.fb_in1k | 224 (uji 288) | 84.862 | 288 | 1000 | apache-2.0 | timm |
convnext_large.fb_in22k | 224 | - | - | 21841 | apache-2.0 | timm |
convnext_large.fb_in22k_ft_in1k | 224 (uji 288) | 87.012 | 288 | 1000 | apache-2.0 | timm |
convnext_large.fb_in22k_ft_in1k_384 | 384 | 87.456 | 384 | 1000 | apache-2.0 | timm |
convnext_small.fb_in1k | 224 (uji 288) | 83.704 | 288 | 1000 | apache-2.0 | timm |
convnext_small.fb_in22k | 224 | - | - | 21841 | apache-2.0 | timm |
convnext_small.fb_in22k_ft_in1k | 224 (uji 288) | 85.276 | 288 | 1000 | apache-2.0 | timm |
convnext_small.fb_in22k_ft_in1k_384 | 384 | 85.752 | 384 | 1000 | apache-2.0 | timm |
convnext_tiny.fb_in1k | 224 (uji 288) | 82.694 | 288 | 1000 | apache-2.0 | timm |
convnext_tiny.fb_in22k | 224 | - | - | 21841 | apache-2.0 | timm |
convnext_tiny.fb_in22k_ft_in1k | 224 (uji 288) | 78.950 | 288 | 1000 | apache-2.0 | timm |
convnext_tiny.fb_in22k_ft_in1k_384 | 384 | 84.098 | 384 | 1000 | apache-2.0 | timm |
convnext_xlarge.fb_in22k | 224 | - | - | 21841 | apache-2.0 | timm |
convnext_xlarge.fb_in22k_ft_in1k | 224 (uji 288) | 87.378 | 288 | 1000 | apache-2.0 | timm |
convnext_xlarge.fb_in22k_ft_in1k_384 | 384 | 87.764 | 384 | 1000 | apache-2.0 | timm |
tag d1
| Bobot | Ukuran input | Top-1 (%) | Ukuran evaluasi | Kelas | Lisensi | Hub |
|---|---|---|---|---|---|---|
convnext_femto.d1_in1k | 224 (uji 288) | 78.712 | 288 | 1000 | apache-2.0 | timm |
convnext_femto_ols.d1_in1k | 224 (uji 288) | 78.916 | 288 | 1000 | apache-2.0 | timm |
convnext_pico.d1_in1k | 224 (uji 288) | 80.426 | 288 | 1000 | apache-2.0 | timm |
convnext_pico_ols.d1_in1k | 224 (uji 288) | 80.458 | 288 | 1000 | apache-2.0 | timm |
dinov3 tag
| Bobot | Ukuran input | Top-1 (%) | Ukuran evaluasi | Kelas | Lisensi | Hub |
|---|---|---|---|---|---|---|
convnext_base.dinov3_lvd1689m | 224 | - | - | none | dinov3-license | timm |
convnext_large.dinov3_lvd1689m | 224 | - | - | none | dinov3-license | timm |
convnext_small.dinov3_lvd1689m | 224 | - | - | none | dinov3-license | timm |
convnext_tiny.dinov3_lvd1689m | 224 | - | - | none | dinov3-license | timm |
eupe tags
| Bobot | Ukuran input | Top-1 (%) | Ukuran evaluasi | Kelas | Lisensi | Hub |
|---|---|---|---|---|---|---|
convnext_base.eupe_lvd1689m | 256 | - | - | none | fair-noncommercial-research-license | timm |
convnext_small.eupe_lvd1689m | 256 | - | - | none | fair-noncommercial-research-license | timm |
convnext_tiny.eupe_lvd1689m | 256 | - | - | none | fair-noncommercial-research-license | timm |
tag d1h
| Bobot | Ukuran input | Top-1 (%) | Ukuran evaluasi | Kelas | Lisensi | Hub |
|---|---|---|---|---|---|---|
convnext_nano.d1h_in1k | 224 (uji 288) | 81.474 | 288 | 1000 | apache-2.0 | timm |
convnext_nano_ols.d1h_in1k | 224 (uji 288) | 81.612 | 288 | 1000 | apache-2.0 | timm |
tag ra4
| Bobot | Ukuran input | Top-1 (%) | Ukuran evaluasi | Kelas | Lisensi | Hub |
|---|---|---|---|---|---|---|
convnext_zepto_rms.ra4_e3600_r224_in1k | 224 | 72.820 | 224 | 1000 | apache-2.0 | timm |
convnext_zepto_rms_ols.ra4_e3600_r224_in1k | 224 | 73.234 | 224 | 1000 | apache-2.0 | timm |
Lainnya
| Bobot | Ukuran input | Top-1 (%) | Ukuran evaluasi | Kelas | Lisensi | Hub |
|---|---|---|---|---|---|---|
convnext_atto.d2_in1k | 224 (uji 288) | 77.016 | 288 | 1000 | apache-2.0 | timm |
convnext_atto_ols.a2_in1k | 224 (uji 288) | 77.220 | 288 | 1000 | apache-2.0 | timm |
convnext_nano.in12k | 224 | - | - | 11821 | apache-2.0 | timm |
convnext_nano.in12k_ft_in1k | 224 (uji 288) | 82.882 | 288 | 1000 | apache-2.0 | timm |
convnext_nano.r384_ad_in12k | 384 | - | - | 11821 | apache-2.0 | timm |
convnext_nano.r384_in12k | 384 | - | - | 11821 | apache-2.0 | timm |
convnext_nano.r384_in12k_ft_in1k | 384 | 83.334 | 384 | 1000 | apache-2.0 | timm |
convnext_small.in12k | 224 | - | - | 11821 | apache-2.0 | timm |
convnext_small.in12k_ft_in1k | 224 (uji 288) | 85.324 | 288 | 1000 | apache-2.0 | timm |
convnext_small.in12k_ft_in1k_384 | 384 | 86.186 | 384 | 1000 | apache-2.0 | timm |
convnext_tiny.in12k | 224 | - | - | 11821 | apache-2.0 | timm |
convnext_tiny.in12k_ft_in1k | 224 (uji 288) | 84.448 | 288 | 1000 | apache-2.0 | timm |
convnext_tiny.in12k_ft_in1k_384 | 384 | 85.154 | 384 | 1000 | apache-2.0 | timm |
convnext_tiny_hnf.a2h_in1k | 224 (uji 288) | 82.608 | 288 | 1000 | apache-2.0 | timm |
test_convnext.r160_in1k | 160 | 47.774 | 160 | 1000 | apache-2.0 | timm |
test_convnext2.r160_in1k | 160 | 53.520 | 160 | 1000 | apache-2.0 | timm |
test_convnext3.r160_in1k | 160 | 53.302 | 160 | 1000 | apache-2.0 | timm |
Sitasi
@Article{liu2022convnet,
author = {Zhuang Liu and Hanzi Mao and Chao-Yuan Wu and Christoph Feichtenhofer and Trevor Darrell and Saining Xie},
title = {A ConvNet for the 2020s},
journal = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
year = {2022},
}
@article{Woo2023ConvNeXtV2,
title={ConvNeXt V2: Co-designing and Scaling ConvNets with Masked Autoencoders},
author={Sanghyun Woo, Shoubhik Debnath, Ronghang Hu, Xinlei Chen, Zhuang Liu, In So Kweon and Saining Xie},
year={2023},
journal={arXiv preprint arXiv:2301.00808},
}