mukhbiir commited on
Commit
2625ac2
·
verified ·
1 Parent(s): 6d98567

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +34 -0
README.md ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - image-classification
5
+ - swin-transformer
6
+ - swin-v2
7
+ - vision
8
+ - gradients
9
+ - jet-colormap
10
+ ---
11
+
12
+ # SwinV2-Large Classifier (384x384, Jet-Colored Gradients)
13
+
14
+ 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.
15
+
16
+ ## 🧩 Model Details
17
+
18
+ - **Architecture**: SwinV2-Large (384x384 input resolution)
19
+ - **Framework**: PyTorch
20
+ - **Training Data**: Jet-colored gradients (Mukhbir dataset on kaggle)
21
+ - **Use case**: Classification of Real or Fake(Gan)
22
+
23
+ ## 🛠️ How to Use
24
+
25
+ ```python
26
+ from huggingface_hub import hf_hub_download
27
+ import torch
28
+
29
+ # Download the model
30
+ model_path = hf_hub_download(repo_id="mukhbiir/Swin_Classifier", filename="model.pt")
31
+ model = torch.load(model_path)
32
+ model.eval()
33
+ ```
34
+