Downloads · 30 days
8
12% of all-time downloads
ViT-Walkability
AEnigmista/ViT-Walkability
ViT-Walkability is a image classification model from AEnigmista. Use it when you need a label for an image. It is set up for pytorch. The card lists the license as mit.
This repository contains a fine-tuned Vision Transformer model for walkability-related image classification tasks.
All-time downloads
69
Public
Parameters
85.8M
687 MB on disk
Likes
1
Public
Repo makeup
Click a slice to open those files.
.pt343 MB · 50%
At a glance
- Task
- Image Classification
- Library
- pytorch
- License
- mit
- Model type
- vit
- Access
- Public
- Created
- May 21, 2026
- Updated
- May 21, 2026
- SHA
- 30ecbc29
Model card
From the Hugging Face model README
ViT-Walkability
This repository contains a fine-tuned Vision Transformer model for walkability-related image classification tasks.
Checkpoint details
- Source file:
best.pt - Top-level type:
dict - Base model:
google/vit-base-patch16-224 - Num labels:
5 - Labels:
['LABEL_0', 'LABEL_1', 'LABEL_2', 'LABEL_3', 'LABEL_4'] - Keys:
['model_state_dict', 'epoch']
Loading notes
The uploaded files are in standard Hugging Face Transformers format and can be loaded directly with from_pretrained.
- Task
- Image Classification
- Library
- pytorch
- Type
- vit
- License
- mit
- Created
- May 21, 2026
- Updated
- May 21, 2026