bread-or-dog / README.md
xhiroga's picture
Upload folder using huggingface_hub
9f934ac verified
---
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
```powershell
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
```powershell
# Before Fine-Tuning
python vgg16.py
# After Fine-Tunin
python app.py
```
## Deploy
```powershell
gradio deploy
```
## References
- [PyTorchの学習済みモデルで画像分類(VGG, ResNetなど) | note.nkmk.me](https://note.nkmk.me/python-pytorch-pretrained-models-image-classification/)
- [PyTorchによるファインチューニングの実装 - 機械学習ともろもろ](https://venoda.hatenablog.com/entry/2020/10/18/014516)