Merge remote-tracking branch 'origin/master'

tdhooghe · Jun 9, 2022 · 04e75ae · 04e75ae
2 parents 59722d1 + 78af0dd
commit 04e75ae
Show file tree

Hide file tree

Showing 6 changed files with 13 additions and 6 deletions.
diff --git a/exp1_perf.py b/exp1_perf.py
@@ -85,8 +85,9 @@ def run_exp1_perf(models, precisions):
 
 
 if __name__ == "__main__":
-    run_exp1_perf(MODELS, PRECISION)
-    run_exp1_speed()
+    run_exp1_perf(['yolov5l6'], ['fp16', 'fp32'])
+    run_exp1_speed(['yolov5l6'], ['fp16', 'fp32'])
+
 
 
 
diff --git a/exp1_speed.py b/exp1_speed.py
@@ -9,16 +9,16 @@
 PRECISION = ['int8', 'fp16', 'fp32']
 
 
-def run_exp1_speed():
+def run_exp1_speed(models, precisions):
     column_names = ["model", "precision", "prep_time", "NMS_time", "latency", "inference_time",
                     "total_time",
                     "FPS", "experiment_time"]
     exp1_speed = pd.DataFrame(columns=column_names)
 
     counter = 0
-    for model in MODELS:
+    for model in models:
         imgsize = 1280 if '6' in model else 640
-        for precision in PRECISION:
+        for precision in precisions:
             if model == 'yolov5l6' and precision == 'int8':
                 break
             start_experiment = datetime.now()
@@ -54,4 +54,4 @@ def run_exp1_speed():
 
 
 if __name__ == "__main__":
-    run_exp1_speed()
+    run_exp1_speed(['yolov5l6'], ['fp16', 'fp32'])
diff --git a/results/experiments/exp1/20220609_exp1_perf.csv b/results/experiments/exp1/20220609_exp1_perf.csv
@@ -0,0 +1,3 @@
+,model,img_size,precision,mAP50,mAP,mAP50_8class,mAP_8class,person_mAP.5,car_mAP.5,motorcycle_mAP.5,bus_mAP.5,truck_mAP.5,baseball_bat_mAP.5,knife_mAP.5,cell_phone_mAP.5,person_mAP,car_mAP,motorcycle_mAP,bus_mAP,truck_mAP,baseball_bat_mAP,knife_mAP,cell_phone_mAP,experiment_time
+0,yolov5l6,1280,fp16,0.7061042757231983,0.5227280554509581,0.7434818681153712,0.5324955592698563,0.8598390698011174,0.787859037238027,0.8079996242721972,0.8850437886752031,0.6673509598451254,0.7169808653401972,0.521954379672731,0.7008272200783721,0.6350552259668647,0.5494377454819125,0.5568775414703249,0.7704109662868025,0.4866806525207699,0.4243228408447344,0.34254745134870773,0.49463205023873397,6032
+1,yolov5l6,1280,fp32,0.7059118815833638,0.5228428252450639,0.7431273980514019,0.5321621208044109,0.859570592146744,0.7882380991650857,0.8070124688242202,0.8860877959175956,0.6669472257287786,0.7179466397444454,0.5192162308709506,0.7000001320133953,0.6359903312155086,0.5497617085113294,0.5565396262999778,0.7719261950705294,0.48700068198104296,0.4183031495369353,0.34314265535611216,0.4946326184638516,11163
diff --git a/results/experiments/exp1/20220609_exp1_perf.pkl b/results/experiments/exp1/20220609_exp1_perf.pkl
diff --git a/results/experiments/exp1/220609_speed.csv b/results/experiments/exp1/220609_speed.csv
@@ -0,0 +1,3 @@
+,model,precision,prep_time,NMS_time,latency,inference_time,total_time,FPS,experiment_time
+0,yolov5l6,fp16,10.344834518432616,6.876286220550536,17.221120738983153,1136.2903309345245,1153.5114516735077,0.8669181381330942,5939
+1,yolov5l6,fp32,10.763452577590941,7.065650749206543,17.829103326797483,2182.0223227500915,2199.8514260768893,0.45457615371023147,11185
diff --git a/results/experiments/exp1/220609_speed.pkl b/results/experiments/exp1/220609_speed.pkl