Resnet 152

🚀 ResNet-152 v1.5

ImageNet-1kで事前学習された画像分類用のResNetモデルです。

🚀 クイックスタート

このモデルは、224x224の解像度でImageNet-1kデータセットを使って事前学習されたResNetモデルです。Heらによる論文 Deep Residual Learning for Image Recognition で紹介されました。

✨ 主な機能

モデル概要

ResNet（Residual Network）は、残差学習とスキップ接続の概念を広めた畳み込みニューラルネットワークです。これにより、より深いモデルの学習が可能になります。

これはResNet v1.5で、元のモデルとは異なります。ダウンサンプリングが必要なボトルネックブロックでは、v1では最初の1x1畳み込みでstride = 2ですが、v1.5では3x3畳み込みでstride = 2になっています。この違いにより、ResNet50 v1.5はv1よりもわずかに精度が高く（~0.5% top1）、Nvidiaによると、少しのパフォーマンスの低下（~5% imgs/sec）があります。

想定用途と制限

この生モデルは画像分類に使用できます。関心のあるタスクで微調整されたバージョンを探すには、モデルハブを参照してください。

💻 使用例

基本的な使用法

以下は、このモデルを使ってCOCO 2017データセットの画像を1,000のImageNetクラスのいずれかに分類する方法です。

from transformers import AutoFeatureExtractor, ResNetForImageClassification
import torch
from datasets import load_dataset

dataset = load_dataset("huggingface/cats-image")
image = dataset["test"]["image"][0]

feature_extractor = AutoFeatureExtractor.from_pretrained("microsoft/resnet-152")
model = ResNetForImageClassification.from_pretrained("microsoft/resnet-152")

inputs = feature_extractor(image, return_tensors="pt")

with torch.no_grad():
    logits = model(**inputs).logits

# model predicts one of the 1000 ImageNet classes
predicted_label = logits.argmax(-1).item()
print(model.config.id2label[predicted_label])

より多くのコード例については、ドキュメントを参照してください。

BibTeX引用

@inproceedings{he2016deep,
  title={Deep residual learning for image recognition},
  author={He, Kaiming and Zhang, Xiangyu and Ren, Shaoqing and Sun, Jian},
  booktitle={Proceedings of the IEEE conference on computer vision and pattern recognition},
  pages={770--778},
  year={2016}
}

📄 ライセンス

このモデルはApache-2.0ライセンスの下で提供されています。

⚠️ 重要提示

ResNetをリリースしたチームはこのモデルのモデルカードを作成していないため、このモデルカードはHugging Faceチームによって作成されました。