prakashchhipa commited on
Commit
8fe897f
1 Parent(s): a345821

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +42 -1
README.md CHANGED
@@ -1,4 +1,45 @@
1
  ---
2
  license: apache-2.0
 
 
 
 
 
 
 
 
 
 
 
 
3
  ---
4
- MPD integrated DINO and SimCLR models coming soon!
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
+ datasets:
4
+ - ILSVRC/imagenet-1k
5
+ language:
6
+ - en
7
+ pipeline_tag: image-classification
8
+ tags:
9
+ - Robust SSL
10
+ - DINO
11
+ - SimCLR
12
+ - Perspective Distortion
13
+ - MPD
14
+ - ImageNet-PD
15
  ---
16
+ **Self-Supervised Pretrained Models with MPD Integration**
17
+
18
+ Publication- *Möbius Transform for Mitigating Perspective Distortions in Representation Learning, European Conference on Computer Vision (ECCV 2024)*
19
+
20
+
21
+ **Model Description**
22
+
23
+ This release includes two self-supervised pretrained models integrated with the Mitigating Perspective Distortion (MPD) technique. The models are:
24
+
25
+ 1. ResNet50 pretrained using SimCLR (Simple Framework for Contrastive Learning of Visual Representations)
26
+ 2. ViT-small pretrained using DINO (Self-Distillation with No Labels)
27
+
28
+ Both models were trained with a *batch size of 512* over *100 epochs*. The MPD method enhances the robustness of these models by simulating real-world perspective distortions, making them more robust in various computer vision tasks.
29
+
30
+
31
+
32
+ **Training Details**
33
+
34
+ Algorithms- SimCLR for ResNet50, DINO for ViT-small
35
+ Batch Size- 512
36
+ Epochs- 100
37
+ Perspective Distortion Synthesis: MPD
38
+
39
+ **Performance**
40
+
41
+ The integration of MPD in both SimCLR and DINO frameworks significantly improves the models' performance on tasks affected by perspective distortion. The models can be used directly for downstream tasks or further fine-tuned for specific applications. Refer results in MPD paper.
42
+
43
+ **Source Code**
44
+
45
+ Two minutes summary on MPD and links to access source code repository and ImageNet-PD bacnhmark are available at https://prakashchhipa.github.io/projects/mpd/