Update app.py

app.py

@@ -79,12 +79,12 @@ hyperparameters = {
 }
 
 results_data = {
-    'SSD300': {'Transfer Pruning':{'map': [0.6, 0.65, 0.7, 0.75], 'flops': [1e9, 1.1e9, 1.2e9, 1.3e9], 'params': [1e6, 1.1e6, 1.2e6, 1.3e6]},
-              'VIB Pruning':{'map': [0.9, 0.65, 0.7, 0.75], 'flops': [1e9, 1.1e9, 1.2e9, 1.3e9], 'params': [1e6, 1.1e6, 1.2e6, 1.3e6]}},
-    'SSD512': {'Transfer Pruning':{'map': [0.6, 0.65, 0.7, 0.75], 'flops': [1e9, 1.1e9, 1.2e9, 1.3e9], 'params': [1e6, 1.1e6, 1.2e6, 1.3e6]},
-              'VIB Pruning':{'map': [0.9, 0.65, 0.7, 0.75], 'flops': [1e9, 1.1e9, 1.2e9, 1.3e9], 'params': [1e6, 1.1e6, 1.2e6, 1.3e6]}},
-    'DETR': {'SPARK':{'map': [0.6, 0.65, 0.7, 0.75], 'flops': [1e9, 1.1e9, 1.2e9, 1.3e9], 'params': [1e6, 1.1e6, 1.2e6, 1.3e6]},
-              'VOC':{'map': [0.9, 0.65, 0.7, 0.75], 'flops': [1e9, 1.1e9, 1.2e9, 1.3e9], 'params': [1e6, 1.1e6, 1.2e6, 1.3e6]}},
+    'SSD300': {'Transfer Pruning':{'map': [77.79, 77.86 , 77.06, 75.08], 'flops': [11.1, 6.85, 5.08, 3.38],'flopsd':['0.0%','0.0%','0.0%',"0.0%"], 'params': [49.2, 32.5, 25.7, 19.4],'paramsd':['0.0%','0.0%','0.0%',"0.0%"]},
+              'VIB Pruning':{'map': [77.79, 0.65, 0.7], 'flops': [1e9, 1.1e9, 1.2e9],'flopsd':['0.0%','0.0%','0.0%'], 'params': [1e6, 1.1e6, 1.2e6, 1.3e6],'paramsd':['0.0%','0.0%','0.0%'],}},
+    'SSD512': {'Transfer Pruning':{'map': [80.9, 81.05 , 80.45, 78.82], 'flops': [46.2, 31.42, 25.6, 20.1],'flopsd':['0.0%','0.0%','0.0%',"0.0%"], 'params': [58.5, 41.8, 35.0, 28.7],'paramsd':['0.0%','0.0%','0.0%',"0.0%"],},
+              'VIB Pruning':{'map': [80.9, 0.65], 'flops': [1e9, 1.1e9],'flopsd':['0.0%','0.0%'], 'params': [1e6, 1.1e6],'flopsd':['0.0%','0.0%']}},
+    'DETR': {'SPARK':{'map': [0.6, 0.65, 0.7], 'flops': [1e9, 1.1e9, 1.2e9],'flopsd':['0.0%','0.0%','0.0%'], 'params': [1e6, 1.1e6, 1.2e6],'flopsd':['0.0%','0.0%','0.0%'],},
+              'VOC':{'map': [0.9, 0.65, 0.7], 'flops': [1e9, 1.1e9, 1.2e9],'flopsd':['0.0%','0.0%','0.0%'], 'params': [1e6, 1.1e6, 1.2e6],'flopsd':['0.0%','0.0%','0.0%'],}},
 }
 
 # Title of the research
@@ -125,14 +125,18 @@ with col1:
             'Model': [model]*4,
             'mAP': results[pruning]['map'],
             'FLOPs': results[pruning]['flops'],
-            'Params': results[pruning]['params']
+            'down by': results[pruning]['flopsd'],
+            'Params': results[pruning]['params'],
+            'down by': results[pruning]['paramsd'],
         })
     else:
         df_results = pd.DataFrame({
             'Model': [model]*4,
             'mAP': results[dataset]['map'],
             'FLOPs': results[dataset]['flops'],
+            'down by': results[dataset]['flopsd'],
             'Params': results[dataset]['params']
+            'down by': results[dataset]['paramsd'],
         })