https://github.com/bessouat40/vit-transformers-finetuning

vision-transformers-finetunning
https://github.com/bessouat40/vit-transformers-finetuning

artificial-intelligence computer-vision fine-tuning huggingface-transformers python transformers

Last synced: about 2 months ago
JSON representation

vision-transformers-finetunning

• Host: GitHub
• URL: https://github.com/bessouat40/vit-transformers-finetuning
• Owner: Bessouat40
• Created: 2023-04-30T18:55:11.000Z (over 1 year ago)
• Default Branch: main
• Last Pushed: 2024-02-22T16:50:59.000Z (11 months ago)
• Last Synced: 2024-02-22T18:01:16.670Z (11 months ago)
• Topics: artificial-intelligence, computer-vision, fine-tuning, huggingface-transformers, python, transformers
• Language: Python
• Homepage:
• Size: 257 KB
• Stars: 1
• Watchers: 2
• Forks: 0
• Open Issues: 2
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# Fine tunning of a ViT transformers model for image classification

ViT is a transformers model that can be used for image classification.

This repository will show you how to fine tune this model with your own database.

## Requirements

### Python dependencies

First you need to install all python libraries :

```bash

python -m pip install -r requirements.txt

```

### Data folder structure

Create 3 subfolders : `test`, `train`, `val`.

Into this three folders, create a folder for each category of your data.

## Usage

```bash

python runner.py

```

## Inference Test

```bash

python inferenceTest.py

```