Nidhal-ch commited on
Commit
4c2ac72
·
verified ·
1 Parent(s): 266497c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -6
app.py CHANGED
@@ -79,12 +79,17 @@ hyperparameters = {
79
  }
80
 
81
  results_data = {
82
- 'SSD300': {'Transfer Pruning':{'model':["SSD300-ResNet50",'SSD300-ITPCC-A','SSD300-ITPCC-B','SSD300-ITPCC-C'],'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%"]},
83
- 'VIB Pruning':{'model':['SSD300-ResNet50','SSD300-VIB-v1','SSD300-VIB-v2'],'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],'paramsd':['0.0%','0.0%','0.0%'],}},
84
- 'SSD512': {'Transfer Pruning':{'model':["SSD512-ResNet50",'SSD512-ITPCC-A','SSD512-ITPCC-B','SSD512-ITPCC-C'],'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%"],},
85
- 'VIB Pruning':{'model':["SSD512-ResNet50",'SSD512-VIB-v1'],'map': [80.9, 0.65], 'flops': [1e9, 1.1e9],'flopsd':['0.0%','0.0%'], 'params': [1e6, 1.1e6],'paramsd':['0.0%','0.0%']}},
86
- 'DETR': {'SPARK':{'model':["DETR-baseline",'DETR-SPARK-A','DETR-SPARK-B'],'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],'paramsd':['0.0%','0.0%','0.0%'],},
87
- 'VOC':{'model':["DETR-baseline",'DETR-VOC-A','DETR-VOC-B'],'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],'paramsd':['0.0%','0.0%','0.0%'],}},
 
 
 
 
 
88
  }
89
 
90
  # Title of the research
 
79
  }
80
 
81
  results_data = {
82
+ 'SSD300': {
83
+ 'VIB Pruning':{'model':['SSD300-ResNet50','SSD300-VIB-v1','SSD300-VIB-v2'],'map': ["77.79", "78.71", "77.41"], 'flops': ["11.1", "5.04", "3.49"],'flopsd':['0.0%','54.55%','68.54%'], 'params': ["49.2", "19.84", "11.18"],'paramsd':['0.0%','59.68%','77.28%'],},
84
+ 'Transfer Pruning':{'model':["SSD300-ResNet50",'SSD300-ITPCC-A','SSD300-ITPCC-B','SSD300-ITPCC-C'],'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%"]},
85
+ }
86
+ 'SSD512': {
87
+ 'VIB Pruning':{'model':["SSD512-ResNet50",'SSD512-VIB-v1'],'map': ["80.9","81.43"], 'flops': ["46.24", "9.73"],'flopsd':['0.0%','78.94%'], 'params': ["58.52","27.2"],'paramsd':['0.0%','53.42%']},
88
+ 'Transfer Pruning':{'model':["SSD512-ResNet50",'SSD512-ITPCC-A','SSD512-ITPCC-B','SSD512-ITPCC-C'],'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%"],},
89
+ }
90
+
91
+ 'DETR': {'SPARK':{'model':["DETR-baseline",'DETR-SPARK-A','DETR-SPARK-B'],'map': ["96.77", "94.5", "95.18"], 'flops': ["85", "56", "58"],'flopsd':['0.0%','34.1%','31.7%'], 'params': ["41.2", "23.3", "26.6"],'paramsd':['0.0%','47.3%','45.4%'],},
92
+ 'VOC':{'model':["DETR-baseline",'DETR-VOC-A','DETR-VOC-B'],'map': ["79.34", "77.2", "78.0"], 'flops': ["85", "55", "60"],'flopsd':['0.0%','35.29%','29.41%'], 'params': ["41.2", "21.71", "22.47"],'paramsd':['0.0%','42.65%','35.5%'],}},
93
  }
94
 
95
  # Title of the research