README.md

metadata

title: bread-or-dog
app_file: app.py
sdk: gradio
sdk_version: 4.19.0

Fine-Tuning VGG16 to detect bread or dog

See <>.

Prerequisites

conda env create -f environment.yml
conda activate fine-tuning-vgg16-bread-or-dog
Invoke-WebRequest -Uri https://raw.githubusercontent.com/anishathalye/imagenet-simple-labels/master/imagenet-simple-labels.json -OutFile ./data/imagenet-simple-labels.json

Demo

# Before Fine-Tuning
python vgg16.py

# After Fine-Tunin
python app.py

Deploy

gradio deploy

References

• PyTorchの学習済みモデルで画像分類（VGG, ResNetなど） | note.nkmk.me
• PyTorchによるファインチューニングの実装 - 機械学習ともろもろ