timm documentation

TResNet

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v1.0.31).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

TResNet

Kategori: CNN · Kode: timm/models/tresnet.py · Arsitektur: 4 · Bobot pralatih: 10 (8 dengan hasil ImageNet-1k)

TResNet adalah desain berbasis ResNet untuk throughput GPU yang tinggi, dengan stem space-to-depth, downsampling anti-alias, dan lapisan SE.

Makalah

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('tresnet_m.miil_in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)

Dukungan API Model

FiturTResNet
feature maps features_only=True✓ (strides 4, 8, 16, 32)
forward_intermediates()✓
prune_intermediate_layers()✓
Ukuran input apa pun, tidak perlu perubahan✓
set_input_size()–
dynamic_img_size=True–
set_grad_checkpointing()✓
Penurunan LR per lapisan (group_matcher())✓
reset_classifier()✓

✓ 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 (TResNet dengan tresnet_m pada 224), arsitektur lain dari kelas yang sama diharapkan berperilaku sama. Lihat halaman Model API untuk detailnya.

Arsitektur

ArsitekturParameter (M)GMACsUkuran inputBobot terlatih
tresnet_l55.9910.902242
tresnet_m31.395.752244
tresnet_v2_l46.178.852242
tresnet_xl78.4415.202242

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.

BobotUkuran inputTop-1 (%)Ukuran evaluasiKelasLisensiHub
tresnet_l.miil_in1k22481.4982241000apache-2.0timm
tresnet_l.miil_in1k_44844882.2784481000apache-2.0timm
tresnet_m.miil_in1k22480.8162241000apache-2.0timm
tresnet_m.miil_in1k_44844881.7104481000apache-2.0timm
tresnet_m.miil_in21k224--11221apache-2.0timm
tresnet_m.miil_in21k_ft_in1k22483.0902241000apache-2.0timm
tresnet_v2_l.miil_in21k224--11221apache-2.0timm
tresnet_v2_l.miil_in21k_ft_in1k22483.9102241000apache-2.0timm
tresnet_xl.miil_in1k22482.0602241000apache-2.0timm
tresnet_xl.miil_in1k_44844883.0664481000apache-2.0timm
Update on GitHub