Resnet18 Catdog Classifier

🚀 Resnet18 猫犬分類器

このモデルは、カスタムデータセットで微調整された画像分類モデルで、猫と犬の画像を高精度に分類することができます。

🚀 クイックスタート

このモデルは、microsoft/resnet-18 を カスタム データセットで微調整したバージョンです。モデルの構築には、Kaggleから取得した「Cats & Dogs Classification」データセットを使用しました。モデル構築の過程では、事前学習済みのResnet-18を用いたPytorchフレームワークを使用し、分類モデルの構築にはデータセットを用いた微調整手法を採用しています。

✨ 主な機能

• 猫と犬の画像を高精度に分類することができます。
• 事前学習済みのResnet-18をベースに微調整されているため、少ないデータでも良好な性能を発揮します。

📚 ドキュメント

学習結果

学習手順

学習ハイパーパラメータ

学習時には以下のハイパーパラメータを使用しました。

• loss_function = CrossEntropyLoss
• optimizer = AdamW
• learning_rate: 0.0001
• batch_size: 16
• num_epochs: 10

フレームワークバージョン

• Transformers 4.33.2
• Pytorch 2.0.1+cu118
• Datasets 2.14.5
• Tokenizers 0.13.3

📄 ライセンス

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