Got all inference times and plotted

antoninoLorenzo · Jul 5, 2024 · 74f7114 · 74f7114
1 parent 111daad
commit 74f7114
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/static/images/inference_times_plot_RTX.png b/static/images/inference_times_plot_RTX.png
diff --git a/static/inference_times_plots.py b/static/inference_times_plots.py
@@ -9,8 +9,9 @@
 if __name__ == "__main__":
     CASES = ['conversion', 'inference']  # inference == planning
     GPUS = ['GTX-1660-Ti', 'RTX-3080']
+    # GPUS = ['RTX-3080']
 
-    fig, axes = plt.subplots(nrows=2, ncols=2, figsize=(16, 16))
+    fig, axes = plt.subplots(nrows=len(CASES), ncols=len(GPUS), figsize=(16, 16))
     fig.suptitle('Inference Times')
     sns.set(style="whitegrid")
 
@@ -38,7 +39,10 @@
     for i, case in enumerate(CASES):
         for j, gpu in enumerate(GPUS):
             df = times[f'{case}_{gpu}']
-            ax = axes[i, j]
+            if len(GPUS) > 1:
+                ax = axes[i, j]
+            else:
+                ax = axes[i]
             sns.barplot(x='model', y='time', data=df, ax=ax)
 
             ax.set_title(f'{case[:1].upper()}{case[1:]} Times ({gpu})')
@@ -48,7 +52,7 @@
             ax.tick_params(axis='x', rotation=45)
 
     plt.savefig(
-        './images/inference_times_plot.png',
+        './images/inference_times_plot_RTX.png',
         dpi=300,
         bbox_inches='tight'
     )