Merge pull request #183 from KushajveerSingh/results_diff

Add rank difference, and top-1/top-5 difference from ImageNet-1k validation for the 3 additional test sets

Author: rwightman

results/README.md

@@ -35,5 +35,4 @@ An ImageNet test set of 10,000 images sampled from new images roughly 10 years a
 A collection of 7500 images covering 200 of the 1000 ImageNet classes. Images are naturally occuring adversarial examples that confuse typical ImageNet classifiers. This is a challenging dataset, your typical ResNet-50 will score 0% top-1.
 
 ## TODO
-* Add rank difference, and top-1/top-5 difference from ImageNet-1k validation for the 3 additional test sets
 * Explore adding a reduced version of ImageNet-C (Corruptions) and ImageNet-P (Perturbations) from https://github.com/hendrycks/robustness. The originals are huge and image size specific.

results/generate_csv_results.py

@@ -0,0 +1,52 @@
+import numpy as np
+import pandas as pd
+
+results = {
+    'results-imagenet.csv'                    : pd.read_csv('results-imagenet.csv'), 
+    'results-imagenetv2-matched-frequency.csv': pd.read_csv('results-imagenetv2-matched-frequency.csv'),
+    'results-sketch.csv'                      : pd.read_csv('results-sketch.csv'),
+    'results-imagenet-a.csv'                  : pd.read_csv('results-imagenet-a.csv'),
+}
+
+def diff(csv_file):    
+    base_models = results['results-imagenet.csv']['model'].values
+    csv_models  = results[csv_file]['model'].values
+
+    rank_diff = np.zeros_like(csv_models, dtype='object')
+    top1_diff = np.zeros_like(csv_models, dtype='object')
+    top5_diff = np.zeros_like(csv_models, dtype='object')
+    
+    for rank, model in enumerate(csv_models):
+        if model in base_models:            
+            base_rank = int(np.where(base_models==model)[0])
+            top1_d = results[csv_file]['top1'][rank]-results['results-imagenet.csv']['top1'][base_rank]
+            top5_d = results[csv_file]['top5'][rank]-results['results-imagenet.csv']['top5'][base_rank]
+            
+            # rank_diff
+            if   rank == base_rank: rank_diff[rank] = f'='
+            elif rank >  base_rank: rank_diff[rank] = f'-{rank-base_rank}'
+            else:                   rank_diff[rank] = f'+{base_rank-rank}'
+                
+            # top1_diff
+            if top1_d >= .0: top1_diff[rank] = f'+{top1_d:.3f}'
+            else           : top1_diff[rank] = f'-{abs(top1_d):.3f}'
+            
+            # top5_diff
+            if top5_d >= .0: top5_diff[rank] = f'+{top5_d:.3f}'
+            else           : top5_diff[rank] = f'-{abs(top5_d):.3f}'
+                
+        else: 
+            rank_diff[rank] = 'X'
+            top1_diff[rank] = 'X'
+            top5_diff[rank] = 'X'
+    
+    results[csv_file]['rank_diff'] = rank_diff
+    results[csv_file]['top1_diff'] = top1_diff
+    results[csv_file]['top5_diff'] = top5_diff
+    
+    results[csv_file]['param_count'] = results[csv_file]['param_count'].map('{:,.2f}'.format)
+
+    results[csv_file].to_csv(csv_file, index=False, float_format='%.3f')
+
+for csv_file in results:
+    if csv_file != 'results-imagenet.csv': diff(csv_file)