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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -80,7 +80,7 @@ hyperparameters = {
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, 1.3e6],'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%'],},
 
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%'],},