File size: 931 Bytes
2625ac2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
---
license: mit
tags:
- image-classification
- swin-transformer
- swin-v2
- vision
- gradients
- jet-colormap
---
# SwinV2-Large Classifier (384x384, Jet-Colored Gradients)
This model is a **SwinV2-Large (384x384)** vision transformer trained for image classification on **Jet-colored gradient maps**. The model learns to identify visual patterns in synthetic or colormap-encoded data to be suitable for detecting GAN generated images.
## 🧩 Model Details
- **Architecture**: SwinV2-Large (384x384 input resolution)
- **Framework**: PyTorch
- **Training Data**: Jet-colored gradients (Mukhbir dataset on kaggle)
- **Use case**: Classification of Real or Fake(Gan)
## 🛠️ How to Use
```python
from huggingface_hub import hf_hub_download
import torch
# Download the model
model_path = hf_hub_download(repo_id="mukhbiir/Swin_Classifier", filename="model.pt")
model = torch.load(model_path)
model.eval()
```
|