戰(zhàn):timm.create_model 調(diào)用與 Top5 預(yù)測詳解)
【免費(fèi)下載鏈接】res2net50_26w_4s.in1k項(xiàng)目地址https://ai.gitcode.com/hf_mirrors/timm/res2net50_26w_4s.in1k點(diǎn)擊查看免費(fèi)下載res2net50_26w_4s.in1k是一個(gè)在 ImageNet-1k 上預(yù)訓(xùn)練的 Res2Net 圖像分類模型可通過 timm 庫一行代碼加載用于圖片識別與 Top5 分類預(yù)測。本文帶你從零開始完成下載模型、timm.create_model調(diào)用、數(shù)據(jù)預(yù)處理和 Top5 預(yù)測輸出的完整流程。一、模型概覽res2net50_26w_4s.in1k 是什么Res2Net 是一種多尺度Multi-Scale殘差骨干網(wǎng)絡(luò)它在 ResNet 基礎(chǔ)上把每個(gè)通道再拆分為多個(gè)尺度特征使網(wǎng)絡(luò)能同時(shí)感知局部細(xì)節(jié)和整體結(jié)構(gòu)因此在分類任務(wù)上表現(xiàn)更穩(wěn)。關(guān)鍵指標(biāo)數(shù)值說明參數(shù)量25.7 M輕量級適合 CPU/GPU 快速推理計(jì)算量4.3 GMACs推理開銷低輸入尺寸224 × 224標(biāo)準(zhǔn) ImageNet 輸入分類數(shù)1000 類ImageNet-1k 全類別預(yù)訓(xùn)練數(shù)據(jù)ImageNet-1k論文作者官方權(quán)重模型命名解讀res2net50_26w_4s表示 50 層結(jié)構(gòu)的 Res2Net每個(gè)殘差塊按26w_4s的比例把通道劃分為 4 個(gè)尺度.in1k后綴表示該權(quán)重在 ImageNet 1000 類數(shù)據(jù)集上訓(xùn)練得到。 模型卡與完整用法說明見 README.md架構(gòu)參數(shù)見 config.json。二、下載模型權(quán)重Git LFS 說明如果希望本地保留權(quán)重文件可以 clone 倉庫需安裝 Git LFS否則下載到的是指針文件git lfs install git clone https://gitcode.com/hf_mirrors/timm/res2net50_26w_4s.in1k倉庫內(nèi)兩個(gè)權(quán)重文件互為等價(jià)格式model.safetensorssafetensors 格式權(quán)重約 98 MB加載更安全、更快pytorch_model.bin傳統(tǒng) PyTorch pickle 格式權(quán)重config.json模型架構(gòu)與預(yù)處理配置架構(gòu)名、輸入尺寸、mean/std 等 實(shí)際上你不需要手動(dòng)下載在 timm 中指定pretrainedTrue時(shí)權(quán)重會(huì)自動(dòng)緩存到本地。三、最快上手timm.create_model 一行調(diào)用1. 安裝依賴pip install timm torch pillow2. 創(chuàng)建模型import timm # 一行代碼創(chuàng)建模型pretrainedTrue 自動(dòng)下載 in1k 預(yù)訓(xùn)練權(quán)重 model timm.create_model(res2net50_26w_4s.in1k, pretrainedTrue) model model.eval() # 切換到推理模式timm.create_model的核心參數(shù)說明參數(shù)示例值作用pretrainedTrue加載 ImageNet-1k 預(yù)訓(xùn)練權(quán)重num_classes0移除分類頭只輸出特征向量features_onlyTrue只提取各階段特征圖3. 自動(dòng)獲取數(shù)據(jù)預(yù)處理管線每個(gè)模型有自己訓(xùn)練時(shí)的歸一化參數(shù)。不要手寫mean/std讓 timm 自動(dòng)解析data_config timm.data.resolve_model_data_config(model) transforms timm.data.create_transform(**data_config, is_trainingFalse)對于 res2net50_26w_4s.in1k該管線會(huì)自動(dòng)完成中心裁剪 87.5% → 縮放到 224×224 → 按 ImageNet 均值/標(biāo)準(zhǔn)差歸一化。這些參數(shù)mean[0.485, 0.456, 0.406]、std[0.229, 0.224, 0.225]、crop_pct0.875都來自 config.json 中的pretrained_cfg字段。四、Top5 預(yù)測詳解從圖片到概率分?jǐn)?shù)下面是對一張圖片做 Top5 分類預(yù)測的完整流程import torch from PIL import Image img Image.open(cat.jpg).convert(RGB) output model(transforms(img).unsqueeze(0)) # 單圖補(bǔ) batch 維 top5_prob, top5_idx torch.topk(output.softmax(dim1) * 100, k5) print(top5_prob) print([model.decode_classes(i) for i in top5_idx])逐步拆解unsqueeze(0)模型輸入要求形狀為(batch, 3, H, W)單張圖片需要補(bǔ)一個(gè) batch 維度softmax(dim1)模型輸出的是 1000 維 logits未歸一化分?jǐn)?shù)經(jīng) softmax 后每類和為 1可解釋為屬于該類別的概率torch.topk(..., k5)取出概率最高的 5 個(gè)類別返回(top5_probabilities, top5_class_indices)兩個(gè)張量decode_classes可選把類別索引翻譯成 ImageNet 類別名稱如n02123045 (golden retriever)。?? 注意model.decode_classes()需要 timm 已緩存imagenet.py類別表pip install timm自帶如果你只做自定義分類微調(diào)索引對應(yīng)的是你自己的類別表。五、進(jìn)階用法速覽提取特征圖用于分割、檢測骨干model timm.create_model(res2net50_26w_4s.in1k, pretrainedTrue, features_onlyTrue)輸出 5 組逐級降采樣特征圖形狀依次為[1,64,112,112]→[1,256,56,56]→[1,512,28,28]→[1,1024,14,14]→[1,2048,7,7]。提取圖像向量用于檢索、聚類model timm.create_model(res2net50_26w_4s.in1k, pretrainedTrue, num_classes0) embedding model(transforms(img).unsqueeze(0)) # 輸出 2048 維向量num_classes0會(huì)移除fc分類層直接輸出 2048 維全局特征向量可用于相似圖片檢索等下游任務(wù)。六、常見問題 FAQQ1為什么輸入必須是 224×224config.json中input_size為[3, 224, 224]。雖然fixed_input_sizefalse允許其他尺寸用雙線性插值池化但尺寸偏離越大精度損失越明顯建議保持 224×224。Q2pretrainedTrue會(huì)重復(fù)下載權(quán)重嗎不會(huì)。timm 會(huì)將權(quán)重緩存到~/.cache/torch/hub第二次調(diào)用直接走本地緩存。Q3.in1k后綴還能換成別的嗎t(yī)imm 中同一架構(gòu)可能提供advprop對抗魯棒版等變體后綴對于 res2net50_26w_4s 系列.in1k是官方推薦的主力權(quán)重其他變體可用timm.list_models(*res2net*)查看。Q4CPU 上推理慢怎么辦該模型僅 25.7M 參數(shù)、4.3 GMACs本身很輕??蓪⑤斎肟s到 224×224 以下或使用torch.no_grad()包裹推理model.eval() 無梯度上下文以避免不必要的計(jì)算圖構(gòu)建。七、文件清單與延伸閱讀文件說明README.md模型卡模型信息、三種用法示例分類/特征/向量、論文引用config.json架構(gòu)名、1000 類、2048 維特征、歸一化參數(shù)等配置model.safetensorssafetensors 權(quán)重Git LFS 管理pytorch_model.binPyTorch 權(quán)重Git LFS 管理 小結(jié)timm.create_model(res2net50_26w_4s.in1k, pretrainedTrue)create_transform自動(dòng)預(yù)處理 softmax → topk三步就能完成一個(gè)工業(yè)級 ImageNet 圖像分類與 Top5 預(yù)測流水線。接下來你可以嘗試把分類頭替換為自己的類別數(shù)在少量數(shù)據(jù)上微調(diào)出專用分類器。贊分享【免費(fèi)下載鏈接】res2net50_26w_4s.in1k項(xiàng)目地址https://ai.gitcode.com/hf_mirrors/timm/res2net50_26w_4s.in1k點(diǎn)擊查看免費(fèi)下載相關(guān)推薦efficientnet_em.ra2_in1k 圖像分類實(shí)戰(zhàn)教程Top5預(yù)測與數(shù)據(jù)預(yù)處理完整演示efficientnet_em.ra2_in1k 圖像分類實(shí)戰(zhàn)教程Top5預(yù)測與數(shù)據(jù)預(yù)處理完整演示 efficientnet_em.ra2_in1k 是 ti從加載圖片到TOP5預(yù)測tf_efficientnetv2_b3.in21k_ft_in1k圖像分類實(shí)戰(zhàn)教程附完整代碼從加載圖片到TOP5預(yù)測tf_efficientnetv2_b3.in21k_ft_in1k圖像分類實(shí)戰(zhàn)教程附完整代碼 本教程帶你使用 timm 中的預(yù)訓(xùn)人工智能計(jì)算機(jī)視覺Habitat-Lab 快速上手從安裝到訓(xùn)練具身 AI 智能體的 5 條主線Habitat Lab 快速上手從安裝到訓(xùn)練具身 AI 智能體的 5 條主線 Habitat Lab 是 Meta 開源的模塊化具身 AI 庫基于 Habi人工智能具身智能強(qiáng)化學(xué)習(xí)上一篇從TOML到LuaPebrel Lua配置系統(tǒng)完全教程含熱重載與校驗(yàn)實(shí)戰(zhàn)下一篇ng-zorro-antd TreeView 樹視圖組件完全指南從數(shù)據(jù)源接入到虛擬滾動(dòng)的源碼級實(shí)踐創(chuàng)作聲明:本文部分內(nèi)容由AI輔助生成(AIGC),僅供參考